arXiv reaDer
非IID連合学習のためのデータフリー知識蒸留によるグローバルモデルの微調整
Fine-tuning Global Model via Data-Free Knowledge Distillation for Non-IID Federated Learning
連合学習(FL)は、プライバシーの制約下にある新しい分散学習パラダイムです。データの不均一性はFLの主な課題の1つであり、収束が遅くなり、パフォーマンスが低下します。ほとんどの既存のアプローチは、クライアントでのローカルモデルの更新を制限することによって異質性の課題に取り組むだけであり、直接的なグローバルモデルの集約によって引き起こされるパフォーマンスの低下を無視します。代わりに、サーバー内のグローバルモデル(FedFTG)を微調整するためのデータフリーの知識蒸留方法を提案します。これにより、モデルの直接集約の問題が軽減されます。具体的には、FedFTGはジェネレーターを介してローカルモデルの入力スペースを探索し、それを使用してローカルモデルからグローバルモデルに知識を転送します。さらに、トレーニング全体で効果的な知識の抽出を実現するために、ハードサンプルマイニングスキームを提案します。さらに、カスタマイズされたラベルサンプリングとクラスレベルのアンサンブルを開発して、知識を最大限に活用します。これにより、クライアント間の分散の不一致が暗黙的に軽減されます。広範な実験により、FedFTGは最先端の(SOTA)FLアルゴリズムを大幅に上回り、FedAvg、FedProx、FedDyn、およびSCAFFOLDを強化するための強力なプラグインとして機能できることが示されています。
Federated Learning (FL) is an emerging distributed learning paradigm under privacy constraint. Data heterogeneity is one of the main challenges in FL, which results in slow convergence and degraded performance. Most existing approaches only tackle the heterogeneity challenge by restricting the local model update in client, ignoring the performance drop caused by direct global model aggregation. Instead, we propose a data-free knowledge distillation method to fine-tune the global model in the server (FedFTG), which relieves the issue of direct model aggregation. Concretely, FedFTG explores the input space of local models through a generator, and uses it to transfer the knowledge from local models to the global model. Besides, we propose a hard sample mining scheme to achieve effective knowledge distillation throughout the training. In addition, we develop customized label sampling and class-level ensemble to derive maximum utilization of knowledge, which implicitly mitigates the distribution discrepancy across clients. Extensive experiments show that our FedFTG significantly outperforms the state-of-the-art (SOTA) FL algorithms and can serve as a strong plugin for enhancing FedAvg, FedProx, FedDyn, and SCAFFOLD.
updated: Thu Mar 17 2022 11:18:17 GMT+0000 (UTC)
published: Thu Mar 17 2022 11:18:17 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アソシエイト