跳到正文
OpenReview · Representation learning· Jaebyeong Jeon; Hyeonseo Jang; Jy-yong Sohn; Kibok Lee·· 2025-09-19精选AI 评分73

面向等变表征学习的软任务感知专家路由

Soft Task-Aware Routing of Experts for Equivariant Representation Learning

AI 导读

基于摘要分析。本文研究不变与等变表征联合学习中的特征冗余问题,提出 Soft Task-Aware Routing(STAR),将投影头建模为专家,通过路由促使专家分别捕获共享信息或任务特定信息,以提高模型容量的利用效率。 原问题中,等变表征学习关注输入变换在表征空间中引起的变化,不变表征学习则忽略这些变换以编码语义信息。常见方案使用独立投影头联合学习两类表征;作者认为,这种设计未充分利用两类学习目标之间的共享信息。STAR 的核心区别是引入任务感知的软路由,而非仅将投影头相互分离。具体路由计算、专家配置、损失函数及训练流程尚未验证。 作者报告,不变与等变嵌入之间的典型相关性降低,并将其作为特征冗余减少的证据;作者还报告在多种迁移学习任务上获得一致改善。摘要未提供任务名称、数据集、划分、对照基线或量化指标,实验协议、消融及统计信息尚未验证。典型相关性变化与下游收益之间的关系仍需结合正文核对。摘要提供了代码地址,但实现内容与复现条件尚未核验。 平台推测(待验证):该机制可能用于协调 EEG 中需要保持稳定的任务语义与需要保留的变换信息,但前提是能够明确界定哪些变换应保持标签、哪些变化具有可监督的等变关系。可复用部分是投影头专家与任务感知路由,需改造部分包括 EEG 变换设计及等变目标。低信噪比、跨被试差异、通道不一致和小数据可能使路由学习到伪相关,或使等变目标保留无关噪声。一个可证伪的小实验是:固定同一 EEG 编码器与数据划分,在已核验标签保持性的变换下,对比独立投影头与 STAR,同时检验下游任务指标和嵌入典型相关性;相关性降低但性能不升,不能视为迁移有效。已有 EEG 相关工作尚未检索确认。

阅读价值

值得核对 STAR 如何通过投影头专家路由协调不变与等变学习,以及典型相关性降低是否与迁移性能改善相对应;其 EEG/BCI 适用性尚未验证。

来源:OpenReview · Representation learning · openreview.net