用新指标选最佳预训练时机,提升医学图像分割效率
Enhancing pretraining efficiency for medical image segmentation via transferability metrics
- 基于对比学习设计可衡量迁移鲁棒性的新指标
- 较短预训练时间反而获得更好下游性能
- 适合需要高效训练医学图像分割模型的研究者
在医学图像分割任务中,标注数据稀缺导致深度网络训练困难。常用方法是使用U-Net结构,并在ImageNet等大规模通用数据集上预训练编码器。然而,这些方法耗时且无法保证下游任务性能提升。本文研究了300多种模型、数据集和训练方法组合,在医学图像分割数据集上的表现。结果发现,较短的预训练周期往往带来更好的下游性能,进一步证实ImageNet准确率不能有效预测下游表现。本文提出一种基于对比学习的新型可迁移性度量方法,能评估预训练模型对目标数据的鲁棒表示能力。该方法适用于从ImageNet分类到医学图像分割的迁移任务。通过在预训练过程中持续测量该得分,可识别最优迁移时机,减少预训练时间并提升目标任务性能。
原文摘要 · Abstract (English)
In medical image segmentation tasks, the scarcity of labeled training data poses a significant challenge when training deep neural networks. When using U-Net-style architectures, it is common practice to address this problem by pretraining the encoder part on a large general-purpose dataset like ImageNet. However, these methods are resource-intensive and do not guarantee improved performance on the downstream task. In this paper we investigate a variety of training setups on medical image segmentation datasets, using ImageNet-pretrained models. By examining over 300 combinations of models, datasets, and training methods, we find that shorter pretraining often leads to better results on the downstream task, providing additional proof to the well-known fact that the accuracy of the model on ImageNet is a poor indicator for downstream performance. As our main contribution, we introduce a novel transferability metric, based on contrastive learning, that measures how robustly a pretrained model is able to represent the target data. In contrast to other transferability scores, our method is applicable to the case of transferring from ImageNet classification to medical image segmentation. We apply our robustness score by measuring it throughout the pretraining phase to indicate when the model weights are optimal for downstream transfer. This reduces pretraining time and improves results on the target task.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。