探究投影头对表征学习的益处
Investigating the Benefits of Projection Head for Representation Learning
基于摘要分析。本文研究为何在训练时给编码器添加投影头、随后丢弃投影头并使用投影前表征,能够获得更好的表征。核心贡献是从训练算法的隐式偏置解释不同层的特征分配,并讨论这种机制如何受到数据增强、输入特征及非线性的影响。 作者首先分析采用自监督对比损失训练的线性模型,指出训练算法的隐式偏置会产生逐层递进的特征加权:越深层的特征权重越不均衡,因此较低层的表征相对更均衡、专门化程度更低。作者给出这类表征更有利的理论情形,强调数据增强与输入特征之间的相互作用。这里不应将“投影前表征没有被损失直接优化”理解为编码器不接受训练,而应关注损失作用位置与最终使用表征位置的区别。 作者进一步报告,引入非线性后,较低层可以学到在较高层完全不再出现的特征,并讨论这一机制如何改善监督对比学习与监督学习的鲁棒性。作者报告在 CIFAR-10/100、UrbanCars 及 ImageNet 的分布偏移版本上进行了实证验证,还提出一种更易解释和控制的潜在投影头替代方案;摘要未给出其具体形式、指标数值及完整对照。理论假设、实验协议、消融及统计信息尚未验证。 平台推测(待验证):该机制可能对应 EEG 自监督学习中增强操作改变任务相关特征权重的问题,但依赖于增强是否保留神经信号语义,以及各层是否确实呈现类似的特征分化。可复用的是“比较投影前后及不同深度表征”的分析思路,编码器和增强策略则需适配 EEG 的时序与通道结构;低信噪比、被试差异和小数据可能使较低层保留更多伪迹,而非更可迁移的任务信息。一个可证伪的小实验是在固定 Cross-subject 划分、训练预算和下游分类器的条件下,比较各层冻结表征及有无投影头的表现,并检验结论是否随增强策略变化。已有 EEG 相关工作尚未检索确认。
该研究从训练隐式偏置与逐层特征加权解释投影头的作用,为选择编码器表征及审视数据增强带来的特征偏向提供了可核对的理论视角,但其对 EEG 的适用性尚未验证。
来源:OpenReview · Representation learning · openreview.net