arXiv reaDer
HCDG:医療画像セグメンテーションのドメイン一般化のための階層的一貫性フレームワーク
HCDG: A Hierarchical Consistency Framework for Domain Generalization on Medical Image Segmentation
現代のディープニューラルネットワークは、実際のアプリケーションに展開するときに、知識を伝達し、ドメイン間で一般化するのに苦労しています。ドメイン一般化(DG)は、複数のソースドメインからユニバーサル表現を学習して、見えないターゲットドメインでのネットワーク一般化機能を向上させることを目的としています。以前のDG手法は、さまざまな整合性スキームの相乗的な正則化を考慮せずに、データレベルの整合性スキームに主に焦点を当ててディープネットワークの一般化機能を向上させていました。この論文では、外因性一貫性と内因性一貫性をアンサンブルすることにより、ドメイン一般化(HCDG)のための新しい階層的一貫性フレームワークを提示します。特に、外部整合性については、複数のソースドメインにわたる知識を活用して、データレベルの整合性を強化します。また、このような一貫性を強化するために、フーリエベースのデータ拡張のための新しい振幅ガウス混合戦略を設計します。本質的一貫性については、デュアルタスク形式で同じインスタンスに対してタスクレベルの一貫性を実行します。提案されたHCDGフレームワークを、2つの医療画像セグメンテーションタスク、つまり眼底画像での光学カップ/ディスクセグメンテーションと前立腺MRIセグメンテーションで評価します。広範な実験結果は、HCDGフレームワークの有効性と多様性を示しています。コードは、承認されると利用可能になります。
Modern deep neural networks struggle to transfer knowledge and generalize across domains when deploying to real-world applications. Domain generalization (DG) aims to learn a universal representation from multiple source domains to improve the network generalization ability on unseen target domains. Previous DG methods mostly focus on the data-level consistency scheme to advance the generalization capability of deep networks, without considering the synergistic regularization of different consistency schemes. In this paper, we present a novel Hierarchical Consistency framework for Domain Generalization (HCDG) by ensembling Extrinsic Consistency and Intrinsic Consistency. Particularly, for Extrinsic Consistency, we leverage the knowledge across multiple source domains to enforce data-level consistency. Also, we design a novel Amplitude Gaussian-mixing strategy for Fourier-based data augmentation to enhance such consistency. For Intrinsic Consistency, we perform task-level consistency for the same instance under the dual-task form. We evaluate the proposed HCDG framework on two medical image segmentation tasks, i.e., optic cup/disc segmentation on fundus images and prostate MRI segmentation. Extensive experimental results manifest the effectiveness and versatility of our HCDG framework. Code will be available once accept.
updated: Mon Sep 13 2021 07:07:23 GMT+0000 (UTC)
published: Mon Sep 13 2021 07:07:23 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アソシエイト