arXiv reaDer
MR 画像再構成のための特異性保存フェデレーテッド ラーニング
Specificity-Preserving Federated Learning for MR Image Reconstruction
連合学習 (FL) を使用して、ローカル データを集約する必要なく複数の機関が協力できるようにすることで、データのプライバシーと磁気共鳴 (MR) 画像再構成の効率を向上させることができます。ただし、異なる MR イメージング プロトコルによって引き起こされるドメイン シフトは、FL モデルのパフォーマンスを大幅に低下させる可能性があります。最近の FL 技術は、グローバル モデルの一般化を強化することでこれを解決する傾向がありますが、デバイス プロパティに関する重要な情報を含み、ローカル再構築に役立つドメイン固有の機能を無視しています。この論文では、MR 画像再構成 (FedMRI) のための特異性保存 FL アルゴリズムを提案します。中核となるアイデアは、MR 再構成モデルを 2 つの部分に分割することです。グローバル レベルで一般化された表現を取得するためのグローバル共有エンコーダーと、各クライアントのドメイン固有のプロパティを保持するためのクライアント固有のデコーダーです。これは、共同作業にとって重要です。クライアントが一意の分布を持っている場合の再構築。このようなスキームは、周波数空間と画像空間でそれぞれ実行され、異なる空間で一般化された表現とクライアント固有のプロパティを同時に探索できます。さらに、ドメインシフトが存在する場合にグローバル共有エンコーダーの収束をさらに促進するために、最適化中にクライアントとサーバー間の偏差を直接修正するために、重み付けされた対照的な正則化が導入されます。広範な実験は、FedMRI の再構成された結果が多施設データのグラウンド トゥルースに最も近く、最先端の FL 法よりも優れていることを示しています。
Federated learning (FL) can be used to improve data privacy and efficiency in magnetic resonance (MR) image reconstruction by enabling multiple institutions to collaborate without needing to aggregate local data. However, the domain shift caused by different MR imaging protocols can substantially degrade the performance of FL models. Recent FL techniques tend to solve this by enhancing the generalization of the global model, but they ignore the domain-specific features, which may contain important information about the device properties and be useful for local reconstruction. In this paper, we propose a specificity-preserving FL algorithm for MR image reconstruction (FedMRI). The core idea is to divide the MR reconstruction model into two parts: a globally shared encoder to obtain a generalized representation at the global level, and a client-specific decoder to preserve the domain-specific properties of each client, which is important for collaborative reconstruction when the clients have unique distribution. Such scheme is then executed in the frequency space and the image space respectively, allowing exploration of generalized representation and client-specific properties simultaneously in different spaces. Moreover, to further boost the convergence of the globally shared encoder when a domain shift is present, a weighted contrastive regularization is introduced to directly correct any deviation between the client and server during optimization. Extensive experiments demonstrate that our FedMRI's reconstructed results are the closest to the ground-truth for multi-institutional data, and that it outperforms state-of-the-art FL methods.
updated: Mon Aug 22 2022 20:56:02 GMT+0000 (UTC)
published: Thu Dec 09 2021 22:13:35 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アソシエイト