Few-shot Learner Parameterization by Diffusion Time-steps

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Yue, Zhongqi, Zhou, Pan, Hong, Richang, Zhang, Hanwang, Sun, Qianru
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909151949488128
author Yue, Zhongqi
Zhou, Pan
Hong, Richang
Zhang, Hanwang
Sun, Qianru
author_facet Yue, Zhongqi
Zhou, Pan
Hong, Richang
Zhang, Hanwang
Sun, Qianru
contents Even when using large multi-modal foundation models, few-shot learning is still challenging -- if there is no proper inductive bias, it is nearly impossible to keep the nuanced class attributes while removing the visually prominent attributes that spuriously correlate with class labels. To this end, we find an inductive bias that the time-steps of a Diffusion Model (DM) can isolate the nuanced class attributes, i.e., as the forward diffusion adds noise to an image at each time-step, nuanced attributes are usually lost at an earlier time-step than the spurious attributes that are visually prominent. Building on this, we propose Time-step Few-shot (TiF) learner. We train class-specific low-rank adapters for a text-conditioned DM to make up for the lost attributes, such that images can be accurately reconstructed from their noisy ones given a prompt. Hence, at a small time-step, the adapter and prompt are essentially a parameterization of only the nuanced class attributes. For a test image, we can use the parameterization to only extract the nuanced class attributes for classification. TiF learner significantly outperforms OpenCLIP and its adapters on a variety of fine-grained and customized few-shot learning tasks. Codes are in https://github.com/yue-zhongqi/tif.
format Preprint
id arxiv_https___arxiv_org_abs_2403_02649
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Few-shot Learner Parameterization by Diffusion Time-steps
Yue, Zhongqi
Zhou, Pan
Hong, Richang
Zhang, Hanwang
Sun, Qianru
Computer Vision and Pattern Recognition
Even when using large multi-modal foundation models, few-shot learning is still challenging -- if there is no proper inductive bias, it is nearly impossible to keep the nuanced class attributes while removing the visually prominent attributes that spuriously correlate with class labels. To this end, we find an inductive bias that the time-steps of a Diffusion Model (DM) can isolate the nuanced class attributes, i.e., as the forward diffusion adds noise to an image at each time-step, nuanced attributes are usually lost at an earlier time-step than the spurious attributes that are visually prominent. Building on this, we propose Time-step Few-shot (TiF) learner. We train class-specific low-rank adapters for a text-conditioned DM to make up for the lost attributes, such that images can be accurately reconstructed from their noisy ones given a prompt. Hence, at a small time-step, the adapter and prompt are essentially a parameterization of only the nuanced class attributes. For a test image, we can use the parameterization to only extract the nuanced class attributes for classification. TiF learner significantly outperforms OpenCLIP and its adapters on a variety of fine-grained and customized few-shot learning tasks. Codes are in https://github.com/yue-zhongqi/tif.
title Few-shot Learner Parameterization by Diffusion Time-steps
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.02649