arXiv reaDer
マルチモーダル3DMRI脳腫瘍セグメンテーションと自己監視事前トレーニングのためのCNNとトランスフォーマーの組み合わせ
Combining CNNs With Transformer for Multimodal 3D MRI Brain Tumor Segmentation With Self-Supervised Pretraining
BraTS 2021チャレンジのセグメンテーションタスクには、変更されたTransBTS、nnU-Net、および両方の組み合わせのアンサンブルを適用します。実際、Squeeze-and-Excitationブロックを追加し、CNNレイヤーの数を増やし、Transformerブロックの位置エンコーディングを学習可能な多層パーセプトロン(MLP)埋め込みに置き換えることで、TransBTSモデルの元のアーキテクチャを変更します。これにより、Transformerは任意の場所に調整可能になります。推論中の入力サイズ。これらの変更により、TransBTSのパフォーマンスを大幅に向上させることができます。 nnU-Netフレームワークに触発されて、nnU-Net内のアーキテクチャをカスタムモデルに変更することにより、変更されたTransBTSと組み合わせることにしました。 BraTS 2021の検証セットでは、これらのアプローチのアンサンブルは、腫瘍、腫瘍コア、および腫瘍全体をそれぞれ強化するために、0.8496、0.8698、0.9256ダイススコアおよび15.72、11.057、3.374HD95を達成します。私たちのコードは公開されています。
We apply an ensemble of modified TransBTS, nnU-Net, and a combination of both for the segmentation task of the BraTS 2021 challenge. In fact, we change the original architecture of the TransBTS model by adding Squeeze-and-Excitation blocks, an increasing number of CNN layers, replacing positional encoding in Transformer block with a learnable Multilayer Perceptron (MLP) embeddings, which makes Transformer adjustable to any input size during inference. With these modifications, we are able to largely improve TransBTS performance. Inspired by a nnU-Net framework we decided to combine it with our modified TransBTS by changing the architecture inside nnU-Net to our custom model. On the Validation set of BraTS 2021, the ensemble of these approaches achieves 0.8496, 0.8698, 0.9256 Dice score and 15.72, 11.057, 3.374 HD95 for enhancing tumor, tumor core, and whole tumor, correspondingly. Our code is publicly available.
updated: Fri Oct 15 2021 08:03:27 GMT+0000 (UTC)
published: Fri Oct 15 2021 08:03:27 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アソシエイト