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.