arXiv reaDer
深部肺炎:胸部X線でのクラス不均衡な肺炎病変認識のための注意ベースの対照学習
Deep Pneumonia: Attention-Based Contrastive Learning for Class-Imbalanced Pneumonia Lesion Recognition in Chest X-rays
コンピュータ支援X線肺炎病変の認識は、肺炎の正確な診断にとって重要です。深層学習の出現により、肺炎の識別精度は大幅に向上しましたが、胸部X線のぼやけた外観のためにまだいくつかの課題があります。この論文では、クラス不均衡なX線肺炎病変認識(深部肺炎と呼ばれる)のための注意ベースの対照学習という名前の深層学習フレームワークを提案します。限られた利用可能なデータセットを完全にマイニングするために余分な肺炎データを使用せずにモデルを事前トレーニングするために、自己教師あり対照学習戦略を採用しています。医師が丹念にマークした病変領域の位置情報を活用するために、注意マップと抽出された特徴にそれぞれ適用されるマスク誘導ハードアテンション戦略と特徴学習を提案し、モデルを認識性能を改善するためのより識別可能な特徴を含む病変領域により多くの注意を向ける。さらに、分類の損失関数として、従来のクロスエントロピーの代わりにクラスバランス損失を採用して、データセット内の異なるクラスの肺炎間の深刻なクラスの不均衡の問題に取り組んでいます。実験結果は、提案されたフレームワークが信頼性の高いコンピューター支援肺炎診断システムとして使用でき、医師が肺炎の症例をより正確に診断するのを支援できることを示しています。
Computer-aided X-ray pneumonia lesion recognition is important for accurate diagnosis of pneumonia. With the emergence of deep learning, the identification accuracy of pneumonia has been greatly improved, but there are still some challenges due to the fuzzy appearance of chest X-rays. In this paper, we propose a deep learning framework named Attention-Based Contrastive Learning for Class-Imbalanced X-Ray Pneumonia Lesion Recognition (denoted as Deep Pneumonia). We adopt self-supervised contrastive learning strategy to pre-train the model without using extra pneumonia data for fully mining the limited available dataset. In order to leverage the location information of the lesion area that the doctor has painstakingly marked, we propose mask-guided hard attention strategy and feature learning with contrastive regulation strategy which are applied on the attention map and the extracted features respectively to guide the model to focus more attention on the lesion area where contains more discriminative features for improving the recognition performance. In addition, we adopt Class-Balanced Loss instead of traditional Cross-Entropy as the loss function of classification to tackle the problem of serious class imbalance between different classes of pneumonia in the dataset. The experimental results show that our proposed framework can be used as a reliable computer-aided pneumonia diagnosis system to assist doctors to better diagnose pneumonia cases accurately.
updated: Sat Jul 23 2022 02:28:37 GMT+0000 (UTC)
published: Sat Jul 23 2022 02:28: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アソシエイト