IBNorm:受信息瓶颈启发的表征学习归一化方法
IBNorm: Information-Bottleneck Inspired Normalization for Representation Learning
基于摘要分析。本文研究通用深度学习中的归一化问题,提出受 Information Bottleneck(信息瓶颈)原则启发的 IBNorm,试图在保持训练稳定性与兼容性的同时,使表征保留预测信息并抑制无关变异。主要研究对象是语言与视觉模型,而非 EEG 或 BCI。 机制上,IBNorm 引入有界压缩操作,将归一化的目标从控制表征统计量扩展到信息保留与压缩。摘要未给出操作公式、信息瓶颈目标的具体定义、是否额外使用标签或互信息估计,以及训练与推理阶段的实现差异,因而尚不能判断其是否可直接替换现有归一化层。摘要将 BatchNorm、LayerNorm 和 RMSNorm 统称为方差导向方法,但三者的具体统计操作不同,不宜据此理解为它们都执行相同的零均值、单位方差变换。 作者报告,理论分析表明 IBNorm 相比方差导向方法具有更高的 IB 目标值及更紧的泛化界;相关目标定义、比较条件与证明假设尚未验证。作者报告,在 LLaMA、GPT-2 语言模型及 ResNet、ViT 视觉模型上的实验优于 BatchNorm、LayerNorm 和 RMSNorm,并以互信息分析支持其信息瓶颈行为。摘要未提供数据集、模型规模、评估指标、具体增益或训练预算,实验协议、消融及统计信息尚未验证。作者表示将公开代码,材料不能确认代码已发布。 平台推测(待验证):假设有界压缩能够削弱与任务无关的幅值波动,同时保留微弱判别信息,该机制可能适用于 EEG 的噪声与跨被试变异问题。可考虑复用压缩算子,但需核对其作用维度、通道尺度处理及监督需求;低信噪比、小数据条件下,压缩也可能抹除有效信号。一个可证伪的小实验是在固定 EEG 解码器与跨被试划分下,仅替换归一化模块,与原模块比较同一任务指标及多随机种子波动,且不利用测试被试标签调参。该迁移尚无本文证据支持,已有 EEG 相关工作尚未检索确认。
值得核对其有界压缩操作及信息瓶颈理论成立的条件,并复现相对标准归一化的收益;摘要尚不足以确认其对 EEG 表征学习的适用性。
来源:OpenReview · Representation learning · openreview.net