arXiv reaDer
皮膚鏡画像分類のための分布外検出
Out-of-Distribution Detection for Dermoscopic Image Classification
医用画像診断は、各疾患クラスに十分な多様なトレーニングデータがあれば、ディープニューラルネットワークによって実現できます。ただし、トレーニング中に遭遇しなかったこれまで知られていなかった疾患クラスは、たとえ低い確率で予測されたとしても、必然的に誤分類されます。この問題は、これまで知られていなかった疾患の画像が診断のために提示される場合、特に画像が皮膚鏡の皮膚画像などの同じ画像領域からのものである場合、医療画像診断にとって特に重要です。現在の分布外検出アルゴリズムは、分布内クラスが不均衡な場合、トレーニングセット内で最も多くの疾患を優先することにより、不公平に機能します。これは、しばしば医学的に重要であるまれなケースの誤った診断につながる可能性があります。ニューラルネットワークをトレーニングするための斬新でシンプルな方法を開発しました。これにより、ニューラルネットワークは、分布内の皮膚鏡による皮膚疾患の画像を分類し、テスト時に皮膚鏡の画像から新しい疾患を検出できます。 BinaryHeadsモデルは、データが不均衡な場合に分類のバランスの取れた精度を損なうだけでなく、バランスの取れた精度を一貫して向上させることを示します。また、実際の設定で発生する可能性のあるさまざまな量の分布外データの存在に基づいて、分布外検出方法の有効性を調査するための重要な方法を紹介します。
Medical image diagnosis can be achieved by deep neural networks, provided there is enough varied training data for each disease class. However, a hitherto unknown disease class not encountered during training will inevitably be misclassified, even if predicted with low probability. This problem is especially important for medical image diagnosis, when an image of a hitherto unknown disease is presented for diagnosis, especially when the images come from the same image domain, such as dermoscopic skin images. Current out-of-distribution detection algorithms act unfairly when the in-distribution classes are imbalanced, by favouring the most numerous disease in the training sets. This could lead to false diagnoses for rare cases which are often medically important. We developed a novel yet simple method to train neural networks, which enables them to classify in-distribution dermoscopic skin disease images and also detect novel diseases from dermoscopic images at test time. We show that our BinaryHeads model not only does not hurt classification balanced accuracy when the data is imbalanced, but also consistently improves the balanced accuracy. We also introduce an important method to investigate the effectiveness of out-of-distribution detection methods based on presence of varying amounts of out-of-distribution data, which may arise in real-world settings.
updated: Mon Apr 19 2021 05:47:57 GMT+0000 (UTC)
published: Thu Apr 15 2021 23:34:53 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アソシエイト