arXiv reaDer
FedDG:連続周波数空間でのエピソード学習による医療画像セグメンテーションのフェデレーションドメインの一般化
FedDG: Federated Domain Generalization on Medical Image Segmentation via Episodic Learning in Continuous Frequency Space
統合学習により、分散型医療機関はプライバシー保護を備えた共有予測モデルを共同で学習できます。臨床展開中、フェデレーション学習でトレーニングされたモデルは、フェデレーション外の完全に見えない病院に適用された場合でも、パフォーマンスの低下に悩まされる可能性があります。この論文では、複数の分散ソースドメインからフェデレーションモデルを学習して、見えないターゲットドメインに直接一般化できるようにすることを目的とした、フェデレーションドメイン一般化(FedDG)の新しい問題設定を指摘して解決します。データ分散化の困難な制約の下で各クライアントがマルチソースデータ分散を活用できるようにすることで、この問題に対するエピソード学習(ELCFS)という名前の新しいアプローチを紹介します。私たちのアプローチは、効果的な連続周波数空間補間メカニズムを通じて、プライバシーを保護する方法でクライアント間で配信情報を送信します。転送されたマルチソース分布を使用して、境界指向のエピソード学習パラダイムをさらに注意深く設計し、ローカル学習をドメイン分布シフトにさらし、特に医療画像セグメンテーションシナリオでのモデル一般化の課題に対応します。私たちの方法の有効性は、2つの医療画像セグメンテーションタスクでの最先端の詳細なアブレーション実験よりも優れたパフォーマンスで実証されています。コードは「https://github.com/liuquande/FedDG-ELCFS」で入手できます。
Federated learning allows distributed medical institutions to collaboratively learn a shared prediction model with privacy protection. While at clinical deployment, the models trained in federated learning can still suffer from performance drop when applied to completely unseen hospitals outside the federation. In this paper, we point out and solve a novel problem setting of federated domain generalization (FedDG), which aims to learn a federated model from multiple distributed source domains such that it can directly generalize to unseen target domains. We present a novel approach, named as Episodic Learning in Continuous Frequency Space (ELCFS), for this problem by enabling each client to exploit multi-source data distributions under the challenging constraint of data decentralization. Our approach transmits the distribution information across clients in a privacy-protecting way through an effective continuous frequency space interpolation mechanism. With the transferred multi-source distributions, we further carefully design a boundary-oriented episodic learning paradigm to expose the local learning to domain distribution shifts and particularly meet the challenges of model generalization in medical image segmentation scenario. The effectiveness of our method is demonstrated with superior performance over state-of-the-arts and in-depth ablation experiments on two medical image segmentation tasks. The code is available at "https://github.com/liuquande/FedDG-ELCFS".
updated: Wed Mar 10 2021 13:05:23 GMT+0000 (UTC)
published: Wed Mar 10 2021 13:05: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アソシエイト