跳到正文
OpenReview · Representation learning· Yuwei Sun; Hideya Ochiai; Zhirong Wu; Stephen Lin; Ryota Kanai·· 2023-10-27精选AI 评分73

Associative Transformer 是一种稀疏表征学习器

Associative Transformer Is A Sparse Representation Learner

AI 导读

基于摘要分析。本文研究通用 Transformer 的稀疏表征学习,提出 Associative Transformers(AiT),通过低秩显式记忆引导共享工作空间中的瓶颈注意力,并将该记忆作为 Hopfield 网络联想记忆中的吸引子;主要验证对象是视觉任务,而非 EEG 或 BCI。 机制上,Set Transformer 和 Perceiver 通过潜在空间汇聚交叉注意力,形成容量有限的注意力瓶颈。AiT 在此基础上借鉴 Global Workspace Theory 与联想记忆,使显式记忆既充当注意力先验,也承担吸引子作用。作者报告,AiT 能经由瓶颈学习不同先验,并声称其复杂度不随输入数量及维度变化;摘要未给出复杂度定义与推导,不能据此认定完整模型的总计算量与输入规模无关。 作者报告,AiT 在多种视觉任务上优于 Set Transformer、Vision Transformer 和 Coordination,但摘要未列出数据集、指标、数值及评估协议,因此尚不能判断收益幅度、适用任务边界与比较条件。损失函数、训练配置、记忆容量、实验协议、消融及统计信息尚未验证。 平台推测(待验证):若低秩记忆能提取可复用的信号模式,该机制可能用于缓解 EEG 时空 token 交互的计算负担;这是假设,并非本文已证实的 BCI 效果。可复用部分是记忆引导的注意力瓶颈,需改造 EEG 的通道与时间编码、输入表征及训练目标;原研究对监督方式和数据规模的依赖尚不明确。低信噪比、跨被试差异、通道变化和小样本可能使记忆学习到伪迹或被试特征,也可能让瓶颈丢失短暂判别信息。一个可检验的小实验是在固定 EEG 数据划分与训练预算下,对比有无显式联想记忆的瓶颈注意力,分别检查被试内与跨被试表现及计算开销。已有 EEG 相关工作尚未检索确认。

阅读价值

值得关注的是以低秩显式记忆同时连接瓶颈注意力与 Hopfield 联想记忆的机制;其视觉任务优势属于作者报告,能否改善 EEG 表征及计算开销尚未验证。

来源:OpenReview · Representation learning · openreview.net