ConceptGuard: Continual Personalized Text-to-Image Generation with Forgetting and Confusion Mitigation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Guo, Zirun, Jin, Tao
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909536707674112
author Guo, Zirun
Jin, Tao
author_facet Guo, Zirun
Jin, Tao
contents Diffusion customization methods have achieved impressive results with only a minimal number of user-provided images. However, existing approaches customize concepts collectively, whereas real-world applications often require sequential concept integration. This sequential nature can lead to catastrophic forgetting, where previously learned concepts are lost. In this paper, we investigate concept forgetting and concept confusion in the continual customization. To tackle these challenges, we present ConceptGuard, a comprehensive approach that combines shift embedding, concept-binding prompts and memory preservation regularization, supplemented by a priority queue which can adaptively update the importance and occurrence order of different concepts. These strategies can dynamically update, unbind and learn the relationship of the previous concepts, thus alleviating concept forgetting and confusion. Through comprehensive experiments, we show that our approach outperforms all the baseline methods consistently and significantly in both quantitative and qualitative analyses.
format Preprint
id arxiv_https___arxiv_org_abs_2503_10358
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle ConceptGuard: Continual Personalized Text-to-Image Generation with Forgetting and Confusion Mitigation
Guo, Zirun
Jin, Tao
Computer Vision and Pattern Recognition
Machine Learning
Diffusion customization methods have achieved impressive results with only a minimal number of user-provided images. However, existing approaches customize concepts collectively, whereas real-world applications often require sequential concept integration. This sequential nature can lead to catastrophic forgetting, where previously learned concepts are lost. In this paper, we investigate concept forgetting and concept confusion in the continual customization. To tackle these challenges, we present ConceptGuard, a comprehensive approach that combines shift embedding, concept-binding prompts and memory preservation regularization, supplemented by a priority queue which can adaptively update the importance and occurrence order of different concepts. These strategies can dynamically update, unbind and learn the relationship of the previous concepts, thus alleviating concept forgetting and confusion. Through comprehensive experiments, we show that our approach outperforms all the baseline methods consistently and significantly in both quantitative and qualitative analyses.
title ConceptGuard: Continual Personalized Text-to-Image Generation with Forgetting and Confusion Mitigation
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2503.10358