arXiv reaDer
生物医学画像セグメンテーションに対する神経ネットワークの自信過剰を処理するためのダイス損失のキャリブレーション
Calibrating the Dice loss to handle neural network overconfidence for biomedical image segmentation
ダイス類似係数(DSC)は、クラスの不均衡に対するロバスト性のため、生物医学画像セグメンテーションに広く使用されているメトリックと損失関数の両方です。ただし、DSC損失のキャリブレーションが不十分であるため、予測が過信になり、生物医学的および臨床的実践で有用に解釈できないことはよく知られています。パフォーマンスは、多くの場合、ディープニューラルネットワークによって生成されたセグメンテーションを評価するために使用される唯一のメトリックであり、キャリブレーションはしばしば無視されます。ただし、キャリブレーションは、生物医学および臨床診療への翻訳にとって重要であり、科学者や臨床医による解釈のための予測をモデル化するための重要なコンテキスト情報を提供します。この研究では、不十分なキャリブレーションを深層学習ベースの生物医学画像セグメンテーションの新たな課題として特定します。自信過剰で誤った予測に関連するペナルティを選択的に調整する、DSC ++損失と呼ばれるDSC損失のシンプルで効果的な拡張を提供します。スタンドアロンの損失関数として、DSC ++損失は、5つの十分に検証されたオープンソースの生物医学画像データセット全体で、従来のDSC損失よりも大幅に改善されたキャリブレーションを実現します。同様に、DSC ++損失を4つのDSCベースの損失関数に統合すると大幅に改善されます。最後に、softmaxしきい値を使用して、適切に調整された出力により、適合率-想起バイアスの調整が可能になることを示します。これは、生物医学的または臨床的タスクに合わせてモデル予測を適応させる重要な後処理手法です。 DSC ++損失は、DSCの主要な制限を克服し、生物医学および臨床診療で使用する深層学習セグメンテーションモデルをトレーニングするための適切な損失関数を提供します。
The Dice similarity coefficient (DSC) is both a widely used metric and loss function for biomedical image segmentation due to its robustness to class imbalance. However, it is well known that the DSC loss is poorly calibrated, resulting in overconfident predictions that cannot be usefully interpreted in biomedical and clinical practice. Performance is often the only metric used to evaluate segmentations produced by deep neural networks, and calibration is often neglected. However, calibration is important for translation into biomedical and clinical practice, providing crucial contextual information to model predictions for interpretation by scientists and clinicians. In this study, we identify poor calibration as an emerging challenge of deep learning based biomedical image segmentation. We provide a simple yet effective extension of the DSC loss, named the DSC++ loss, that selectively modulates the penalty associated with overconfident, incorrect predictions. As a standalone loss function, the DSC++ loss achieves significantly improved calibration over the conventional DSC loss across five well-validated open-source biomedical imaging datasets. Similarly, we observe significantly improved when integrating the DSC++ loss into four DSC-based loss functions. Finally, we use softmax thresholding to illustrate that well calibrated outputs enable tailoring of precision-recall bias, an important post-processing technique to adapt the model predictions to suit the biomedical or clinical task. The DSC++ loss overcomes the major limitation of the DSC, providing a suitable loss function for training deep learning segmentation models for use in biomedical and clinical practice.
updated: Sun Oct 31 2021 16:02:02 GMT+0000 (UTC)
published: Sun Oct 31 2021 16:02:02 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト