arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2510.22650cs.CV
收藏

自注意力分解用于训练自由扩散编辑

Self-Attention Decomposition For Training Free Diffusion Editing

ELLIS Institute Finland(芬兰ELLIS研究所);Department of Computer Science, Aalto University, Finland(芬兰奥卢大学计算机科学系)

Tharun Anand, Mohammad Hassan Vali, Arno Solin, Green Rosh, BH Pawan Prasad

AI总结:

本文提出了一种基于自注意力权重矩阵特征向量提取的扩散模型编辑方法,无需额外数据或微调,实现高效高质量的图像编辑。

摘要:

扩散模型在图像合成中实现了卓越的保真度,但对其输出进行精确控制以实现目标编辑仍具有挑战性。实现可控性的关键步骤是识别模型潜在表示中可解释的方向,这些方向对应于语义属性。现有方法寻找可解释方向通常依赖于采样大量图像或训练辅助网络,这限制了效率。我们提出了一种分析方法,直接从预训练的扩散模型参数中推导出语义编辑方向,无需额外数据或微调。我们的发现是自注意力权重矩阵编码了数据分布在训练过程中学习的丰富结构信息。通过计算这些权重矩阵的特征向量,我们获得了稳健且可解释的编辑方向。实验表明,我们的方法在多个数据集上产生了高质量的编辑,同时通过比当前基准减少60%的编辑时间。

英文摘要:

Diffusion models achieve remarkable fidelity in image synthesis, yet precise control over their outputs for targeted editing remains challenging. A key step toward controllability is to identify interpretable directions in the model's latent representations that correspond to semantic attributes. Existing approaches for finding interpretable directions typically rely on sampling large sets of images or training auxiliary networks, which limits efficiency. We propose an analytical method that derives semantic editing directions directly from the pretrained parameters of diffusion models, requiring neither additional data nor fine-tuning. Our insight is that self-attention weight matrices encode rich structural information about the data distribution learned during training. By computing the eigenvectors of these weight matrices, we obtain robust and interpretable editing directions. Experiments demonstrate that our method produces high-quality edits across multiple datasets while reducing editing time significantly by 60% over current benchmarks.

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

↑