医学影像迁移学习中,源模型选择的评估指标稳定性差,易受数据集微小变化影响。
Robustness of transferability estimation metrics for medical imaging

- 构建不同规模和随机种子的小型数据集,隔离测试目标数据集变化的影响
- 发现目标数据集微调即导致评估排名大幅变动,跨实验可复现性差
- 提示当前迁移能力评估指标在医学影像中可靠性不足,适合关注模型选择稳健性的研究者
在迁移学习中,源模型的选择显著影响目标数据集上的性能。然而,在医学影像领域,如何在通用数据集(如ImageNet)与领域特定数据集间选择合适的源模型仍具挑战。转移能力估计(TE)指标旨在以计算高效的方式预测最佳源模型。但先前研究因实验设置差异报告了矛盾的结果。多数TE指标针对自然图像设计并以准确率优化,而医学影像更常用对类别不平衡鲁棒的指标。本文通过构建不同样本量和随机种子的微型数据集,单独考察目标数据集变化的影响,并研究评估指标对参考排序的影响。结果表明,目标数据集的微小修改会改变排名;且评估指标的选择也会影响参考排序,进而影响对TE指标的评价。总体上,TE指标与参考排序之间的一致性较低。代码、模型检查点和数据划分已公开于https://github.com/niclasclassen/robustness-of-transferability-estimation-metrics-for-medical-imaging。
原文摘要 · Abstract (English)
In transfer learning, the choice of source model largely influences the performance on a target dataset. Still, selecting a fitting source remains a challenging task, especially in medical imaging where one has to decide between models pre-trained on off-the-shelf options, such as ImageNet, and domain specific datasets. Transferability estimation (TE) metrics address this problem by aiming to predict the best performing source model in a computationally cost effective way. However, previous work has reported conflicting TE metric performances due to differences in experimental setups. Moreover, most TE metrics are designed for and evaluated on natural images, while being optimized for accuracy, whereas in medical imaging metrics that are more robust to class imbalance are typically used. We study the impact of varying the target dataset as an isolated factor, by constructing miniature populations of different sample sizes and random seeds. In addition, we investigate the influence of the evaluation metric used to obtain the reference ranking. We find that small modifications to the target dataset change the rankings. Furthermore, we show that the choice of evaluation metric affects the reference rankings and therefore the evaluation of TE metrics. Overall, we observe a low agreement between rankings from TE metrics and reference. The code, model checkpoints and data splits used in this work are available through https://github.com/niclasclassen/robustness-of-transferability-estimation-metrics-for-medical-imaging.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。