arXiv reaDer
スパイラル対比学習: 注釈のない CT 病変の効率的な 3D 表現学習方法
Spiral Contrastive Learning: An Efficient 3D Representation Learning Method for Unannotated CT Lesions
病理学的注釈を含むコンピューター断層撮影 (CT) サンプルは入手が困難です。その結果、コンピューター支援診断 (CAD) アルゴリズムは小さなデータセット (1,018 サンプルの LIDC-IDRI など) でトレーニングされ、精度と信頼性が制限されます。過去 5 年間で、2 次元 (2D) および 3 次元 (3D) の自己教師あり学習 (SSL) アルゴリズムを介して、CT 病変の教師なし表現に合わせたいくつかの研究が行われました。 2D アルゴリズムは 3D 情報をキャプチャするのが難しく、既存の 3D アルゴリズムは計算量が多くなります。軽量の 3D SSL は、引き続き探求すべき境界です。この論文では、計算効率の高い方法で 3D 表現を生成するスパイラル コントラスト学習 (SCL) を提案します。 SCL は、最初に情報保存スパイラル変換を使用して 3D 病変を 2D 平面に変換し、次に 2D 対照学習を使用して変換不変の特徴を学習します。拡張については、自然画像拡張と医療画像拡張を検討します。埋め込みレイヤーで分類ヘッドをトレーニングすることにより、SCL を評価します。実験結果は、SCL が教師なし表現学習の LIDC-IDRI (89.72%)、LNDb (82.09%)、および TianChi (90.16%) で最先端の精度を達成することを示しています。微調整用の 10% の注釈付きデータを使用すると、SCL のパフォーマンスは、教師あり学習アルゴリズムのパフォーマンスに匹敵します (LIDC-IDRI で 85.75% 対 85.03%、LNDb で 78.20% 対 73.44%、LNDb で 87.85% 対 83.34%)。 TianChi、それぞれ)。一方、SCL は、他の 3D SSL アルゴリズムと比較して計算量を 66.98% 削減し、教師なしの事前トレーニングにおける提案された方法の効率を示しています。
Computed tomography (CT) samples with pathological annotations are difficult to obtain. As a result, the computer-aided diagnosis (CAD) algorithms are trained on small datasets (e.g., LIDC-IDRI with 1,018 samples), limiting their accuracies and reliability. In the past five years, several works have tailored for unsupervised representations of CT lesions via two-dimensional (2D) and three-dimensional (3D) self-supervised learning (SSL) algorithms. The 2D algorithms have difficulty capturing 3D information, and existing 3D algorithms are computationally heavy. Light-weight 3D SSL remains the boundary to explore. In this paper, we propose the spiral contrastive learning (SCL), which yields 3D representations in a computationally efficient manner. SCL first transforms 3D lesions to the 2D plane using an information-preserving spiral transformation, and then learn transformation-invariant features using 2D contrastive learning. For the augmentation, we consider natural image augmentations and medical image augmentations. We evaluate SCL by training a classification head upon the embedding layer. Experimental results show that SCL achieves state-of-the-art accuracy on LIDC-IDRI (89.72%), LNDb (82.09%) and TianChi (90.16%) for unsupervised representation learning. With 10% annotated data for fine-tune, the performance of SCL is comparable to that of supervised learning algorithms (85.75% vs. 85.03% on LIDC-IDRI, 78.20% vs. 73.44% on LNDb and 87.85% vs. 83.34% on TianChi, respectively). Meanwhile, SCL reduces the computational effort by 66.98% compared to other 3D SSL algorithms, demonstrating the efficiency of the proposed method in unsupervised pre-training.
updated: Tue Aug 23 2022 02:31:03 GMT+0000 (UTC)
published: Tue Aug 23 2022 02:31:03 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アソシエイト