arXiv:2410.00700cs.CVcs.AI2024-10ICLR被引 3

用扩散模型的分类器得分实现无存储负担的持续个性化

Mining Your Own Secrets: Diffusion Classifier Scores for Continual Personalization of Text-to-Image Diffusion Models

  • 利用扩散分类器得分对参数和函数空间进行正则化
  • 在多个数据集上优于SOTA方法C-LoRA,保留旧概念能力更强
  • 无需存储历史数据或额外参数,适合隐私敏感场景

个性化文本到图像扩散模型能高效从用户提供的文本描述和少量图像中学习新概念。但在实际应用中,用户可能需逐个个性化多个概念,且因存储或隐私限制无法访问历史数据。面对这种持续学习(CL)场景,多数个性化方法难以兼顾新概念学习与旧概念保留,这正是持续个性化(CP)的目标。受成功依赖类别特异性信息的持续学习方法启发,我们利用扩散分类器(DC)得分——即内在的类别条件密度估计——来正则化文本到图像扩散模型的参数空间和函数空间,实现持续个性化。通过多种评估设置、数据集和指标验证,我们的基于正则化的CP方法优于当前最佳的C-LoRA及其他基线。此外,在无需回放数据的持续学习设置下,结合低秩适配器,本方法在存储和参数方面均无额外开销,相较现有方案更高效。

原文摘要 · Abstract (English)

Personalized text-to-image diffusion models have grown popular for their ability to efficiently acquire a new concept from user-defined text descriptions and a few images. However, in the real world, a user may wish to personalize a model on multiple concepts but one at a time, with no access to the data from previous concepts due to storage/privacy concerns. When faced with this continual learning (CL) setup, most personalization methods fail to find a balance between acquiring new concepts and retaining previous ones -- a challenge that continual personalization (CP) aims to solve. Inspired by the successful CL methods that rely on class-specific information for regularization, we resort to the inherent class-conditioned density estimates, also known as diffusion classifier (DC) scores, for continual personalization of text-to-image diffusion models. Namely, we propose using DC scores for regularizing the parameter-space and function-space of text-to-image diffusion models, to achieve continual personalization. Using several diverse evaluation setups, datasets, and metrics, we show that our proposed regularization-based CP methods outperform the state-of-the-art C-LoRA, and other baselines. Finally, by operating in the replay-free CL setup and on low-rank adapters, our method incurs zero storage and parameter overhead, respectively, over the state-of-the-art. Our project page: https://srvcodes.github.io/continual_personalization/

文本生成扩散模型持续学习个性化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。