arXiv reaDer
Auto-FedAvg:多施設医療画像セグメンテーションのための学習可能なフェデレーション平均
Auto-FedAvg: Learnable Federated Averaging for Multi-Institutional Medical Image Segmentation
フェデレーテッドラーニング(FL)は、各参加者のプライバシーを保護しながら、協調的なモデルトレーニングを可能にします。これは、医療分野にとって特に有益です。 FedAvgは、固定の重みを使用する標準アルゴリズムであり、多くの場合、各クライアントのデータセットサイズに由来し、FLプロセス中にサーバー上の分散学習モデルを集約します。ただし、FLの非iid問題として知られる、クライアント間でのデータの分散が同一でない場合、固定集計の重みを設定するためのこの仮定が最適ではなくなる可能性があります。この作業では、新しいデータ駆動型アプローチ、つまりAuto-FedAvgを設計します。このアプローチでは、データサイロ全体のデータ分布とモデルの現在のトレーニングの進捗状況に応じて、集計の重みが動的に調整されます。パラメータセットをローカルモデルパラメータとグローバル集約パラメータの2つの部分に解きほぐし、通信効率の高いアルゴリズムで繰り返し更新します。最初に、CIFAR-10の異種データ分割での画像認識のための最先端のFLメソッドを上回ることにより、アプローチの有効性を示します。さらに、2つの多施設医療画像分析タスク、すなわち胸部CTでのCOVID-19病変セグメンテーションと腹部CTでの膵臓セグメンテーションに対するアルゴリズムの有効性を示します。
Federated learning (FL) enables collaborative model training while preserving each participant's privacy, which is particularly beneficial to the medical field. FedAvg is a standard algorithm that uses fixed weights, often originating from the dataset sizes at each client, to aggregate the distributed learned models on a server during the FL process. However, non-identical data distribution across clients, known as the non-i.i.d problem in FL, could make this assumption for setting fixed aggregation weights sub-optimal. In this work, we design a new data-driven approach, namely Auto-FedAvg, where aggregation weights are dynamically adjusted, depending on data distributions across data silos and the current training progress of the models. We disentangle the parameter set into two parts, local model parameters and global aggregation parameters, and update them iteratively with a communication-efficient algorithm. We first show the validity of our approach by outperforming state-of-the-art FL methods for image recognition on a heterogeneous data split of CIFAR-10. Furthermore, we demonstrate our algorithm's effectiveness on two multi-institutional medical image analysis tasks, i.e., COVID-19 lesion segmentation in chest CT and pancreas segmentation in abdominal CT.
updated: Tue Apr 20 2021 18:29:44 GMT+0000 (UTC)
published: Tue Apr 20 2021 18:29:44 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アソシエイト