arXiv reaDer
コミュニケーションに向けて-効率的でプライバシーを保護する連合表現学習
Towards Communication-Efficient and Privacy-Preserving Federated Representation Learning
この論文は、通信コストとプライバシー保護の制約の下での連合表現学習の実現可能性を調査します。既存の作品は、頻繁なコミュニケーションを必要とする注釈に基づくローカルトレーニングを実施するか、プライバシーにさらされる可能性のある重みの平均化を介してクライアントモデルを集約します。上記の問題に取り組むために、まず、自己監視型の対照的なローカルトレーニングが、同一に分散されていないデータに対して堅牢であることを確認します。これにより、より長いローカルトレーニングの実現可能性が提供され、通信コストが削減されます。次に、前述のロバスト性に基づいて、このロバスト性を利用したアンサンブル類似性蒸留〜(FLESD)を備えた新しいフェデレーション表現学習フレームワークを提案します。通信の各ラウンドで、サーバーは最初に、公開データセット上のクライアントの推定類似性マトリックスの一部を収集します。次に、類似性行列をアンサンブルし、類似性蒸留を介してグローバルモデルをトレーニングします。一連の実証実験によってFLESDの有効性を検証し、より厳しい制約にもかかわらず、複数のデータセットの複数の設定で同等の結果が得られることを示します。
This paper investigates the feasibility of federated representation learning under the constraints of communication cost and privacy protection. Existing works either conduct annotation-guided local training which requires frequent communication or aggregates the client models via weight averaging which has potential risks of privacy exposure. To tackle the above problems, we first identify that self-supervised contrastive local training is robust against the non-identically distributed data, which provides the feasibility of longer local training and thus reduces the communication cost. Then based on the aforementioned robustness, we propose a novel Federated representation Learning framework with Ensemble Similarity Distillation~(FLESD) that utilizes this robustness. At each round of communication, the server first gathers a fraction of the clients' inferred similarity matrices on a public dataset. Then it ensembles the similarity matrices and train the global model via similarity distillation. We verify the effectiveness of FLESD by a series of empirical experiments and show that, despite stricter constraints, it achieves comparable results under multiple settings on multiple datasets.
updated: Sat Jan 15 2022 16:02:01 GMT+0000 (UTC)
published: Wed Sep 29 2021 02:13:22 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アソシエイト