自動MRI脳腫瘍セグメンテーションは、疾患の診断、監視、および治療計画にとって非常に重要です。この論文では、脳腫瘍の小領域セグメンテーションのための2段階のエンコーダ-デコーダベースのモデルを提案します。オーバーフィットの問題を防ぐために、両方の段階で変分オートエンコーダ正則化が利用されます。第2段階のネットワークはアテンションゲートを採用し、第1段階の出力によって形成された拡張データセットを使用してさらにトレーニングされます。 BraTS 2020検証データセットでは、提案された方法により、腫瘍全体、腫瘍コア、および増強腫瘍について、それぞれ0.9041、0.8350、および0.7958の平均ダイススコアと4.953、6.299、および23.608のハウスドルフ距離(95%)が達成されます。 。 BraTS 2020テストデータセットの対応する結果は、ダイススコアで0.8729、0.8357、0.8205、ハウスドルフ距離で11.4288、19.9690、15.6711です。コードはhttps://github.com/shu-hai/two-stage-VAE-Attention-gate-BraTS2020で公開されています。
Automatic MRI brain tumor segmentation is of vital importance for the disease diagnosis, monitoring, and treatment planning. In this paper, we propose a two-stage encoder-decoder based model for brain tumor subregional segmentation. Variational autoencoder regularization is utilized in both stages to prevent the overfitting issue. The second-stage network adopts attention gates and is trained additionally using an expanded dataset formed by the first-stage outputs. On the BraTS 2020 validation dataset, the proposed method achieves the mean Dice score of 0.9041, 0.8350, and 0.7958, and Hausdorff distance (95%) of 4.953, 6.299, and 23.608 for the whole tumor, tumor core, and enhancing tumor, respectively. The corresponding results on the BraTS 2020 testing dataset are 0.8729, 0.8357, and 0.8205 for Dice score, and 11.4288, 19.9690, and 15.6711 for Hausdorff distance. The code is publicly available at https://github.com/shu-hai/two-stage-VAE-Attention-gate-BraTS2020.