arXiv reaDer
ローカル学習の問題:連合学習におけるデータの不均一性の再考
Local Learning Matters: Rethinking Data Heterogeneity in Federated Learning
連合学習(FL)は、クライアントのネットワーク(つまり、エッジデバイス)を使用してプライバシーを保護する分散学習を実行するための有望な戦略です。ただし、クライアント間のデータ分散は本質的に非IIDであることが多く、効率的な最適化が困難になります。この問題を軽減するために、多くのFLアルゴリズムは、グローバルモデルに関してローカル更新を制限するために、さまざまな近位項を導入することによってクライアント間のデータの不均一性の影響を軽減することに焦点を当てています。代わりに、近位の制限ではなく、ローカル学習の一般性に焦点を当てて、FLのデータの不均一性に対するソリューションを再考することを検討します。この目的のために、最初に、FLでのアルゴリズムの有効性をよりよく理解するために、2次指標によって通知される体系的な研究を提示します。興味深いことに、標準の正則化手法は、データの不均一性の影響を軽減する上で驚くほど強力なパフォーマンスを発揮します。私たちの調査結果に基づいて、データの不均一性と以前の方法の落とし穴を克服するために、シンプルで効果的な方法であるFedAlignをさらに提案します。 FedAlignは、計算とメモリのオーバーヘッドを最小限に抑えながら、さまざまな設定で最先端のFLメソッドを使用して競争力のある精度を実現します。コードは公開されます。
Federated learning (FL) is a promising strategy for performing privacy-preserving, distributed learning with a network of clients (i.e., edge devices). However, the data distribution among clients is often non-IID in nature, making efficient optimization difficult. To alleviate this issue, many FL algorithms focus on mitigating the effects of data heterogeneity across clients by introducing a variety of proximal terms, some incurring considerable compute and/or memory overheads, to restrain local updates with respect to the global model. Instead, we consider rethinking solutions to data heterogeneity in FL with a focus on local learning generality rather than proximal restriction. To this end, we first present a systematic study informed by second-order indicators to better understand algorithm effectiveness in FL. Interestingly, we find that standard regularization methods are surprisingly strong performers in mitigating data heterogeneity effects. Based on our findings, we further propose a simple and effective method, FedAlign, to overcome data heterogeneity and the pitfalls of previous methods. FedAlign achieves competitive accuracy with state-of-the-art FL methods across a variety of settings while minimizing computation and memory overhead. Code will be publicly available.
updated: Sun Nov 28 2021 19:03:39 GMT+0000 (UTC)
published: Sun Nov 28 2021 19:03:39 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アソシエイト