arXiv reaDer
それらすべてを統合するための1つのモデル:マルチコントラストMRI合成のパーソナライズされた連合学習
One Model to Unite Them All: Personalized Federated Learning of Multi-Contrast MRI Synthesis
学習ベースのMRI変換には、ソースコントラストをターゲットコントラスト画像にマッピングする合成モデルが含まれます。複数の機関によるコラボレーションは、幅広いデータセットにわたる合成モデルのトレーニングの鍵ですが、集中型のトレーニングにはプライバシーリスクが伴います。連合学習(FL)は、画像データの共有を回避し、プライバシーの懸念を軽減するために、代わりに分散型トレーニングを採用するコラボレーションフレームワークです。ただし、FLでトレーニングされたモデルは、画像データの分布に固有の不均一性によって損なわれる可能性があります。一方では、ソースとターゲットの構成が固定された一般的な翻訳タスクの場合でも、サイト間で画像分布の暗黙的なシフトが明らかになります。逆に、ソースとターゲットの構成が異なるさまざまな翻訳タスクが規定されている場合、サイト内およびサイト間で明示的なシフトが発生します。ドメインシフトに対する信頼性を向上させるために、ここではMRI合成(pFLSynth)用の最初のパーソナライズされたFLメソッドを紹介します。 pFLSynthは、個々のサイトとソースとターゲットのコントラストに固有の潜在性を生成するマッパーを備えた敵対モデルに基づいています。これは、これらの潜在性に基づいてジェネレータ全体の機能マップの統計と重み付けを適応的に調整する新しいパーソナライズブロックを活用します。サイトの特異性をさらに高めるために、ジェネレーターのダウンストリームレイヤーで部分的なモデル集約が採用され、アップストリームレイヤーはローカルに保持されます。そのため、pFLSynthを使用すると、複数のサイトや翻訳タスクにまたがって確実に一般化できる統合合成モデルのトレーニングが可能になります。マルチサイトデータセットでの包括的な実験は、マルチコントラストMRI合成における以前のフェデレーションメソッドに対するpFLSynthのパフォーマンスの向上を明確に示しています。
Learning-based MRI translation involves a synthesis model that maps a source-contrast onto a target-contrast image. Multi-institutional collaborations are key to training synthesis models across broad datasets, yet centralized training involves privacy risks. Federated learning (FL) is a collaboration framework that instead adopts decentralized training to avoid sharing imaging data and mitigate privacy concerns. However, FL-trained models can be impaired by the inherent heterogeneity in the distribution of imaging data. On the one hand, implicit shifts in image distribution are evident across sites, even for a common translation task with fixed source-target configuration. Conversely, explicit shifts arise within and across sites when diverse translation tasks with varying source-target configurations are prescribed. To improve reliability against domain shifts, here we introduce the first personalized FL method for MRI Synthesis (pFLSynth). pFLSynth is based on an adversarial model equipped with a mapper that produces latents specific to individual sites and source-target contrasts. It leverages novel personalization blocks that adaptively tune the statistics and weighting of feature maps across the generator based on these latents. To further promote site-specificity, partial model aggregation is employed over downstream layers of the generator while upstream layers are retained locally. As such, pFLSynth enables training of a unified synthesis model that can reliably generalize across multiple sites and translation tasks. Comprehensive experiments on multi-site datasets clearly demonstrate the enhanced performance of pFLSynth against prior federated methods in multi-contrast MRI synthesis.
updated: Wed Jul 13 2022 20:14:16 GMT+0000 (UTC)
published: Wed Jul 13 2022 20:14:16 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アソシエイト