arXiv:2512.02438cs.CVcs.AI2025-12

用动量自蒸馏提升医疗视觉语言模型,单卡训练也能高效出成果。

Boosting Medical Vision-Language Pretraining via Momentum Self-Distillation under Limited Computing Resources

  • 用动量自蒸馏增强多模态学习,提升模型知识提取能力。
  • 少样本适配时AUC-ROC超90%,检索任务性能提升2-3%。
  • 单张显卡运行,训练效率高,适合资源有限的研究者。

在医疗健康领域,详细标注获取困难,亟需鲁棒的视觉-语言模型(VLM)。预训练的VLM可微调小数据集或实现零样本推理,性能接近专用模型。对比学习(CL)是训练VLM的关键范式,但通常需要大批次以有效学习,计算成本高,常限于资源充足机构。此外,医疗数据有限,训练中需同时从数据和模型中挖掘知识以提升性能。为此,我们结合动量方法与蒸馏技术,兼顾计算效率与知识利用。贡献包括:(1) 提出动量自蒸馏增强多模态学习;(2) 将动量机制与梯度累积结合,扩大有效批次大小而不增加资源消耗。本方法在零样本分类上达到SOTA水平,少样本适应时AUC-ROC超过90%,检索任务提升2-3%。重要的是,仅用单张GPU即可高效训练,保持合理时间。代码已公开于https://github.com/phphuc612/MSD。

原文摘要 · Abstract (English)

In medical healthcare, obtaining detailed annotations is challenging, highlighting the need for robust Vision-Language Models (VLMs). Pretrained VLMs enable fine-tuning on small datasets or zero-shot inference, achieving performance comparable to task-specific models. Contrastive learning (CL) is a key paradigm for training VLMs but inherently requires large batch sizes for effective learning, making it computationally demanding and often limited to well-resourced institutions. Moreover, with limited data in healthcare, it is important to prioritize knowledge extraction from both data and models during training to improve performance. Therefore, we focus on leveraging the momentum method combined with distillation to simultaneously address computational efficiency and knowledge exploitation. Our contributions can be summarized as follows: (1) leveraging momentum self-distillation to enhance multimodal learning, and (2) integrating momentum mechanisms with gradient accumulation to enlarge the effective batch size without increasing resource consumption. Our method attains competitive performance with state-of-the-art (SOTA) approaches in zero-shot classification, while providing a substantial boost in the few-shot adaption, achieving over 90% AUC-ROC and improving retrieval tasks by 2-3%. Importantly, our method achieves high training efficiency with a single GPU while maintaining reasonable training time. Our approach aims to advance efficient multimodal learning by reducing resource requirements while improving performance over SOTA methods. The implementation of our method is available at https://github.com/phphuc612/MSD .

医疗AI多模态自蒸馏单卡训练

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