arXiv reaDer
深いショートカット学習を克服するための注釈の粒度の再考:胸部X線写真に関する後ろ向き研究
Rethinking annotation granularity for overcoming deep shortcut learning: A retrospective study on chest radiographs
ディープラーニングは、放射線科医と同等またはそれ以上の放射線写真スクリーニング性能を実証しています。ただし、最近の研究によると、胸部疾患分類の詳細モデルは、通常、外部データに適用するとパフォーマンスが低下することが示されています。このような現象は、ショートカット学習に分類できます。この場合、ディープモデルは、同じように分散されたトレーニングとテストセットに適合することができるが、他の分布に一般化できない意図しない決定ルールを学習します。この欠陥を軽減する自然な方法は、病変を明示的に示し、意図された特徴の学習にモデルを集中させることです。この論文では、一般的な胸部疾患分類モデルであるCheXNetと胸部病変検出モデルであるCheXDetを比較するために、広範な遡及的実験を実施します。最初に、2つのモデルが、多くのシナリオで有意差がなく、内部テストセットで同様の画像レベルの分類パフォーマンスを達成したことを示しました。一方、外部トレーニングデータを組み込むと、CheXNetのパフォーマンスが低下することさえありました。次に、病変の位置特定タスクでのモデルの内部パフォーマンスを比較し、80%少ないトレーニングデータが与えられた場合でも、CheXDetがCheXNetよりも大幅に優れたパフォーマンスを達成したことを示しました。モデルの意思決定領域をさらに視覚化することにより、CheXNetが標的病変以外のパターンを学習したことを明らかにし、そのショートカット学習の欠陥を示しました。さらに、CheXDetは、画像レベルの分類タスクと病変の位置特定タスクの両方で、CheXNetよりも大幅に優れた外部パフォーマンスを実現しました。私たちの調査結果は、臨床使用のための将来の深層学習ベースの診断システムを向上させる有望な方法として、深層学習システムをトレーニングするための注釈の粒度を改善することを示唆しています。
Deep learning has demonstrated radiograph screening performances that are comparable or superior to radiologists. However, recent studies show that deep models for thoracic disease classification usually show degraded performance when applied to external data. Such phenomena can be categorized into shortcut learning, where the deep models learn unintended decision rules that can fit the identically distributed training and test set but fail to generalize to other distributions. A natural way to alleviate this defect is explicitly indicating the lesions and focusing the model on learning the intended features. In this paper, we conduct extensive retrospective experiments to compare a popular thoracic disease classification model, CheXNet, and a thoracic lesion detection model, CheXDet. We first showed that the two models achieved similar image-level classification performance on the internal test set with no significant differences under many scenarios. Meanwhile, we found incorporating external training data even led to performance degradation for CheXNet. Then, we compared the models' internal performance on the lesion localization task and showed that CheXDet achieved significantly better performance than CheXNet even when given 80% less training data. By further visualizing the models' decision-making regions, we revealed that CheXNet learned patterns other than the target lesions, demonstrating its shortcut learning defect. Moreover, CheXDet achieved significantly better external performance than CheXNet on both the image-level classification task and the lesion localization task. Our findings suggest improving annotation granularity for training deep learning systems as a promising way to elevate future deep learning-based diagnosis systems for clinical usage.
updated: Wed Apr 21 2021 14:21:37 GMT+0000 (UTC)
published: Wed Apr 21 2021 14:21:37 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アソシエイト