arXiv reaDer
堅牢な医用画像セグメンテーションのためのカリキュラムベースの拡張フーリエ ドメイン適応
Curriculum-Based Augmented Fourier Domain Adaptation for Robust Medical Image Segmentation
正確かつ堅牢な医療画像セグメンテーションは、コンピュータ支援診断および介入システムの自律性を強化するために基本的かつ重要です。医療データの収集には通常、さまざまなスキャナー、プロトコル、母集団が含まれるため、ドメイン アダプテーション (DA) は、展開サイトでのモデルの劣化を軽減するための非常に要求の高い研究分野となっています。複数のテスト ドメインにわたってモデルのパフォーマンスを維持するために、この研究では、堅牢な医療画像セグメンテーションのためのカリキュラムベースの拡張フーリエ ドメイン適応 (Curri-AFDA) を提案しています。特に、私たちのカリキュラム学習戦略は、展開フェーズにおけるさまざまなレベルのデータシフトにおけるモデルの因果関係に基づいており、シフトが大きくなるほど、差異を認識するのが難しくなります。これを考慮して、カリキュラム形式のトレーニング中に周波数空間でターゲット領域からソース領域へより多くの振幅情報を徐々に導入し、より簡単からより困難に意味論的な知識の伝達をスムーズにスケジュールします。さらに、トレーニング時の連鎖拡張ミキシングを組み込んで、ドメイン不変のセマンティクスを維持しながらデータ分布を拡張します。これは、取得されたモデルがより堅牢になり、目に見えないドメインに対してより適切に一般化するのに有益です。複数の部位およびスキャナーから収集された網膜と核の 2 つのセグメンテーション タスクに関する広範な実験により、提案された方法が優れた適応および一般化パフォーマンスを生み出すことが示唆されています。一方、私たちのアプローチは、さまざまな種類の破損や増加する重大度レベルの下でより堅牢であることが証明されています。さらに、私たちの方法が皮膚病変データセットを使用したドメイン適応型分類タスクにも有益であることを示します。コードは https://github.com/lofrienger/Curri-AFDA で入手できます。
Accurate and robust medical image segmentation is fundamental and crucial for enhancing the autonomy of computer-aided diagnosis and intervention systems. Medical data collection normally involves different scanners, protocols, and populations, making domain adaptation (DA) a highly demanding research field to alleviate model degradation in the deployment site. To preserve the model performance across multiple testing domains, this work proposes the Curriculum-based Augmented Fourier Domain Adaptation (Curri-AFDA) for robust medical image segmentation. In particular, our curriculum learning strategy is based on the causal relationship of a model under different levels of data shift in the deployment phase, where the higher the shift is, the harder to recognize the variance. Considering this, we progressively introduce more amplitude information from the target domain to the source domain in the frequency space during the curriculum-style training to smoothly schedule the semantic knowledge transfer in an easier-to-harder manner. Besides, we incorporate the training-time chained augmentation mixing to help expand the data distributions while preserving the domain-invariant semantics, which is beneficial for the acquired model to be more robust and generalize better to unseen domains. Extensive experiments on two segmentation tasks of Retina and Nuclei collected from multiple sites and scanners suggest that our proposed method yields superior adaptation and generalization performance. Meanwhile, our approach proves to be more robust under various corruption types and increasing severity levels. In addition, we show our method is also beneficial in the domain-adaptive classification task with skin lesion datasets. The code is available at https://github.com/lofrienger/Curri-AFDA.
updated: Tue Jun 06 2023 08:56:58 GMT+0000 (UTC)
published: Tue Jun 06 2023 08:56:58 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アソシエイト