arXiv reaDer
医療画像セグメンテーションのための因果関係に触発された単一ソースドメインの一般化
Causality-inspired Single-source Domain Generalization for Medical Image Segmentation
ディープラーニングモデルは通常、ドメインシフトの問題に悩まされており、1つのソースドメインでトレーニングされたモデルは、他の見えないドメインにうまく一般化されません。この作業では、単一ソースドメインの一般化の問題を調査します。トレーニングデータは、医用画像アプリケーションで一般的な1つのソースドメインからのみ利用可能であるという条件の下で、見えないドメインに対して堅牢なディープネットワークをトレーニングします。クロスドメイン医療画像セグメンテーションのコンテキストでこの問題に取り組みます。このシナリオでは、ドメインのシフトは主にさまざまな取得プロセスによって引き起こされます。セグメンテーションモデルを合成されたドメインシフトトレーニングの例に公開するための、単純な因果関係に触発されたデータ拡張アプローチを提案します。具体的には、1)画像の強度とテクスチャの不一致に対して深いモデルを堅牢にするために、ランダムに重み付けされた浅いネットワークのファミリを採用しています。それらは、多様な外観変換を使用してトレーニング画像を補強します。 2)さらに、画像内のオブジェクト間の疑似相関がドメインの堅牢性に悪影響を与えることを示します。これらの相関関係は、予測を行うためのドメイン固有の手がかりとしてネットワークによって取得される可能性があり、見えないドメインで壊れることがあります。因果的介入を介してこれらの疑似相関を削除します。これは、潜在的に相関するオブジェクトの外観を個別にリサンプリングすることによって実現されます。提案されたアプローチは、クロスモダリティ(CT-MRI)腹部画像セグメンテーション、クロスシーケンス(bSSFP-LGE)心臓MRIセグメンテーション、およびクロスセンター前立腺MRIセグメンテーションの3つのクロスドメインセグメンテーションタスクで検証されます。提案されたアプローチは、目に見えないドメインでテストした場合、競合する方法と比較して一貫したパフォーマンスの向上をもたらします。
Deep learning models usually suffer from domain shift issues, where models trained on one source domain do not generalize well to other unseen domains. In this work, we investigate the single-source domain generalization problem: training a deep network that is robust to unseen domains, under the condition that training data is only available from one source domain, which is common in medical imaging applications. We tackle this problem in the context of cross-domain medical image segmentation. Under this scenario, domain shifts are mainly caused by different acquisition processes. We propose a simple causality-inspired data augmentation approach to expose a segmentation model to synthesized domain-shifted training examples. Specifically, 1) to make the deep model robust to discrepancies in image intensities and textures, we employ a family of randomly-weighted shallow networks. They augment training images using diverse appearance transformations. 2) Further we show that spurious correlations among objects in an image are detrimental to domain robustness. These correlations might be taken by the network as domain-specific clues for making predictions, and they may break on unseen domains. We remove these spurious correlations via causal intervention. This is achieved by resampling the appearances of potentially correlated objects independently. The proposed approach is validated on three cross-domain segmentation tasks: cross-modality (CT-MRI) abdominal image segmentation, cross-sequence (bSSFP-LGE) cardiac MRI segmentation, and cross-center prostate MRI segmentation. The proposed approach yields consistent performance gains compared with competitive methods when tested on unseen domains.
updated: Wed Dec 01 2021 15:42:30 GMT+0000 (UTC)
published: Wed Nov 24 2021 14:45:17 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アソシエイト