跳到正文
OpenReview · Representation learning· Jiho Jang; Seonhoon Kim; KiYoon Yoo; Jangho Kim; Chaerin Kong; Nojun Kwak·· 2023-01-05精选AI 评分73

自蒸馏自监督表征学习

Self-Distilled Self-supervised Representation Learning

AI 导读

基于摘要分析。本文研究图像自监督表征学习,提出 Self-Distilled Self-Supervised Learning(SDSSL):不仅在最终表征上学习,还通过对比损失让中间层向最终层学习,以改善 ViT 不同深度的表征及提前退出子网络的性能。 核心机制是将监督学习中的自蒸馏思路引入自监督训练。针对图像的两个视图,方法在最终层学习之外增加对中间表征的指导,使其更适合实例判别。作者认为,中间层能力增强还可降低最终层代理任务的难度,进而改善最终表征。摘要未给出具体损失形式、梯度处理方式、中间层选择及各项损失权重,这些实现细节尚未验证。 作者报告,在采用 ViT、覆盖多种任务与数据集的实验中,SDSSL 优于 SimCLR、BYOL 和 MoCo v3 等基线;在线性评估与 k-NN 协议下,最终层及多数较低层均有性能改善。作者还报告,提前退出子网络相较完整网络的性能下降较小,并通过定性与定量分析考察沿 Transformer 层的表征形成过程。摘要未提供数据集名称、具体指标、收益幅度或计算开销,实验协议、消融及统计信息尚未验证。材料仅称代码将提供,不能据此确认已发布。 平台推测(待验证):若 EEG 自监督编码器也存在中间层判别能力不足的问题,可假设最终层到中间层的蒸馏有助于学习可提前退出的表征。原方法依赖图像双视图及实例判别训练,迁移时需改造 EEG 视图构造与中间表征对齐;低信噪比、通道差异和小数据可能使最终层的噪声或被试特征向浅层传播。一个可证伪的小实验是在固定 Cross-subject 划分、增强方式与训练预算下,对比同一 EEG 编码器加入蒸馏前后最终层及提前退出层的 Balanced Accuracy 和推理开销。此为迁移假设,并非本文已验证结果;已有 EEG 相关工作尚未检索确认。

阅读价值

值得核对的是以最终层表征指导中间层的自蒸馏机制,及其能否同时改善 ViT 的分层表征与提前退出性能;摘要报告了相对多种基线的优势,但具体实验协议与收益幅度尚未验证。

来源:OpenReview · Representation learning · openreview.net