让编码器和解码器一起预训练,提升密集预测任务表现
Beyond the Encoder: Joint Encoder-Decoder Contrastive Pre-Training Improves Dense Prediction
- 设计联合对比学习框架,让编码器和解码器共同预训练
- 在COCO上用ResNet-50训练,检测和分割指标分别提升0.37和0.32 AP
- 效果跨模型、数据集、任务泛化,尤其在小样本场景下更优
自监督对比学习通常只预训练编码器,而解码器则在下游任务中单独训练。本文提出DeCon,一种高效的编码器-解码器自监督学习框架,支持联合对比预训练。通过扩展现有架构以适配多样解码器及对应对比损失,并引入非竞争性加权损失函数,实现编码器与解码器的协同优化。在ImageNet-1K、COCO和COCO+上预训练时,DeCon在多数评估任务中达到领先性能。例如,在COCO上使用ResNet-50编码器预训练,目标检测和实例分割分别提升0.37 AP和0.32 AP;Pascal VOC语义分割提升1.42 mIoU,Cityscapes提升0.50 mIoU。该效果在多种骨干网络、解码器、数据集和密集预测任务(包括分割与检测)中均成立,且在域外场景(如小样本设置)中依然有效,证明联合预训练显著提升了密集预测的表征质量。
原文摘要 · Abstract (English)
Contrastive learning methods in self-supervised settings have primarily focused on pre-training encoders, while decoders are typically introduced and trained separately for downstream dense prediction tasks. However, this conventional approach overlooks the potential benefits of jointly pre-training both encoder and decoder. In this paper, we propose DeCon, an efficient encoder-decoder self-supervised learning (SSL) framework that supports joint contrastive pre-training. We first extend existing SSL architectures to accommodate diverse decoders and their corresponding contrastive losses. Then, we introduce a weighted encoder-decoder contrastive loss with non-competing objectives to enable the joint pre-training of encoder-decoder architectures. By adapting a contrastive SSL framework for dense prediction, DeCon establishes consistent state-of-the-art performance on most of the evaluated tasks when pre-trained on Imagenet-1K, COCO and COCO+. Notably, when pre-training a ResNet-50 encoder on COCO dataset, DeCon improves COCO object detection and instance segmentation compared to the baseline framework by +0.37 AP and +0.32 AP, respectively, and boosts semantic segmentation by +1.42 mIoU on Pascal VOC and by +0.50 mIoU on Cityscapes. These improvements generalize across recent backbones, decoders, datasets, and dense tasks beyond segmentation and object detection, and persist in out-of-domain scenarios, including limited-data settings, demonstrating that joint pre-training significantly enhances representation quality for dense prediction. Code is available at https://github.com/sebquetin/DeCon.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。