arXiv reaDer
MRI脳腫瘍セグメンテーションのための変分オートエンコーダとアテンションゲートを備えた2段階カスケードモデル
A Two-Stage Cascade Model with Variational Autoencoders and Attention Gates for MRI Brain Tumor Segmentation
自動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.
updated: Sat Nov 28 2020 06:48:15 GMT+0000 (UTC)
published: Wed Nov 04 2020 05:55:06 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アソシエイト