用动态标签平滑提升医疗影像模型预测可信度
Improving Predictive Confidence in Medical Imaging via Online Label Smoothing
- 根据模型自身预测模式实时调整软标签,动态修正分类误差
- 在RadImageNet上使准确率提升,特征嵌入更紧凑分离
- 适合需要高可靠性、可解释性医疗AI系统的研究者
深度学习模型在医学图像分类中表现优异,但常产生过度自信的预测,影响其在医疗场景中的可靠性。传统标签平滑方法忽略类别间关系,对所有非目标类一视同仁。本文提出在线标签平滑(OLS),一种基于模型预测模式动态调整软标签的方法。在大型RadImageNet数据集上,使用ResNet-50、MobileNetV2和VGG-19三种架构验证,结果表明OLS在Top-1和Top-5准确率上均优于标准训练、硬标签、传统标签平滑及无教师知识蒸馏方法。此外,OLS生成的特征嵌入更紧凑且分离度更高,表明其提升了表征学习能力。这些发现说明OLS不仅增强预测性能,还改善模型校准,是构建可信医疗AI系统的有效方案。
原文摘要 · Abstract (English)
Deep learning models, especially convolutional neural networks, have achieved impressive results in medical image classification. However, these models often produce overconfident predictions, which can undermine their reliability in critical healthcare settings. While traditional label smoothing offers a simple way to reduce such overconfidence, it fails to consider relationships between classes by treating all non-target classes equally. In this study, we explore the use of Online Label Smoothing (OLS), a dynamic approach that adjusts soft labels throughout training based on the model's own prediction patterns. We evaluate OLS on the large-scale RadImageNet dataset using three widely used architectures: ResNet-50, MobileNetV2, and VGG-19. Our results show that OLS consistently improves both Top-1 and Top-5 classification accuracy compared to standard training methods, including hard labels, conventional label smoothing, and teacher-free knowledge distillation. In addition to accuracy gains, OLS leads to more compact and well-separated feature embeddings, indicating improved representation learning. These findings suggest that OLS not only strengthens predictive performance but also enhances calibration, making it a practical and effective solution for developing trustworthy AI systems in the medical imaging domain.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。