通过稀疏率缩减构建白盒 Transformer
White-Box Transformers via Sparse Rate Reduction
基于摘要分析。本文研究表征学习的目标及 Transformer 的数学解释,提出以 sparse rate reduction(稀疏率缩减)统一衡量表示质量,并据此构建白盒 Transformer 类架构。主要研究对象是通用表征学习及视觉数据,而非 EEG 或 BCI。 核心机制是将 token 集合的分布压缩、变换为支撑在非相干子空间上的低维高斯混合分布。作者将标准 Transformer 块解释为对目标函数互补部分的交替优化:多头自注意力对应通过降低有损编码率压缩 token 集合的梯度下降步骤,后续多层感知机则被解释为尝试使 token 表示稀疏化。其贡献重点不是单独增加一个模块,而是从显式优化目标推导架构,并赋予各步骤数学解释;具体目标公式、约束及推导适用条件仍需正文核对。 作者报告,在包括 ImageNet 在内的大规模真实视觉数据上,这类网络能够压缩并稀疏化表示,性能接近经过充分工程优化的 ViT。摘要未提供具体指标、分数、训练配置或对照协议,不能据此判断性能差距与计算代价。材料提供了代码仓库,但实现细节、实验协议、消融及统计信息尚未验证。 平台推测(待验证):若 EEG 的时空 token 表示能够近似满足低维子空间结构,显式压缩与稀疏化目标可能为冗余、低信噪比信号的表征学习提供可解释约束。可复用的是目标驱动的迭代模块,需改造的是 EEG token 化、通道位置编码及跨被试归一化;原研究所需监督形式和训练规模尚未验证,视觉结果不代表小样本 EEG 同样有效。噪声、通道差异及跨被试分布变化可能破坏子空间假设,压缩也可能丢失弱判别信息。可检验的小实验是在固定 Cross-subject 划分、相同预处理和训练预算下,对比该架构与常规 Transformer,同时观察任务指标及压缩、稀疏化是否与性能改善一致。已有 EEG 相关工作尚未检索确认。
值得阅读的是其将注意力与表示稀疏化统一为 sparse rate reduction 的迭代优化解释;这一机制为可解释表征学习提供了可核对的设计路径,但其在 EEG 上的适用性尚未验证。
来源:OpenReview · Representation learning · openreview.net