用稀疏编码思想让大模型微调更可解释,提升图像编辑和概念定制效果
Sparse Fine-Tuning of Transformers for Generative Tasks
- 将微调视为稀疏组合基础特征原子,实现可解释的参数更新
- 在图像编辑中通过剔除无关原子,提升文本对齐性能
- 在文生图概念定制任务中,以更少参数实现优于基线的方法
大规模预训练变换器在多个领域革新了人工智能,微调仍是适应下游任务的主流方法,因从零训练成本过高。然而,现有微调方法生成的表示为修改参数的密集组合,难以解释其贡献及模型如何适应新任务。本文提出一种受稀疏编码启发的微调框架,将微调特征表示为基本元素(特征字典原子)的稀疏组合。特征字典原子作为表征的基础构建块,调整原子可实现对下游任务的无缝适应。稀疏系数则指示原子重要性,揭示各原子对更新表征的贡献。利用稀疏系数的原子选择能力,我们首先证明该方法通过移除不重要的特征字典原子,提升了图像编辑中的文本对齐性能。此外,在文本到图像的概念定制任务中,该方法能高效地使用特征字典原子的稀疏组合构建目标概念,表现优于多种基线微调方法。
原文摘要 · Abstract (English)
Large pre-trained transformers have revolutionized artificial intelligence across various domains, and fine-tuning remains the dominant approach for adapting these models to downstream tasks due to the cost of training from scratch. However, in existing fine-tuning methods, the updated representations are formed as a dense combination of modified parameters, making it challenging to interpret their contributions and understand how the model adapts to new tasks. In this work, we introduce a fine-tuning framework inspired by sparse coding, where fine-tuned features are represented as a sparse combination of basic elements, i.e., feature dictionary atoms. The feature dictionary atoms function as fundamental building blocks of the representation, and tuning atoms allows for seamless adaptation to downstream tasks. Sparse coefficients then serve as indicators of atom importance, identifying the contribution of each atom to the updated representation. Leveraging the atom selection capability of sparse coefficients, we first demonstrate that our method enhances image editing performance by improving text alignment through the removal of unimportant feature dictionary atoms. Additionally, we validate the effectiveness of our approach in the text-to-image concept customization task, where our method efficiently constructs the target concept using a sparse combination of feature dictionary atoms, outperforming various baseline fine-tuning methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。