arXiv reaDer
ラベル分布シフトのための医用画像分類ネットのキャリブレーションによるテスト時間の適応
Test-time Adaptation with Calibration of Medical Image Classification Nets for Label Distribution Shift
クラス分布は、深い分類器を学習する上で重要な役割を果たします。テストセット内の各クラスの比率がトレーニングセットと異なる場合、通常、分類ネットのパフォーマンスは低下します。病気の有病率は場所と時間によって変化するため、このようなラベル分布シフトの問題は医療診断で一般的です。本論文では、医用画像分類のためのラベルシフトに取り組む最初の方法を提案します。これは、単一のトレーニングラベル分布から学習したモデルを任意の未知のテストラベル分布に効果的に適応させます。私たちのアプローチは、分布のキャリブレーションを革新して、さまざまな1つの支配的なクラスの分布を処理できる複数の代表的な分類器を学習します。テスト画像が与えられると、未知のテストラベルの分布を処理するために、一貫性に基づくテスト時間の適応によって、さまざまな分類子が動的に集約されます。肝硬変の病期分類とCOVID-19重症度予測を含む2つの重要な医用画像分類タスクで私たちの方法を検証します。私たちの実験は、ラベルシフトの下でモデルのパフォーマンスが低下することを明確に示しています。私たちの方法では、モデルのパフォーマンスは、両方の医用画像診断タスクのラベルシフトが異なるすべてのテストデータセットで大幅に向上します。
Class distribution plays an important role in learning deep classifiers. When the proportion of each class in the test set differs from the training set, the performance of classification nets usually degrades. Such a label distribution shift problem is common in medical diagnosis since the prevalence of disease vary over location and time. In this paper, we propose the first method to tackle label shift for medical image classification, which effectively adapt the model learned from a single training label distribution to arbitrary unknown test label distribution. Our approach innovates distribution calibration to learn multiple representative classifiers, which are capable of handling different one-dominating-class distributions. When given a test image, the diverse classifiers are dynamically aggregated via the consistency-driven test-time adaptation, to deal with the unknown test label distribution. We validate our method on two important medical image classification tasks including liver fibrosis staging and COVID-19 severity prediction. Our experiments clearly show the decreased model performance under label shift. With our method, model performance significantly improves on all the test datasets with different label shifts for both medical image diagnosis tasks.
updated: Sat Jul 09 2022 09:37:38 GMT+0000 (UTC)
published: Sat Jul 02 2022 07:55:23 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アソシエイト