arXiv reaDer
3D医療画像分析のためのSwinTransformersの自己監視事前トレーニング
Self-Supervised Pre-Training of Swin Transformers for 3D Medical Image Analysis
Vision Transformers(ViT)は、ダウンストリームアプリケーションに転送できるグローバルおよびローカル表現の自己監視学習で優れたパフォーマンスを示しています。これらの結果に触発されて、医療画像分析のために調整されたプロキシタスクを備えた新しい自己監視学習フレームワークを紹介します。具体的には、次のことを提案します。(i)Swin UNEt TRansformers(Swin UNETR)と呼ばれる新しい3Dトランスベースのモデルで、自己監視型の事前トレーニング用の階層型エンコーダーを備えています。 (ii)人体解剖学の根底にあるパターンを学習するための調整されたプロキシタスク。さまざまな体の臓器からの5,050の公的に利用可能なコンピューター断層撮影(CT)画像で、提案されたモデルの事前トレーニングが成功したことを示します。私たちのアプローチの有効性は、13の腹部臓器とMedical Segmentation Decathlon(MSD)データセットからのセグメンテーションタスクを使用して、Beyond the Cranial Vault(BTCV)セグメンテーションチャレンジで事前トレーニングされたモデルを微調整することによって検証されます。私たちのモデルは現在、MSDとBTCVの両方のデータセットの公開テストリーダーボードで最先端(つまり1位)です。コード:https://monai.io/research/swin-unetr
Vision Transformers (ViT)s have shown great performance in self-supervised learning of global and local representations that can be transferred to downstream applications. Inspired by these results, we introduce a novel self-supervised learning framework with tailored proxy tasks for medical image analysis. Specifically, we propose: (i) a new 3D transformer-based model, dubbed Swin UNEt TRansformers (Swin UNETR), with a hierarchical encoder for self-supervised pre-training; (ii) tailored proxy tasks for learning the underlying pattern of human anatomy. We demonstrate successful pre-training of the proposed model on 5,050 publicly available computed tomography (CT) images from various body organs. The effectiveness of our approach is validated by fine-tuning the pre-trained models on the Beyond the Cranial Vault (BTCV) Segmentation Challenge with 13 abdominal organs and segmentation tasks from the Medical Segmentation Decathlon (MSD) dataset. Our model is currently the state-of-the-art (i.e. ranked 1st) on the public test leaderboards of both MSD and BTCV datasets. Code: https://monai.io/research/swin-unetr
updated: Mon Nov 29 2021 18:45:20 GMT+0000 (UTC)
published: Mon Nov 29 2021 18:45:20 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アソシエイト