arXiv reaDer
UniMiSS: 次元の壁を打ち破るユニバーサル メディカル セルフスーパーバイズド ラーニング
UniMiSS: Universal Medical Self-Supervised Learning via Breaking Dimensionality Barrier
自己教師あり学習 (SSL) は、注釈がないことでよく知られている医療画像分析に大きなチャンスをもたらします。ただし、コンピューター断層撮影法 (CT) のような大規模な (ラベルのない) 3D 医療画像を集約することは、画像処理のコストが高く、プライバシーの制限があるため、依然として困難です。この論文では、UniMiSS と呼ばれる普遍的な医療自己教師あり表現学習フレームワークを構築することを目的として、3D データの不足を補うために、胸部 X 線のような豊富な 2D 画像をもたらすことを提唱します。次の問題は、次元の壁をどのように壊すか、つまり、2D と 3D の両方の画像で SSL を実行できるようにすることです。これを実現するために、ピラミッド型の U 型の医療用変圧器 (MiT) を設計します。これは、切り替え可能なパッチ埋め込み (SPE) モジュールとトランスフォーマーで構成されています。 SPE モジュールは、入力次元に応じて、2D または 3D パッチ埋め込みに適応的に切り替えます。埋め込まれたパッチは、元のサイズに関係なくシーケンスに変換されます。トランスフォーマーは、長期的な依存関係を連続的にモデル化するため、UniMiSS は 2D 画像と 3D 画像の両方から表現を学習できます。 MiTをバックボーンに、自己蒸留方式でUniMiSSを行っています。セグメンテーションと分類を含む 6 つの 3D/2D 医用画像解析タスクについて、費用のかかる実験を行います。結果は、提案された UniMiSS がさまざまなダウンストリーム タスクで有望なパフォーマンスを達成し、ImageNet の事前トレーニングやその他の高度な SSL 対応物よりも大幅に優れていることを示しています。コードは \rm\small\ttfamily https://github.com/YtongXie/UniMiSS-code で入手できます。
Self-supervised learning (SSL) opens up huge opportunities for medical image analysis that is well known for its lack of annotations. However, aggregating massive (unlabeled) 3D medical images like computerized tomography (CT) remains challenging due to its high imaging cost and privacy restrictions. In this paper, we advocate bringing a wealth of 2D images like chest X-rays as compensation for the lack of 3D data, aiming to build a universal medical self-supervised representation learning framework, called UniMiSS. The following problem is how to break the dimensionality barrier, i.e. , making it possible to perform SSL with both 2D and 3D images? To achieve this, we design a pyramid U-like medical Transformer (MiT). It is composed of the switchable patch embedding (SPE) module and Transformers. The SPE module adaptively switches to either 2D or 3D patch embedding, depending on the input dimension. The embedded patches are converted into a sequence regardless of their original dimensions. The Transformers model the long-term dependencies in a sequence-to-sequence manner, thus enabling UniMiSS to learn representations from both 2D and 3D images. With the MiT as the backbone, we perform the UniMiSS in a self-distillation manner. We conduct expensive experiments on six 3D/2D medical image analysis tasks, including segmentation and classification. The results show that the proposed UniMiSS achieves promising performance on various downstream tasks, outperforming the ImageNet pre-training and other advanced SSL counterparts substantially. Code is available at \rm\small\ttfamily https://github.com/YtongXie/UniMiSS-code.
updated: Tue Aug 16 2022 00:42:25 GMT+0000 (UTC)
published: Fri Dec 17 2021 07:27: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アソシエイト