この論文では、体積医療画像セグメンテーションのためのTransformerアーキテクチャを紹介します。ボリュームセグメンテーションのための計算効率の高いTransformerアーキテクチャを設計することは、困難な作業です。ローカルおよびグローバルな空間キューをエンコードする際に複雑なバランスを保ち、ボリュームデータのすべての軸に沿って情報を保持する必要があります。提案された体積変換器は、入力ボクセル全体を処理するU字型のエンコーダ-デコーダ設計を備えています。私たちのエンコーダーには、ローカルキューとグローバルキューを同時にエンコードするための2つの連続した自己注意レイヤーがあり、デコーダーには、フーリエ位置エンコードを含めることによって境界の詳細をキャプチャするための新しい並列シフトウィンドウベースの自己およびクロスアテンションブロックがあります。私たちが提案する設計の選択により、計算効率の高いアーキテクチャが実現します。これは、脳腫瘍セグメンテーション(BraTS)2021、および腫瘍セグメンテーションの医療セグメンテーション十種競技(膵臓および肝臓)データセットで有望な結果を示しています。さらに、モデルによって学習された表現がデータセット間でより適切に転送され、データの破損に対して堅牢であることを示します。 https://github.com/himashi92/VT-UNetOurコードの実装は公開されています。
This paper presents a Transformer architecture for volumetric medical image segmentation. Designing a computationally efficient Transformer architecture for volumetric segmentation is a challenging task. It requires keeping a complex balance in encoding local and global spatial cues, and preserving information along all axes of the volumetric data. The proposed volumetric Transformer has a U-shaped encoder-decoder design that processes the input voxels in their entirety. Our encoder has two consecutive self-attention layers to simultaneously encode local and global cues, and our decoder has novel parallel shifted window based self and cross attention blocks to capture fine details for boundary refinement by subsuming Fourier position encoding. Our proposed design choices result in a computationally efficient architecture, which demonstrates promising results on Brain Tumor Segmentation (BraTS) 2021, and Medical Segmentation Decathlon (Pancreas and Liver) datasets for tumor segmentation. We further show that the representations learned by our model transfer better across-datasets and are robust against data corruptions. https://github.com/himashi92/VT-UNetOur code implementation is publicly available.