arXiv reaDer
個別化医療のための適応バッチノルムによる連合学習
Federated Learning with Adaptive Batchnorm for Personalized Healthcare
医療に機械学習技術を適用することへの関心が高まっています。最近、フェデレーション機械学習(FL)は、データのプライバシーとセキュリティを損なうことなく強力なモデルをトレーニングできるため、人気が高まっています。ただし、既存のFLアプローチのパフォーマンスは、クライアント間に分布のギャップが存在する非iidの状況に遭遇すると低下することが多く、これまでの取り組みのほとんどは医療におけるパーソナライズに焦点を当てていませんでした。この記事では、ドメインシフトに取り組み、ローカルクライアント向けにパーソナライズされたモデルを取得するためのAdaFedを提案します。 AdaFedは、異なるローカルバッチ正規化で各クライアントの特異性を維持しながら、バッチ正規化レイヤーの統計を介してクライアント間の類似性を学習します。 5つのヘルスケアベンチマークに関する包括的な実験は、AdaFedが最先端の方法(たとえば、PAMAP2の10%以上の精度向上)と比較して、より速い収束速度でより優れた精度を達成することを示しています。
There is a growing interest in applying machine learning techniques for healthcare. Recently, federated machine 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 AdaFed to tackle domain shifts and obtain personalized models for local clients. AdaFed learns the similarity between clients via 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 AdaFed achieves better accuracy compared to state-of-the-art methods (e.g., 10%+ accuracy improvement for PAMAP2) with faster convergence speed.
updated: Fri Apr 08 2022 03:37:57 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アソシエイト