跳到正文
OpenReview · Representation learning· Qin Wang; Kai Krajsek; Hanno Scharr·· 2024-10-13精选AI 评分74

通过图像重建实现基于增强的自监督学习中的等变表征学习

Equivariant Representation Learning for Augmentation-based Self-Supervised Learning via Image Reconstruction

AI 导读

基于摘要分析。该研究面向视觉表征学习,针对基于增强的自监督方法偏重不变特征、可能忽视等变特征的问题,引入图像重建辅助任务,旨在同时保留对增强的不变性与随变换规律变化的表征信息。 核心机制是利用 cross-attention 融合两个增强视图的特征,再重建其中一个视图。与仅要求不同增强视图具有一致表征的思路相比,该辅助任务试图保留重建所需的变换相关信息。作者称该方法不增加额外参数,并可适配多种基于增强视图对的学习方法;但摘要未说明参数复用方式、重建目标、损失权重及训练与推理开销,具体实现条件尚未验证。 作者在人工数据集 3DIEBench 与自然图像数据集 ImageNet 上,通过多项线性回归任务和下游应用评估等变特征。作者报告,相比标准的基于增强的自监督方法及所比较的先进方法,该方法取得改善,尤其在组合增强情形下表现突出,并同时改善不变与等变特征学习。摘要未给出具体指标、数值、基线名称及数据划分,实验协议、消融及统计信息尚未验证,不能据此判断收益幅度或适用边界。 平台推测(待验证):其可迁移假设是,EEG 自监督学习也可能需要区分应被消除的扰动与应被保留的变换信息。可复用的是双视图特征融合与重建辅助任务,需改造的是时序输入、重建目标及增强定义;依赖条件是增强与任务标签的关系明确,并保留足够的可重建信息。低信噪比可能使重建偏向噪声,被试差异、通道布局变化和小数据也可能削弱收益。一个可证伪的小实验是在固定 EEG 编码器、训练划分与增强协议下,对比加入及不加入重建辅助任务的下游表现和变换参数线性可预测性,检查两者能否同时改善。原视觉结果不构成 EEG 或 BCI 有效性的证据,已有相关 EEG 工作尚未检索确认。

阅读价值

值得核对的是,作者通过跨视图图像重建辅助任务兼顾不变与等变表征,并报告在组合增强下的优势;其“不增加额外参数”的实现条件及对 EEG 的适用性尚未验证。

来源:OpenReview · Representation learning · openreview.net