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

利用视觉Transformer基础模型中的层归一化微调进行分类

Exploiting Layer Normalization Fine-tuning in Visual Transformer Foundation Models for Classification

Shanghai Academy of Artificial Intelligence for Science(上海人工智能科学研究院);Xi’an Jiaotong-Liverpool University(西安交通大学利物浦大学);University of Liverpool(利物浦大学);AI 3 Fudan University(复旦大学);Duke Kunshan University(杜克-昆山大学);BII, A ∗ STAR;Zhejiang University(浙江大学)

Zhaorui Tan, Tan Pan, Kaizhu Huang, Weimiao Yu, Kai Yao, Chen Jiang, Qiufeng Wang, Anh Nguyen, Xin Guo, Yuan Cheng, Xi Yang

AI总结:

本文研究ViT中LayerNorm微调动态,提出FSR量化目标域代表性并设计标量λ重缩放机制与循环框架,有效解决数据稀缺与领域偏移下的微调问题。

摘要:

LayerNorm在Vision Transformers (ViTs)中至关重要,但其在数据稀缺和领域偏移下的微调动态仍未被充分探索。本文表明,微调后LayerNorm参数的偏移指示了源域和目标域之间的转换;其有效性取决于目标训练样本准确代表目标域的程度,这由我们提出的Fine-tuning Shift Ratio ($FSR$)量化。基于此,我们提出一种简单有效的重缩放机制,使用与$FSR$负相关的标量$λ$来对齐学习到的LayerNorm偏移与在完全代表性数据下实现的理想偏移,并结合循环框架进一步增强LayerNorm微调。在自然和病理图像上的广泛实验,涵盖分布内 (ID) 和分布外 (OOD) 设置及多种目标训练样本规模,验证了我们的框架。值得注意的是,与ID案例相比,OOD任务倾向于产生更低的$FSR$和更高的$λ$,特别是在数据稀缺时,表明目标训练样本代表性不足。此外,在病理数据上微调的ViTFs表现得更像ID设置,倾向于保守的LayerNorm更新。我们的发现阐明了迁移学习中LayerNorm未被充分探索的动态,并为LayerNorm微调提供了实用策略。

英文摘要:

LayerNorm is pivotal in Vision Transformers (ViTs), yet its fine-tuning dynamics under data scarcity and domain shifts remain underexplored. This paper shows that shifts in LayerNorm parameters after fine-tuning (LayerNorm shifts) are indicative of the transitions between source and target domains; its efficacy is contingent upon the degree to which the target training samples accurately represent the target domain, as quantified by our proposed Fine-tuning Shift Ratio ($FSR$). Building on this, we propose a simple yet effective rescaling mechanism using a scalar $λ$ that is negatively correlated to $FSR$ to align learned LayerNorm shifts with those ideal shifts achieved under fully representative data, combined with a cyclic framework that further enhances the LayerNorm fine-tuning. Extensive experiments across natural and pathological images, in both in-distribution (ID) and out-of-distribution (OOD) settings, and various target training sample regimes validate our framework. Notably, OOD tasks tend to yield lower $FSR$ and higher $λ$ in comparison to ID cases, especially with scarce data, indicating under-represented target training samples. Moreover, ViTFs fine-tuned on pathological data behave more like ID settings, favoring conservative LayerNorm updates. Our findings illuminate the underexplored dynamics of LayerNorm in transfer learning and provide practical strategies for LayerNorm fine-tuning.

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

↑