Discriminative Image Generation with Diffusion Models for Zero-Shot Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Fu, Dingjie, Hou, Wenjin, Chen, Shiming, Chen, Shuhuang, You, Xinge, Khan, Salman, Khan, Fahad Shahbaz
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866908171545608192
author Fu, Dingjie
Hou, Wenjin
Chen, Shiming
Chen, Shuhuang
You, Xinge
Khan, Salman
Khan, Fahad Shahbaz
author_facet Fu, Dingjie
Hou, Wenjin
Chen, Shiming
Chen, Shuhuang
You, Xinge
Khan, Salman
Khan, Fahad Shahbaz
contents Generative Zero-Shot Learning (ZSL) methods synthesize class-related features based on predefined class semantic prototypes, showcasing superior performance. However, this feature generation paradigm falls short of providing interpretable insights. In addition, existing approaches rely on semantic prototypes annotated by human experts, which exhibit a significant limitation in their scalability to generalized scenes. To overcome these deficiencies, a natural solution is to generate images for unseen classes using text prompts. To this end, We present DIG-ZSL, a novel Discriminative Image Generation framework for Zero-Shot Learning. Specifically, to ensure the generation of discriminative images for training an effective ZSL classifier, we learn a discriminative class token (DCT) for each unseen class under the guidance of a pre-trained category discrimination model (CDM). Harnessing DCTs, we can generate diverse and high-quality images, which serve as informative unseen samples for ZSL tasks. In this paper, the extensive experiments and visualizations on four datasets show that our DIG-ZSL: (1) generates diverse and high-quality images, (2) outperforms previous state-of-the-art nonhuman-annotated semantic prototype-based methods by a large margin, and (3) achieves comparable or better performance than baselines that leverage human-annotated semantic prototypes. The codes will be made available upon acceptance of the paper.
format Preprint
id arxiv_https___arxiv_org_abs_2412_17219
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Discriminative Image Generation with Diffusion Models for Zero-Shot Learning
Fu, Dingjie
Hou, Wenjin
Chen, Shiming
Chen, Shuhuang
You, Xinge
Khan, Salman
Khan, Fahad Shahbaz
Computer Vision and Pattern Recognition
Generative Zero-Shot Learning (ZSL) methods synthesize class-related features based on predefined class semantic prototypes, showcasing superior performance. However, this feature generation paradigm falls short of providing interpretable insights. In addition, existing approaches rely on semantic prototypes annotated by human experts, which exhibit a significant limitation in their scalability to generalized scenes. To overcome these deficiencies, a natural solution is to generate images for unseen classes using text prompts. To this end, We present DIG-ZSL, a novel Discriminative Image Generation framework for Zero-Shot Learning. Specifically, to ensure the generation of discriminative images for training an effective ZSL classifier, we learn a discriminative class token (DCT) for each unseen class under the guidance of a pre-trained category discrimination model (CDM). Harnessing DCTs, we can generate diverse and high-quality images, which serve as informative unseen samples for ZSL tasks. In this paper, the extensive experiments and visualizations on four datasets show that our DIG-ZSL: (1) generates diverse and high-quality images, (2) outperforms previous state-of-the-art nonhuman-annotated semantic prototype-based methods by a large margin, and (3) achieves comparable or better performance than baselines that leverage human-annotated semantic prototypes. The codes will be made available upon acceptance of the paper.
title Discriminative Image Generation with Diffusion Models for Zero-Shot Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.17219