arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2009.04057cs.CVcs.LGeess.IV
收藏

一种改进的医学影像分类神经网络可训练标定方法

Improved Trainable Calibration Method for Neural Networks on Medical Imaging Classification

University of Kentucky(肯塔基大学)

Gongbo Liang, Yu Zhang, Xiaoqin Wang, Nathan Jacobs

摘要:

最近的工作表明,深度神经网络可以在医学成像领域的广泛图像分类任务中实现超人的性能。然而,这些工作主要集中在分类精度,忽略了不确定性量化的重要作用。从经验上讲,神经网络经常在预测中被错误校准和过度自信。这种误校准在任何自动决策系统中都可能存在问题,但我们专注于医学领域,其中神经网络误校准有可能导致重大的治疗错误。我们提出了一种新的校准方法,保持整体分类精度,同时显着提高模型校准。所提出的方法是基于预期的校准误差,这是一个共同的度量量化误校准。我们的方法可以很容易地集成到任何分类任务中作为辅助损失项,因此不需要明确的训练轮进行校准。我们表明,我们的方法在各种架构和数据集上显着降低了校准误差。

英文摘要:

Recent works have shown that deep neural networks can achieve super-human performance in a wide range of image classification tasks in the medical imaging domain. However, these works have primarily focused on classification accuracy, ignoring the important role of uncertainty quantification. Empirically, neural networks are often miscalibrated and overconfident in their predictions. This miscalibration could be problematic in any automatic decision-making system, but we focus on the medical field in which neural network miscalibration has the potential to lead to significant treatment errors. We propose a novel calibration approach that maintains the overall classification accuracy while significantly improving model calibration. The proposed approach is based on expected calibration error, which is a common metric for quantifying miscalibration. Our approach can be easily integrated into any classification task as an auxiliary loss term, thus not requiring an explicit training round for calibration. We show that our approach reduces calibration error significantly across various architectures and datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑