跳到正文
arXiv · 多模态与生成机制· Qinyu Li; Yee Whye Teh; Chris Holmes; Joel Jaskari·· 4 天前精选AI 评分72

通过扩散模型实现数据高效的分类器预训练

Data-Efficient Classifier Pretraining Through Diffusion Models

AI 导读

基于摘要分析。本文研究领域特定训练数据有限时分类器容易过拟合的问题,提出 DiffPT:利用扩散建模对给定分类器骨干进行预训练,再用于分类任务。其思路受到传统 auto-encoder 与 denoising auto-encoder 重建式预训练的启发,主要研究对象涵盖时序数据、自然图像和医学影像,并非专门面向 EEG 或 BCI。 作者将 DiffPT 与多种传统及现代无监督、自监督方法比较。作者报告,在 CIFAR-10 和 Tiny ImageNet 的低数据设置下,对比预训练表现更好;在所评估的医学影像与时序任务的多个低数据设置下,DiffPT 优于自监督基线。摘要未提供具体样本规模、划分方式、指标及数值,因此这些结论只能在作者所述评估范围内理解,不能据此认定扩散预训练普遍优于对比学习。 机制上,可确认的贡献是将扩散建模用于分类骨干的预训练;噪声预测或重建目标的具体形式、骨干与扩散组件的连接方式、预训练数据及标签使用方式、下游微调流程均尚未验证。医学影像与时序数据集名称、实验协议、消融及统计信息也尚未验证,复现前需查阅全文确认。 平台推测(待验证):假设扩散预训练能够从有限 EEG 数据中学习有利于分类的信号结构,DiffPT 可能对应小数据 EEG 分类中过拟合与表征学习不足的问题。可考虑复用其骨干预训练思路,但输入表示、噪声设置和通道处理需适配 EEG;低信噪比、伪迹、跨被试差异及通道配置变化可能使模型学习与任务无关的结构。一个可检验的小实验是在固定 EEG 分类骨干与跨被试划分下,仅用训练被试数据预训练,比较 DiffPT、随机初始化及对比预训练在相同标注预算下的表现,并记录多次运行的不确定性。已有 EEG 相关工作尚未检索确认,此建议不构成已验证的迁移效果。

阅读价值

值得关注其在小数据时序与医学影像任务中采用扩散建模预训练分类骨干的机制,以及相较对比预训练呈现的模态差异;其对 EEG 分类的价值尚未验证。

来源:arXiv · 多模态与生成机制 · arxiv.org