利用权重空间对称性近似曲率
Exploiting weight-space symmetries for approximating curvature
基于摘要分析。本文研究现代深度网络中损失函数曲率难以高效近似的问题,提出利用保持损失不变的权重空间对称性,构造可高效估计、存储和求逆的结构化 Hessian 近似。主要贡献是将对称性施加的曲率约束转化为近似方法,并以统一理论视角联系已有曲率估计方法。 核心机制是对保持损失不变的群作用进行解析平均,从单个梯度构造曲率近似。用户指定的对称群决定近似精度与计算成本之间的权衡。作者指出,特定对称群选择可恢复 Shampoo/Muon 类曲率估计;这并不意味着其方法与这些优化器在全部训练设置下等价。摘要未说明具体群、近似公式、数值稳定化或更新规则,相关细节尚未验证。 作者报告在多种网络架构上验证该方法,并将其用于包含小型语言模型在内的二阶优化基准,但摘要未给出数据集、对照基线、评估指标或性能数值。实验协议、消融及统计信息尚未验证。作者还提出不确定性估计、持续学习、压缩与剪枝、训练数据归因等潜在应用,这些不能据摘要视为已验证结果。 平台推测(待验证):假设 EEG 解码网络存在可利用的损失不变对称性,该机制可能复用于训练阶段的曲率估计与预条件更新,而非直接改进神经信号表征。迁移需先核对网络结构、正则化和训练目标是否满足所选对称群的条件,再适配曲率估计与求逆模块。EEG 的低信噪比、小数据及跨被试分布差异可能使梯度估计不稳定,近似误差和计算开销也可能抵消收益。可在固定的被试内数据划分、相同网络及训练预算下,对比原优化器与该近似更新,记录 Accuracy、收敛速度、显存和运行时间,以检验是否存在实际收益。已有 EEG 相关工作尚未检索确认。
值得阅读其如何利用权重空间对称性从单个梯度构造可求逆的结构化 Hessian 近似,以及与 Shampoo/Muon 类曲率估计的理论联系;具体性能和 EEG 适用性尚未验证。
来源:OpenReview · State space models · openreview.net