DEFT让文生图模型高效微调,兼顾个性定制与通用生成能力。
DEFT: Decompositional Efficient Fine-Tuning for Text-to-Image Models
- 将参数更新分解为子空间投影与低秩调整,仅训练两个小矩阵。
- 在多个数据集上达到顶尖性能,个人化与多任务生成均表现优异。
- 适合追求低成本微调且需保持编辑能力的研究者与开发者。
文生图模型的高效微调需在计算资源和可训练参数有限的情况下,实现对目标分布的良好对齐。现有方法难以同时完成个性化概念学习(如用少量图像定制风格)与保持多任务指令理解能力,且影响编辑灵活性。本文提出DEFT——一种基于分解的高效微调框架,通过将预训练权重更新分解为两部分:(1) 投影至由一个低秩矩阵张成的低秩子空间的补空间;(2) 在该子空间内进行低秩更新。其中,一个可训练低秩矩阵定义子空间,另一个则实现灵活参数调整。我们在Dreambooth、Dreambench Plus(个性化)、InsDet(物体与场景适配)、VisualCloze(视觉上下文生成)等数据集上进行了广泛实验,涵盖Stable Diffusion与统一模型。结果表明,DEFT在各项任务中均达当前最优水平,展现出高效微调的涌现特性。代码已开源。
原文摘要 · Abstract (English)
Efficient fine-tuning of pre-trained Text-to-Image (T2I) models involves adjusting the model to suit a particular task or dataset while minimizing computational resources and limiting the number of trainable parameters. However, it often faces challenges in striking a trade-off between aligning with the target distribution: learning a novel concept from a limited image for personalization and retaining the instruction ability needed for unifying multiple tasks, all while maintaining editability (aligning with a variety of prompts or in-context generation). In this work, we introduce DEFT, Decompositional Efficient Fine-Tuning, an efficient fine-tuning framework that adapts a pre-trained weight matrix by decomposing its update into two components with two trainable matrices: (1) a projection onto the complement of a low-rank subspace spanned by a low-rank matrix, and (2) a low-rank update. The single trainable low-rank matrix defines the subspace, while the other trainable low-rank matrix enables flexible parameter adaptation within that subspace. We conducted extensive experiments on the Dreambooth and Dreambench Plus datasets for personalization, the InsDet dataset for object and scene adaptation, and the VisualCloze dataset for a universal image generation framework through visual in-context learning with both Stable Diffusion and a unified model. Our results demonstrated state-of-the-art performance, highlighting the emergent properties of efficient fine-tuning. Our code is available on \href{https://github.com/MAXNORM8650/DEFT}{DEFTBase}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。