论自监督学习的分布外泛化
On the Out-of-Distribution Generalization of Self-Supervised Learning
基于摘要分析。本文研究自监督学习(SSL)的分布外(OOD)泛化,从 mini-batch 构造解释 SSL 可能具备泛化能力的原因,并针对训练中学到的伪相关,提出基于结构因果模型的干预后分布(post-intervention distribution,PID)及相应批次采样策略。 机制与理论:作者从数据生成和因果推断视角分析 SSL,认为伪相关会降低 OOD 泛化。PID 用于描述变量关系不受伪相关影响的情形;作者报告,在每个训练 mini-batch 满足 PID 的条件下,SSL 模型可达到最优的最坏情形 OOD 表现。为落实该条件,方法通过学习潜变量模型实施受 PID 约束的批次采样。作者还报告了潜变量模型的可识别性分析,但摘要未给出具体假设、变量定义、目标函数与采样流程,不能将上述条件性结论理解为任意数据上的普遍保证。 实验与复现:作者报告,该采样策略在多种下游 OOD 任务上有效;摘要未提供任务名称、数据集、划分、基线或指标数值。实验协议、消融及统计信息尚未验证。复现需重点核对 PID 的可操作定义、潜变量模型的可识别条件,以及采样相较原始 SSL 训练增加的计算与数据需求。 平台推测(待验证):其机制可能对应 EEG 自监督预训练中被试或会话相关伪特征导致的域泛化瓶颈,但依赖 EEG 数据中存在可识别、可估计的潜在生成因素。可尝试复用批次采样思想,潜变量模型与 PID 约束则需适配 EEG 数据结构;低信噪比、通道差异及小样本可能使潜因素估计失真,甚至削弱任务相关信息。一个小规模可证伪实验是在固定 EEG 编码器、SSL 目标和训练预算下,对比随机采样与 PID 约束采样,在严格隔离目标被试的 Cross-subject 协议中评估下游表现。此项迁移尚未验证,已有相关 EEG 工作尚未检索确认。
值得关注其将自监督学习的 OOD 泛化问题落实到 mini-batch 构造与因果约束采样的机制,但理论假设、采样实现及下游收益仍需全文核对。
来源:OpenReview · Representation learning · openreview.net