Mar, 2024
探究投影头在表示学习中的益处
Investigating the Benefits of Projection Head for Representation Learning
Yihao Xue, Eric Gan, Jiayi Ni, Siddharth Joshi, Baharan Mirzasoleiman
TL;DR对于提取高质量表示的有效技术是在训练期间在编码器上方添加投影头部,然后丢弃它并使用预投影表示。本研究通过理论解释揭示了这种技术成功的原因,探索了低层特征表示规范化和非线性相互作用对模型性能的影响,并证明了这种机制如何提高有监督对比学习和有监督学习中的鲁棒性。