arXiv reaDer
ローカルキャリブレーションによるフェデレーション医療画像セグメンテーションのパーソナライズ
Personalizing Federated Medical Image Segmentation via Local Calibration
連合学習(FL)の下での医療画像のセグメンテーションは、データセットを一元化することなく、複数の臨床サイトがグローバルモデルを共同で学習できるようにすることで有望な方向性です。ただし、単一のモデルを使用してさまざまなサイトからのさまざまなデータ配布に適応することは非常に困難です。パーソナライズされたFLは、グローバルサーバーから共有される部分的なモデルパラメーターのみを利用することでこの問題に取り組み、残りは各サイトのローカルトレーニングで独自のデータ分散に適応するようにします。ただし、ほとんどの既存の方法は部分的なパラメータ分割に集中していますが、ローカルトレーニング中のサイト間の不整合は考慮されていません。これにより、サイトを介した知識の伝達が容易になり、ローカルの精度を向上させるためのモデル学習に役立ちます。このホワイトペーパーでは、ローカルキャリブレーション(LC-Fed)を使用したパーソナライズされたフェデレーションフレームワークを提案し、機能レベルと予測レベルの両方でサイト間の不整合を活用してセグメンテーションを強化します。具体的には、各ローカルサイトがさまざまな機能に代替の注意を払っているため、最初に、エンコードされた機能を調整するためのチャネル選択操作と組み合わせた対照的なサイト埋め込みを設計します。さらに、予測レベルの不整合の知識を活用して、解剖学的境界などのあいまいな領域でのパーソナライズされたモデリングをガイドすることを提案します。これは、不一致を認識したマップを計算して予測を調整することによって実現されます。私たちの方法の有効性は、異なるモダリティを持つ3つの医療画像セグメンテーションタスクで検証されており、私たちの方法は、最先端のパーソナライズされたFL方法よりも優れたパフォーマンスを一貫して示しています。コードはhttps://github.com/jcwang123/FedLCで入手できます。
Medical image segmentation under federated learning (FL) is a promising direction by allowing multiple clinical sites to collaboratively learn a global model without centralizing datasets. However, using a single model to adapt to various data distributions from different sites is extremely challenging. Personalized FL tackles this issue by only utilizing partial model parameters shared from global server, while keeping the rest to adapt to its own data distribution in the local training of each site. However, most existing methods concentrate on the partial parameter splitting, while do not consider the inter-site in-consistencies during the local training, which in fact can facilitate the knowledge communication over sites to benefit the model learning for improving the local accuracy. In this paper, we propose a personalized federated framework with Local Calibration (LC-Fed), to leverage the inter-site in-consistencies in both feature- and prediction- levels to boost the segmentation. Concretely, as each local site has its alternative attention on the various features, we first design the contrastive site embedding coupled with channel selection operation to calibrate the encoded features. Moreover, we propose to exploit the knowledge of prediction-level in-consistency to guide the personalized modeling on the ambiguous regions, e.g., anatomical boundaries. It is achieved by computing a disagreement-aware map to calibrate the prediction. Effectiveness of our method has been verified on three medical image segmentation tasks with different modalities, where our method consistently shows superior performance to the state-of-the-art personalized FL methods. Code is available at https://github.com/jcwang123/FedLC.
updated: Mon Jul 11 2022 06:30:31 GMT+0000 (UTC)
published: Mon Jul 11 2022 06:30:31 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アソシエイト