arXiv reaDer
BraTS 2021チャレンジ用の一般化されたワッサースタインダイスロス、テスト時間増強、および変圧器
Generalized Wasserstein Dice Loss, Test-time Augmentation, and Transformers for the BraTS 2021 challenge
複数の磁気共鳴画像法(MRI)モダリティからの脳腫瘍のセグメンテーションは、医用画像計算における困難な作業です。主な課題は、さまざまなスキャナーとイメージングプロトコルへの一般化可能性にあります。この論文では、推論時間を増やすことなくモデルのロバスト性を高めるための戦略を探ります。この目的に向けて、さまざまな損失、オプティマイザー、およびトレイン検証データ分割を使用してトレーニングされたモデルから、堅牢なアンサンブルを見つけることを検討します。重要なのは、U-Netアーキテクチャのボトルネックに変圧器を含めることを検討することです。ボトルネックの変圧器のパフォーマンスは平均してベースラインのU-Netよりもわずかに悪いことがわかりますが、一般化されたWassersteinDiceの損失は一貫して優れた結果をもたらします。さらに、より高速で堅牢な推論のために、効率的なテスト時間拡張戦略を採用しています。テスト時間の拡張を伴う7つの3DU-Netの最終的なアンサンブルは、BraTS 2021テストデータセットで評価した場合、平均ダイススコア89.4%、平均ハウスドルフ95%距離10.0mmを生成します。コードとトレーニング済みモデルは、https://github.com/LucasFidon/TRABIT_BraTS2021で公開されています。
Brain tumor segmentation from multiple Magnetic Resonance Imaging (MRI) modalities is a challenging task in medical image computation. The main challenges lie in the generalizability to a variety of scanners and imaging protocols. In this paper, we explore strategies to increase model robustness without increasing inference time. Towards this aim, we explore finding a robust ensemble from models trained using different losses, optimizers, and train-validation data split. Importantly, we explore the inclusion of a transformer in the bottleneck of the U-Net architecture. While we find transformer in the bottleneck performs slightly worse than the baseline U-Net in average, the generalized Wasserstein Dice loss consistently produces superior results. Further, we adopt an efficient test time augmentation strategy for faster and robust inference. Our final ensemble of seven 3D U-Nets with test-time augmentation produces an average dice score of 89.4% and an average Hausdorff 95% distance of 10.0 mm when evaluated on the BraTS 2021 testing dataset. Our code and trained models are publicly available at https://github.com/LucasFidon/TRABIT_BraTS2021.
updated: Fri Dec 24 2021 13:01:44 GMT+0000 (UTC)
published: Fri Dec 24 2021 13:01:44 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アソシエイト