跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

自监督视觉表示学习:预训练微调还是联合训练?

文章摘要

一项新研究系统比较了自监督学习中的两种训练范式:预训练后微调(PFT)和联合训练(JT),发现JT在数据效率和低标签设置下表现更优,而PFT在专门领域更可靠。

来源arXiv Computer Vision作者: Nusrat Munia, Tyler Ward, Nishat Nayla, Matthew A. Massey, Abdullah-Al-Zubaer Imran
自监督视觉表示学习:预训练微调还是联合训练?
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

自监督学习(SSL)通过利用未标注数据学习视觉表示,已成为计算机视觉领域的关键技术。传统方法通常采用两阶段流程:先在未标注数据上进行预训练,再在标注数据上进行微调(PFT)。然而,自监督和监督学习目标之间的相互作用尚未被充分理解。本研究由Nusrat Munia等人开展,旨在探索联合训练(JT)——即在同一个网络中同时优化自监督损失和监督损失——是否能提供更好的替代方案。

研究团队在八个代表性SSL方法(包括SimCLR、MoCo V2、BYOL、SwAV、DINO、SimSiam、Barlow Twins和VICReg)和多种计算机视觉任务上进行了系统实验,涵盖自然图像(ImageNet)、医学影像(Kather、Retina等)、危机响应(灾难图像分类)和遥感数据(卫星图像分类)。他们评估了不同标注数据比例(1%, 5%, 10%, 50%, 100%)下的性能,结果显示PFT和JT的相对有效性高度依赖于具体任务、标注数据的可用性以及领域的复杂度。具体而言,JT在低标签设置下表现出更强的鲁棒性,并持续提升数据和训练效率;而PFT在更专门化的领域中更为可靠。例如,在医学影像任务中,PFT往往优于JT,可能是因为医学标注数据质量较高且领域特征显著;而在自然图像中,JT在低标签比例下优势明显。

进一步的分析涉及表示质量(线性探测和KNN分类)、鲁棒性(对噪声和域偏移的抵抗)和跨领域泛化(使用不同数据源进行迁移学习)。研究表明,JT学习到的表示在低标签场景下更具判别性,而PFT在高标签场景下可能略优。此外,跨领域泛化实验表明,JT在从自然图像迁移到医学图像时表现更好,而PFT在领域内微调时更稳定。研究还发现,JT的收敛速度更快,且对超参数不太敏感。

该工作建立了一个混合SSL半监督学习的综合实证基准,并为不同视觉应用选择训练策略提供了实用指导。研究人员还发布了相关代码和数据(在GitHub上提供),以促进可重复性研究。该论文已提交至arXiv(编号2607.13192),并计划在后续会议中发表。研究还指出,未来的工作可以探索更高级的联合训练策略,如动态调整自监督和监督损失的权重。

展开要点与分析

文章情报

研究者进阶

要点

  • 研究比较了八种自监督方法和多种视觉任务,评估不同标签比例下的性能。
  • 联合训练(JT)同时优化自监督和监督损失,在低标签场景下更鲁棒且高效。
  • 预训练微调(PFT)在专业化领域更可靠,但整体效果取决于任务、标签量和领域复杂度。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。