用于鲁棒表征学习的上下文视觉 Transformer
Contextual Vision Transformers for Robust Representation Learning
基于摘要分析。研究针对不同群组间潜在因素变化导致的图像分布偏移,提出 Contextual Vision Transformers(ContextViT),通过群组上下文条件化图像表征,并在推理时适应新的测试分布。 核心机制借鉴 in-context learning:将与输入图像属于同一群组的图像信息映射为额外的 context token,再将其附加到输入图像 tokens 中,使表征随群组上下文调整。作者另引入 context inference network,从某一群组的一批样本即时预测 context token。该机制依赖可用的群组归属及同组样本;摘要未说明群组的具体定义、上下文聚合方式、训练损失或测试时是否更新模型参数,因此不能将其直接等同于参数更新式测试时自适应。 作者报告,在监督微调设置下,为预训练 ViTs 加入上下文条件化机制,在 iWildCam 和 FMoW 上持续改善分布外泛化;在自监督表征学习设置下,ContextViT 在 Camelyon17 病理图像及 JUMP-CP 显微图像基准上,比对应 ViT 获得更稳定的分布偏移下图像表征。摘要未给出指标数值、数据划分或效应大小,实验协议、消融及统计信息尚未验证。复现时需核对群组构造、上下文批次采样、预训练条件与基线配置。 平台推测(待验证):迁移假设是,EEG 的被试或会话可作为群组,上下文条件化可能帮助处理跨被试或跨会话分布变化。可复用 context token 与上下文推断思路,但需将图像 token 编码改为 EEG 时序或时空编码,并明确测试群组可用样本及监督条件。低信噪比、小批次、通道配置差异,以及同组任务类别比例变化,都可能使上下文混入噪声或任务内容。一个可检验的小实验是在固定 Cross-subject 划分下,以同一 EEG 编码器比较无上下文与同组上下文条件化,并加入打乱群组的对照;严格固定测试上下文采样规则,检验增益是否来自有效群组信息。已有 EEG 相关工作尚未检索确认。
值得核对其由同组样本推断 context token 的机制及测试时分组协议:这为利用群组上下文处理分布偏移提供了具体方法,但其对 EEG 跨被试或跨会话表征的价值尚未验证。
来源:OpenReview · Representation learning · openreview.net