arXiv reaDer
ヘルスケアのための適応バッチノルムによるパーソナライズされた連合学習
Personalized Federated Learning with Adaptive Batchnorm for Healthcare
機械学習技術をヘルスケアに適用することに関心が高まっています。最近、連合学習(FL)は、研究者がデータのプライバシーとセキュリティを損なうことなく強力なモデルをトレーニングできるため、人気が高まっています。ただし、既存のFLアプローチのパフォーマンスは、クライアント間に分布のギャップが存在する非iidの状況に遭遇すると低下することが多く、これまでの取り組みのほとんどは、ヘルスケアのパーソナライズに焦点を当てていませんでした。この記事では、ドメインシフトに取り組み、ローカルクライアント向けにパーソナライズされたモデルを取得するためのFedAPを提案します。 FedAPは、バッチ正規化レイヤーの統計に基づいてクライアント間の類似性を学習し、ローカルバッチ正規化が異なる各クライアントの特異性を維持します。 5つのヘルスケアベンチマークに関する包括的な実験は、FedAPが最先端の方法(たとえば、PAMAP2の10%の精度向上)と比較して、より速い収束速度でより優れた精度を達成することを示しています。
There is a growing interest in applying machine learning techniques to healthcare. Recently, federated learning (FL) is gaining popularity since it allows researchers to train powerful models without compromising data privacy and security. However, the performance of existing FL approaches often deteriorates when encountering non-iid situations where there exist distribution gaps among clients, and few previous efforts focus on personalization in healthcare. In this article, we propose FedAP to tackle domain shifts and then obtain personalized models for local clients. FedAP learns the similarity between clients based on the statistics of the batch normalization layers while preserving the specificity of each client with different local batch normalization. Comprehensive experiments on five healthcare benchmarks demonstrate that FedAP achieves better accuracy compared to state-of-the-art methods (e.g., 10% accuracy improvement for PAMAP2) with faster convergence speed.
updated: Fri May 20 2022 02:13:58 GMT+0000 (UTC)
published: Wed Dec 01 2021 11:36:56 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アソシエイト