arXiv:2410.02387cs.LGcs.AI2024-10被引 2

用双层优化提升自监督预训练与下游任务的对齐,让模型更懂下游目标。

BiSSL: Enhancing the Alignment Between Self-Supervised Pretraining and Downstream Fine-Tuning via Bilevel Optimization

  • 设计双层优化框架,让预训练和微调阶段相互协作
  • 在12个图像分类数据集上准确率显著提升,检测任务也受益
  • 适用于多种预训练方法,适合追求性能的视觉模型开发者

从自监督预训练初始化的模型可能与下游任务对齐不佳,限制了微调时特征适应能力。为解决此问题,我们提出BiSSL,一种在传统自监督预训练后引入的双层优化训练框架。该框架将预训练任务作为下层目标、下游任务作为上层目标,显式建模预训练与微调之间的依赖关系,促进两阶段间的信息共享,从而获得更贴近下游任务的模型初始化。我们提出一个通用算法,兼容多种预训练和下游任务。以SimCLR和BYOL在ImageNet上预训练ResNet-50为例,实验表明,该框架在绝大多数12个下游图像分类数据集及目标检测任务上均显著提升准确率。探索性分析进一步验证了其增强下游对齐的效果。

原文摘要 · Abstract (English)

Models initialized from self-supervised pretraining may suffer from poor alignment with downstream tasks, reducing the extent to which subsequent fine-tuning can adapt pretrained features toward downstream objectives. To mitigate this, we introduce BiSSL, a novel bilevel training framework that enhances the alignment of self-supervised pretrained models with downstream tasks prior to fine-tuning. BiSSL acts as an intermediate training stage conducted after conventional self-supervised pretraining and is tasked with solving a bilevel optimization problem that incorporates the pretext and downstream training objectives in its lower- and upper-level objectives, respectively. This approach explicitly models the interdependence between the pretraining and fine-tuning stages within the conventional self-supervised learning pipeline, facilitating enhanced information sharing between them that ultimately leads to a model initialization better aligned with the downstream task. We propose a general training algorithm for BiSSL that is compatible with a broad range of pretext and downstream tasks. Using SimCLR and Bootstrap Your Own Latent to pretrain ResNet-50 backbones on the ImageNet dataset, we demonstrate that our proposed framework significantly improves accuracy on the vast majority of 12 downstream image classification datasets, as well as on object detection. Exploratory analyses alongside investigative experiments further provide compelling evidence that BiSSL enhances downstream alignment.

自监督学习双层优化模型对齐图像分类

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。