arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2311.18763cs.CVcs.AIcs.LG

使用STAMINA的持续扩散:堆叠与掩码增量适配器

Continual Diffusion with STAMINA: STack-And-Mask INcremental Adapters

Samsung Research America(三星美国研究院);Georgia Institute of Technology(佐治亚理工学院)

James Seale Smith, Yen-Chang Hsu, Zsolt Kira, Yilin Shen, Hongxia Jin

AI总结:

本文提出STAMINA方法,通过低秩注意力掩码适配器和定制MLP令牌增强LoRA的顺序概念学习能力,在50概念持续扩散定制和图像分类持续学习基准上均取得SOTA性能,且无额外推理参数开销。

摘要:

近期工作展示了一种显著的能力,即在顺序(即持续)方式下,仅需为每个概念提供少量示例图像,即可将文本到图像扩散模型定制到多个细粒度概念。这一设定被称为持续扩散。在此,我们提出一个问题:能否将这些方法扩展到更长的概念序列而不发生遗忘?尽管先前的工作缓解了对已学概念的遗忘,但我们表明其学习新任务的能力在较长序列上会达到饱和。我们通过引入一种新颖的方法——堆叠与掩码增量适配器(STAMINA)来应对这一挑战,该方法由低秩注意力掩码适配器和定制化的MLP令牌组成。STAMINA旨在通过以低秩MLP参数化的可学习硬注意力掩码,增强LoRA在顺序概念学习中的鲁棒微调特性,从而通过稀疏适配实现精确、可扩展的学习。值得注意的是,所有引入的可训练参数在训练后均可折叠回模型中,不会产生额外的推理参数开销。我们表明,在一个由地标和人脸组成的50概念基准上,STAMINA在文本到图像持续定制设定中优于先前的SOTA,且无需存储任何回放数据。此外,我们将方法扩展到图像分类的持续学习设定,证明我们的增益同样转化为该标准基准上的最先进性能。

英文摘要:

Recent work has demonstrated a remarkable ability to customize text-to-image diffusion models to multiple, fine-grained concepts in a sequential (i.e., continual) manner while only providing a few example images for each concept. This setting is known as continual diffusion. Here, we ask the question: Can we scale these methods to longer concept sequences without forgetting? Although prior work mitigates the forgetting of previously learned concepts, we show that its capacity to learn new tasks reaches saturation over longer sequences. We address this challenge by introducing a novel method, STack-And-Mask INcremental Adapters (STAMINA), which is composed of low-ranked attention-masked adapters and customized MLP tokens. STAMINA is designed to enhance the robust fine-tuning properties of LoRA for sequential concept learning via learnable hard-attention masks parameterized with low rank MLPs, enabling precise, scalable learning via sparse adaptation. Notably, all introduced trainable parameters can be folded back into the model after training, inducing no additional inference parameter costs. We show that STAMINA outperforms the prior SOTA for the setting of text-to-image continual customization on a 50-concept benchmark composed of landmarks and human faces, with no stored replay data. Additionally, we extended our method to the setting of continual learning for image classification, demonstrating that our gains also translate to state-of-the-art performance in this standard benchmark.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑