教師あり学習方法には、大量の注釈付きデータセットが必要です。このようなデータセットの収集には、時間と費用がかかります。これまで、注釈付きのCOVID-19画像データセットはほとんどありません。自己教師あり学習では、ラベルのないデータを活用してトレーニングをブートストラップできますが、自然画像の一般的な自己教師あり方法では、コンテキストが十分に組み込まれていません。医用画像の場合、望ましい方法は、各解剖学的領域の正常に見える組織からの逸脱を検出するのに十分な感度が必要です。ここでは、解剖学がコンテキストです。 2つのレベルの自己教師あり表現学習目標を備えた新しいアプローチを紹介します。1つは地域の解剖学的レベルで、もう1つは患者レベルです。グラフニューラルネットワークを使用して、異なる解剖学的領域間の関係を組み込みます。グラフの構造は、各患者と解剖学的アトラスの間の解剖学的対応によって通知されます。さらに、グラフ表現には、任意のサイズの画像をフル解像度で処理できるという利点があります。肺画像の大規模なコンピューター断層撮影(CT)データセットでの実験は、私たちのアプローチがコンテキストを考慮しないベースライン方法と比べて遜色がないことを示しています。学習した埋め込みを使用してCOVID-19の臨床的進行を定量化し、私たちの方法がさまざまな病院のCOVID-19患者によく一般化することを示します。定性的な結果は、私たちのモデルが画像内の臨床的に関連する領域を識別できることを示唆しています。
Supervised learning method requires a large volume of annotated datasets. Collecting such datasets is time-consuming and expensive. Until now, very few annotated COVID-19 imaging datasets are available. Although self-supervised learning enables us to bootstrap the training by exploiting unlabeled data, the generic self-supervised methods for natural images do not sufficiently incorporate the context. For medical images, a desirable method should be sensitive enough to detect deviation from normal-appearing tissue of each anatomical region; here, anatomy is the context. We introduce a novel approach with two levels of self-supervised representation learning objectives: one on the regional anatomical level and another on the patient-level. We use graph neural networks to incorporate the relationship between different anatomical regions. The structure of the graph is informed by anatomical correspondences between each patient and an anatomical atlas. In addition, the graph representation has the advantage of handling any arbitrarily sized image in full resolution. Experiments on large-scale Computer Tomography (CT) datasets of lung images show that our approach compares favorably to baseline methods that do not account for the context. We use the learnt embedding to quantify the clinical progression of COVID-19 and show that our method generalizes well to COVID-19 patients from different hospitals. Qualitative results suggest that our model can identify clinically relevant regions in the images.