跳到正文
OpenReview · Representation learning·· 25 天前精选AI 评分79

通过带周期性重置的归一化梯度动力学理解神经网络中的表征对齐

Understanding Representation Alignment in Neural Networks through Normalized Gradient Dynamics with Periodic Resets

AI 导读

基于摘要分析。本文研究神经网络在训练过程中如何形成有用表征,提出尺度不变的 feature-target alignment,用于衡量学习到的特征映射对目标函数的刻画程度,并据此提出带周期性重置的归一化梯度下降。 核心机制是:作者指出,归一化特征学习在训练初期会产生改善特征与目标对齐的方向;周期性重置旨在反复使训练回到有利于目标对齐表征学习的状态。摘要未说明归一化的具体定义、重置对象与周期选择,因此不能将其直接理解为参数重初始化或学习率重启。 理论方面,作者报告,对于一类广泛的参数化特征映射,该方法在总体与有限样本两种设置下,均可在任意给定的有限时间范围内单调改善 feature-target alignment。作者还分析了一个 full-matrix representation model,报告同一学习原则可严格降低模型设定失配风险(misspecified risk),为表征改善能够带来预测收益提供具体例子。上述结论不宜外推为任意神经网络或任意训练配置下的保证;定理假设及其适用边界尚未验证。 作者报告提供了支持理论结果的实证证据,但摘要未给出任务、数据集、模型规模、指标或对照结果,实验协议、消融及统计信息尚未验证。复现前需核对对齐量的计算方式、目标函数的可获得性、归一化与重置实现,以及与普通梯度下降等对照的训练预算。 平台推测(待验证):若有限样本对齐量能用 EEG 任务中的训练标签可靠估计,该原则可作为监督表征学习的优化假设,而非已验证的 BCI 方法。可复用部分是归一化更新与周期性重置思想,需改造的是 EEG 编码器、目标表示和对齐量估计;低信噪比、小数据及跨被试分布差异可能使训练样本上的对齐改善无法转化为泛化收益。一个可检验的小实验是在固定 EEG 编码器与 Cross-subject 划分下,以相同训练预算比较普通梯度下降和该方法,同时观察训练对齐量与留出被试性能是否一致改善。已有 EEG 相关工作尚未检索确认。

阅读价值

值得阅读其如何以尺度不变的 feature-target alignment 连接表征学习动态与预测风险,但理论成立条件、周期性重置的具体操作及实证适用范围仍需全文核对。

来源:OpenReview · Representation learning · openreview.net