arXiv reaDer
Anatomy X-Net:胸腔疾患分類のための半教師あり解剖学認識畳み込みニューラルネットワーク
Anatomy X-Net: A Semi-Supervised Anatomy Aware Convolutional Neural Network for Thoracic Disease Classification
深層学習法を使用した胸部X線写真からの胸部疾患の検出は、過去10年間で活発な研究分野となっています。以前のほとんどの方法は、モデルの予測に大きく寄与する原因となる空間領域を特定することにより、画像の病変臓器に焦点を当てようとします。対照的に、専門の放射線科医は、それらの領域が異常であるかどうかを判断する前に、最初に顕著な解剖学的構造を見つけます。したがって、解剖学的知識を深層学習モデルに統合すると、自動疾患分類が大幅に改善される可能性があります。この作品は、Anatomy X-Netという名前の解剖学を意識した注意ベースのアーキテクチャを提案します。これは、事前に識別された解剖学的領域によって導かれる空間的特徴に優先順位を付けます。臓器レベルの注釈を含むJSRTデータセットを使用した半教師あり学習方法を活用して、NIHおよびCheXpertデータセットの解剖学的セグメンテーションマスク(肺および心臓用)を取得します。提案された Anatomy X-Net は、解剖学的注意学習のためのまとまりのあるフレームワークで、2 つの対応する構造化モジュールである Anatomy Aware Attention (AAA) と Probabilistic Weighted Average Pooling (PWAP) を備えたバックボーン ネットワークとして事前訓練された DenseNet-121 を使用します。私たちの提案する方法は、0.8439のAUCスコアで公式NIHテストセットに新しい最先端のパフォーマンスを設定し、胸部疾患の分類を改善するために解剖学的セグメンテーションの知識を利用することの有効性を証明します。さらに、Anatomy X-Netは、Stanford CheXpertデータセットで平均0.9020のAUCを生成し、提案されたフレームワークの一般化可能性を実証する既存の方法を改善します。
Thoracic disease detection from chest radiographs using deep learning methods has been an active area of research in the last decade. Most previous methods attempt to focus on the diseased organs of the image by identifying spatial regions responsible for significant contributions to the model's prediction. In contrast, expert radiologists first locate the prominent anatomical structures before determining if those regions are anomalous. Therefore, integrating anatomical knowledge within deep learning models could bring substantial improvement in automatic disease classification. This work proposes an anatomy-aware attention-based architecture named Anatomy X-Net, that prioritizes the spatial features guided by the pre-identified anatomy regions. We leverage a semi-supervised learning method using the JSRT dataset containing organ-level annotation to obtain the anatomical segmentation masks (for lungs and heart) for the NIH and CheXpert datasets. The proposed Anatomy X-Net uses the pre-trained DenseNet-121 as the backbone network with two corresponding structured modules, the Anatomy Aware Attention (AAA) and Probabilistic Weighted Average Pooling (PWAP), in a cohesive framework for anatomical attention learning. Our proposed method sets new state-of-the-art performance on the official NIH test set with an AUC score of 0.8439, proving the efficacy of utilizing the anatomy segmentation knowledge to improve the thoracic disease classification. Furthermore, the Anatomy X-Net yields an averaged AUC of 0.9020 on the Stanford CheXpert dataset, improving on existing methods that demonstrate the generalizability of the proposed framework.
updated: Thu Jun 10 2021 17:01:23 GMT+0000 (UTC)
published: Thu Jun 10 2021 17:01: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アソシエイト