用持续自监督预训练提升图像分割模型在小数据场景下的表现
Enhancing Semantic Segmentation with Continual Self-Supervised Pre-training
- 设计新任务GLARE,结合局部与区域一致性增强特征学习
- 仅更新轻量适配器,实现低计算开销的持续预训练
- 适用于数据少的图像分割任务,尤其适合资源受限场景
自监督学习(SSL)已成为训练基础视觉模型的核心范式,利用大规模无标签数据生成具备强泛化能力的表征。这些模型通常在ImageNet等通用数据集上预训练,再通过微调适配下游任务。尽管已有研究探索了参数高效适配方法(如adapter、LoRA、prompt tuning),但针对下游密集预测任务(如语义分割)的持续自监督预训练仍缺乏系统探索,尤其在数据有限的新领域中。本文提出一种面向语义分割的持续自监督预训练方法GLARE(Global Local and Regional Enforcement),通过引入块级增强促进局部一致性,并加入基于空间语义的区域一致性约束。为实现高效持续训练,采用现有SSL模型权重初始化视觉变换器(ViT),仅更新轻量级适配模块UniAdapter,其余主干保持冻结。在多个不同领域的语义分割基准测试中,GLARE在极低计算和参数开销下持续提升下游性能。
原文摘要 · Abstract (English)
Self-supervised learning (SSL) has emerged as a central paradigm for training foundation models by leveraging large-scale unlabeled datasets, often producing representations with strong generalization capabilities. These models are typically pre-trained on general-purpose datasets such as ImageNet and subsequently adapted to various downstream tasks through finetuning. While prior work has investigated parameter-efficient adaptation methods like adapters, LoRA, and prompt tuning, primarily targeting downstream finetuning, extending the SSL pre-training itself in a continual manner to new domains under limited data remains largely underexplored, especially for downstream dense prediction tasks like semantic segmentation. In this work, we address the challenge of adapting vision foundation models to low-data target domains through continual self-supervised pre-training, specifically targeting downstream semantic segmentation. We propose GLARE (Global Local and Regional Enforcement), a novel continual self-supervised pre-training task designed to enhance downstream semantic segmentation performance. GLARE introduces patch-level augmentations to encourage local consistency and incorporates a regional consistency constraint that leverages spatial semantics in the data. For efficient continual pre-training, we initialize Vision Transformers (ViTs) with weights from existing SSL models and update only lightweight adapter modules specifically UniAdapter - while keeping the rest of the backbone frozen. Experiments across multiple semantic segmentation benchmarks on different domains demonstrate that GLARE consistently improves downstream performance with minimal computational and parameter overhead.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。