arXiv reaDer
COVID-19診断を改善するための多段階の注意深い転移学習フレームワーク
A Multi-Stage Attentive Transfer Learning Framework for Improving COVID-19 Diagnosis
コンピュータ断層撮影(CT)イメージングは、COVID-19を診断するための有望なアプローチです。機械学習手法を使用して、ラベル付けされたCT画像からモデルをトレーニングし、症例が陽性か陰性かを予測できます。ただし、正確なモデルをトレーニングするための公開されている大規模なCTデータはありません。この作業では、COVID-19診断を改善するための多段階の注意深い転移学習フレームワークを提案します。私たちが提案するフレームワークは、複数のソースタスクと異なるドメインのデータから知識を学習することで正確な診断モデルをトレーニングするための3つの段階で構成されています。重要なのは、肺CT画像のマルチスケール表現を学習するための新しい自己教師あり学習方法を提案することです。私たちの方法は、肺全体から意味情報をキャプチャし、より良い表現学習のために各肺領域の機能を強調します。次に、この方法は、提案された転移学習フレームワークの最終段階に統合され、同じCT画像から学習された複雑なパターンを再利用します。自己注意(ATTN)と畳み込み演算を統合する基本モデルを使用します。実験結果は、ATTNを備えたネットワークは、ATTNを備えていないネットワークよりも転送学習を通じてパフォーマンスの向上をもたらすことを示しています。これは、注意が畳み込みよりも高い転送可能性を示すことを示しています。我々の結果はまた、提案された自己教師あり学習法がいくつかのベースライン法よりも優れていることを示しています。
Computed tomography (CT) imaging is a promising approach to diagnosing the COVID-19. Machine learning methods can be employed to train models from labeled CT images and predict whether a case is positive or negative. However, there exists no publicly-available and large-scale CT data to train accurate models. In this work, we propose a multi-stage attentive transfer learning framework for improving COVID-19 diagnosis. Our proposed framework consists of three stages to train accurate diagnosis models through learning knowledge from multiple source tasks and data of different domains. Importantly, we propose a novel self-supervised learning method to learn multi-scale representations for lung CT images. Our method captures semantic information from the whole lung and highlights the functionality of each lung region for better representation learning. The method is then integrated to the last stage of the proposed transfer learning framework to reuse the complex patterns learned from the same CT images. We use a base model integrating self-attention (ATTNs) and convolutional operations. Experimental results show that networks with ATTNs induce greater performance improvement through transfer learning than networks without ATTNs. This indicates attention exhibits higher transferability than convolution. Our results also show that the proposed self-supervised learning method outperforms several baseline methods.
updated: Thu Jan 14 2021 01:39:19 GMT+0000 (UTC)
published: Thu Jan 14 2021 01:39:19 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アソシエイト