arXiv reaDer
連合学習における非真の自己知識蒸留によるグローバル知識の保存
Preservation of the Global Knowledge by Not-True Self Knowledge Distillation in Federated Learning
連合学習では、クライアントのローカルでトレーニングされたモデルを集約することにより、強力なグローバルモデルが共同で学習されます。これにより、クライアントのデータに直接アクセスする必要がなくなりますが、グローバルモデルの収束は、データの不均一性に悩まされることがよくあります。この研究は、継続的な学習へのアナロジーから始まり、忘却が連合学習のボトルネックになる可能性があることを示唆しています。グローバルモデルは前のラウンドからの知識を忘れ、ローカルトレーニングはローカル分布の外で知識を忘れることを誘発することを観察します。私たちの調査結果に基づいて、忘却に取り組むことでデータの不均一性の問題が軽減されると仮定します。この目的のために、新規で効果的なアルゴリズムであるFederated Not-True Distillation(FedNTD)を提案します。これは、非真のクラスに対してのみローカルで利用可能なデータのグローバルな視点を保持します。実験では、FedNTDは、データのプライバシーを危険にさらしたり、追加の通信コストを発生させたりすることなく、さまざまなセットアップで最先端のパフォーマンスを示しています。
In federated learning, a strong global model is collaboratively learned by aggregating clients' locally trained models. Although this precludes the need to access clients' data directly, the global model's convergence often suffers from data heterogeneity. This study starts from an analogy to continual learning and suggests that forgetting could be the bottleneck of federated learning. We observe that the global model forgets the knowledge from previous rounds, and the local training induces forgetting the knowledge outside of the local distribution. Based on our findings, we hypothesize that tackling down forgetting will relieve the data heterogeneity problem. To this end, we propose a novel and effective algorithm, Federated Not-True Distillation (FedNTD), which preserves the global perspective on locally available data only for the not-true classes. In the experiments, FedNTD shows state-of-the-art performance on various setups without compromising data privacy or incurring additional communication costs.
updated: Wed Feb 16 2022 10:51:39 GMT+0000 (UTC)
published: Sun Jun 06 2021 11:51:47 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アソシエイト