中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Enhancing Fine-Tuning Performance of Text-to-Image Diffusion Models for Few-Shot Image Generation Through Contrastive Learning

文献类型:会议论文

作者Zhu, Yanlin; Yang,Peipei
出版日期2024-05
会议日期2024-8
会议地点北京
英文摘要

Recent significant progress in the field of few-shot image generation has been achieved by fine-tuning pretrained text-to-image models, notably methods such as Dreambooth and Textual Inversion. To enhance the performance of existing methods, we advocate the explicit utilization of differences between various concepts to enable the generator to more effectively capture the characteristics of any given concept with limited samples. To this end, we introduce a contrastive generation approach that leverages these differences. Specifically, we expand the framework of Dreambooth by applying multimodal contrastive learning to optimize the feature distribution of conditions. Furthermore, we introduce another contrastive mechanism on the fused multimodal features between different concepts to further capture the characteristics of each new concept.
Ultimately, our approach outperforms advanced few-shot generation models in capturing the characteristics of new concepts accurately. Additionally, it supports various applications and facilitates simultaneous learning of multiple new concepts without retraining for each.

源URL[http://ir.ia.ac.cn/handle/173211/57069]  
专题多模态人工智能系统全国重点实验室
通讯作者Yang,Peipei
作者单位1.中国科学院大学
2.中国科学院大学自动化研究所多模态国家重点实验室
推荐引用方式
GB/T 7714
Zhu, Yanlin,Yang,Peipei. Enhancing Fine-Tuning Performance of Text-to-Image Diffusion Models for Few-Shot Image Generation Through Contrastive Learning[C]. 见:. 北京. 2024-8.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。