跳到正文
OpenReview · Representation learning·· 2026-05-23精选AI 评分77

Fibottention:通过跨头多样化注意力实现 Inception 式视觉表征学习

Fibottention: Inceptive Visual Representation Learning with Diverse Attention Across Heads

AI 导读

基于摘要分析。该研究面向视觉 Transformer 中多头自注意力(MHSA)的二次计算复杂度及对大规模训练数据的依赖,提出结构化稀疏自注意力机制 Fibottention,以减少成对交互并提升不同注意力头的功能多样性。原研究对象是视觉表征学习,实验涉及图像分类、视频理解和机器人学习,并非 EEG 或 BCI。 机制上,Fibottention 从 Wythoff array 构造稀疏模式,使不同注意力头采用不同的交互覆盖。作者声称,这种设计可在保证充分且多样覆盖的同时减少冗余成对交互,并将自注意力计算复杂度降至 O(N log N),其中 N 为 token 数;跨头差异形成类似 Inception 的功能多样性。具体模式构造、覆盖性质的证明条件及实现细节尚未验证。 作者报告,将 Fibottention 集成到标准 Transformer 后,在所述多个领域中,相比稠密 MHSA 可取得更优或相当的表现;在摘要所称的典型设置下,仅使用跨注意力头全部成对交互的 2%。作者还报告,在等 FLOPs 比较下优于已有稀疏注意力机制。摘要未给出具体数据集、划分、性能指标及硬件实测,不能将交互比例直接等同于端到端加速或显存降幅;减少大规模训练数据依赖的效果亦需核对。实验协议、消融及统计信息尚未验证。 平台推测(待验证):若 EEG 被编码为较长的时间或时空 token 序列,跨头差异化稀疏覆盖可能缓解注意力开销,但原摘要未说明其对输入空间结构、监督方式和训练规模的具体依赖。可考虑复用稀疏注意力模块,同时改造 token 编码及时间—通道位置表示;低信噪比、通道变化和小样本可能使固定稀疏连接遗漏关键依赖,跨被试稳健性不能由视觉结果推定。一个可检验的小实验是在固定 EEG 编码器、训练预算和跨被试划分下,仅替换注意力模块,与稠密 MHSA 及等 FLOPs 稀疏基线比较任务指标、实际时延和显存占用。相关 EEG 工作尚未检索确认。

阅读价值

值得核对其基于 Wythoff array 的跨头稀疏模式能否在减少注意力交互的同时保持特征多样性,尤其是与稠密 MHSA 及等 FLOPs 稀疏注意力的比较协议;其 EEG 适用性尚未验证。

来源:OpenReview · Representation learning · openreview.net