ディープラーニングモデルは、医療画像分析の一般的な選択肢になっています。ただし、ディープラーニングモデルの一般化パフォーマンスが低いため、医療アプリケーションには堅牢性が重要であるため、モデルを現実の世界に展開することはできません。たとえば、最先端の畳み込みニューラルネットワーク(CNN)は、敵対的サンプルやトレーニング分布から統計的に遠く離れて抽出されたサンプルを検出できません。この作業では、マラリア寄生細胞と非感染細胞を分類する際に、異常な入力サンプルを検出するためのシンプルで効果的な方法であるマハラノビス距離ベースの信頼スコアの堅牢性を実験的に評価します。結果は、マハラノビス信頼スコア検出器が深層学習モデルの改善されたパフォーマンスと堅牢性を示し、分布外(OOD)と敵対的なサンプルの両方で最先端のパフォーマンスを達成することを示しました。
Deep learning models have become a popular choice for medical image analysis. However, the poor generalization performance of deep learning models limits them from being deployed in the real world as robustness is critical for medical applications. For instance, the state-of-the-art Convolutional Neural Networks (CNNs) fail to detect adversarial samples or samples drawn statistically far away from the training distribution. In this work, we experimentally evaluate the robustness of a Mahalanobis distance-based confidence score, a simple yet effective method for detecting abnormal input samples, in classifying malaria parasitized cells and uninfected cells. Results indicated that the Mahalanobis confidence score detector exhibits improved performance and robustness of deep learning models, and achieves stateof-the-art performance on both out-of-distribution (OOD) and adversarial samples.