用大模型知识蒸馏出小模型,提升医疗图像分割精度。
Task-Specific Knowledge Distillation from the Vision Foundation Model for Enhanced Medical Image Segmentation
- 先微调大模型抓取任务特征,再蒸馏给小模型。
- 在80样本下比传统方法高28%的分割准确率。
- 适合数据少的医疗影像场景,尤其擅长小样本学习。
大规模预训练模型如视觉基础模型(VFM)在下游任务中表现出色,尤其在目标数据有限时。然而,其高计算成本及自然图像与医学图像间的领域差距限制了其在医学分割中的应用。为此,我们提出一种通用的任务特定知识蒸馏框架:先在目标分割任务上微调VFM以捕捉任务特异性特征,再通过低秩适配(LoRA)降低微调开销,并引入扩散模型生成的合成数据增强训练集。在五个医学图像数据集上的实验表明,该方法持续优于无任务特性的知识蒸馏及自监督预训练方法(如MoCo v3和MAE)。例如,在KidneyUS数据集上,使用80个标注样本时,本方法的Dice分数比任务无关知识蒸馏高出28%;在CHAOS数据集上,使用100个样本时,性能比MAE提升11%。结果验证了任务特定知识蒸馏在数据受限条件下训练精准高效模型的潜力。
原文摘要 · Abstract (English)
Large-scale pre-trained models, such as Vision Foundation Models (VFMs), have demonstrated impressive performance across various downstream tasks by transferring generalized knowledge, especially when target data is limited. However, their high computational cost and the domain gap between natural and medical images limit their practical application in medical segmentation tasks. Motivated by this, we pose the following important question: "How can we effectively utilize the knowledge of large pre-trained VFMs to train a small, task-specific model for medical image segmentation when training data is limited?" To address this problem, we propose a novel and generalizable task-specific knowledge distillation framework. Our method fine-tunes the VFM on the target segmentation task to capture task-specific features before distilling the knowledge to smaller models, leveraging Low-Rank Adaptation (LoRA) to reduce the computational cost of fine-tuning. Additionally, we incorporate synthetic data generated by diffusion models to augment the transfer set, enhancing model performance in data-limited scenarios. Experimental results across five medical image datasets demonstrate that our method consistently outperforms task-agnostic knowledge distillation and self-supervised pretraining approaches like MoCo v3 and Masked Autoencoders (MAE). For example, on the KidneyUS dataset, our method achieved a 28% higher Dice score than task-agnostic KD using 80 labeled samples for fine-tuning. On the CHAOS dataset, it achieved an 11% improvement over MAE with 100 labeled samples. These results underscore the potential of task-specific knowledge distillation to train accurate, efficient models for medical image segmentation in data-constrained settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。