跳到正文
OpenReview · Representation learning· Yihao Xue; Eric Gan; Jiayi Ni; Siddharth Joshi; Baharan Mirzasoleiman·· 2024-01-01精选AI 评分81

探究投影头对表征学习的益处

Investigating the Benefits of Projection Head for Representation Learning

AI 导读

基于摘要分析。本文研究为何在编码器顶部添加投影头、训练后丢弃投影头并使用投影前表示,能够改善表征质量。作者从采用自监督对比损失训练的线性模型出发,给出理论解释,并进一步考察非线性网络、监督对比学习和监督学习中的相关机制。 核心机制是训练算法的隐式偏置导致逐层渐进的特征加权:随着层次加深,不同特征的权重越来越不均衡,较低层因而保留更均衡、较少专门化的表示。作者从理论上刻画了这类表示更有益的情形,强调数据增强与输入特征之间的相互作用。作者还报告,引入非线性后,较低层可以学习到高层完全不包含的特征,并展示该机制如何改善监督对比学习和监督学习的鲁棒性;具体适用条件仍需结合全文核对。 作者报告在 CIFAR-10/100、UrbanCars 及发生分布偏移的 ImageNet 版本上进行了实证验证,并提出一种可能替代投影头、设计更可解释且可控的方案。摘要未提供该替代方案的具体形式、实验划分、对照配置或定量结果,实验协议、消融及统计信息尚未验证。 平台推测(待验证):若 EEG 自监督对比学习中的数据增强对不同任务相关特征产生不均衡保留,投影头可能影响编码器表征的特征覆盖程度。此假设依赖增强策略、对比监督信号与 EEG 特征之间的对应关系,不代表视觉实验结论已适用于 BCI。可复用的是编码器—投影头训练与分层表征比较思路;需改造的是时序编码器和符合信号语义的增强。低信噪比、通道差异及小数据可能使较低层保留的特征主要是噪声或被试特征。一个小规模检验是在固定跨被试划分和相同训练预算下,对比有无投影头时的编码器表示,并分别线性评估投影前后表示,检查优势是否随增强强度变化。相关 EEG 工作尚未检索确认。

阅读价值

值得阅读其对投影头作用的机制解释:训练中的逐层特征加权可能使投影前表示保留更均衡、较少专门化的特征,但其对 EEG 表征学习的适用性尚未验证。

来源:OpenReview · Representation learning · openreview.net