通过谱分解对齐残差流,提升视觉语言模型的零样本分类性能。
ResiDual Transformer Alignment with Spectral Decomposition
- 基于残差流的谱几何分析,发现视觉头具低维结构和任务专属性。
- 针对性对齐文本与专用头,显著提升零样本分类准确率。
- 提出ResiDual方法,高效可解释,适用于多种模型与数据集。
从残差流视角观察,变压器网络中存在一种奇特现象:残差贡献(如注意力头)有时会专门处理特定任务或输入属性。本文研究视觉变压器中的这一现象,聚焦残差的谱几何特性,探索其在视觉-语言模型跨模态对齐中的意义。首先,我们将其与视觉头表示的内在低维结构联系起来,深入分析其主成分,发现其在多种输入数据分布下均编码特定功能。随后,分析多模态模型中头专属性的影响,发现文本与专用头对齐程度提升可显著改善零样本分类性能。该关联在不同预训练数据、网络规模和目标下保持一致,揭示了一种通过定向对齐增强零样本分类的新机制。最终,我们将这些洞察转化为可操作技术——ResiDual,一种残差流的谱对齐方法。如同淘金,它让无关主成分(即属性)的噪声被冲刷掉,放大任务相关成分。令人惊喜的是,这种双重视角的模态对齐策略,在70个预训练模型-数据集组合(7个模型,10个数据集)上实现了微调级性能,同时具备高度可解释性和参数高效性。
原文摘要 · Abstract (English)
When examined through the lens of their residual streams, a puzzling property emerges in transformer networks: residual contributions (e.g., attention heads) sometimes specialize in specific tasks or input attributes. In this paper, we analyze this phenomenon in vision transformers, focusing on the spectral geometry of residuals, and explore its implications for modality alignment in vision-language models. First, we link it to the intrinsically low-dimensional structure of visual head representations, zooming into their principal components and showing that they encode specialized roles across a wide variety of input data distributions. Then, we analyze the effect of head specialization in multimodal models, focusing on how improved alignment between text and specialized heads impacts zero-shot classification performance. This specialization-performance link consistently holds across diverse pre-training data, network sizes, and objectives, demonstrating a powerful new mechanism for boosting zero-shot classification through targeted alignment. Ultimately, we translate these insights into actionable terms by introducing ResiDual, a technique for spectral alignment of the residual stream. Much like panning for gold, it lets the noise from irrelevant unit principal components (i.e., attributes) wash away to amplify task-relevant ones. Remarkably, this dual perspective on modality alignment yields fine-tuning level performance on different data distributions while modelling an extremely interpretable and parameter-efficient transformation, as we extensively show on 70 pre-trained network-dataset combinations (7 models, 10 datasets).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。