深層学習ベースのモデルは、糖尿病性網膜症(DR)スクリーニングで網膜画像が「好ましい」かどうかを自動的に検出するために開発されています。ただし、入力画像がトレーニング分布から分布的にシフトするため、分類精度は低下します。さらに、入力が網膜画像でない場合でも、標準のDR分類器は、画像が「参照可能」であるという信頼性の高い予測を生成します。私たちの論文は、この問題に対処するためのディリクレ事前ネットワークベースのフレームワークを示しています。分布外(OOD)検出器モデルとDR分類モデルを利用して、OOD画像を識別することにより一般化可能性を向上させます。実世界のデータセットでの実験は、提案されたフレームワークが未知の非網膜画像を排除し、人間の介入のために分布的にシフトした網膜画像を特定できることを示しています。
Deep learning-based models are developed to automatically detect if a retina image is `referable' in diabetic retinopathy (DR) screening. However, their classification accuracy degrades as the input images distributionally shift from their training distribution. Further, even if the input is not a retina image, a standard DR classifier produces a high confident prediction that the image is `referable'. Our paper presents a Dirichlet Prior Network-based framework to address this issue. It utilizes an out-of-distribution (OOD) detector model and a DR classification model to improve generalizability by identifying OOD images. Experiments on real-world datasets indicate that the proposed framework can eliminate the unknown non-retina images and identify the distributionally shifted retina images for human intervention.