arXiv reaDer
CD ^ 2-pFed:連合学習におけるモデルのパーソナライズのための循環蒸留ガイドチャネルデカップリング
CD^2-pFed: Cyclic Distillation-guided Channel Decoupling for Model Personalization in Federated Learning
連合学習(FL)は、複数のクライアントが共有グローバルモデルを共同で学習できるようにする分散学習パラダイムです。最近の進歩にもかかわらず、データの分散が異なると、通常、グローバルモデルが参加している各クライアントに優れた一般化機能を提供できないため、異種データクライアントを処理することは依然として困難です。この論文では、データの不均一性のさまざまな設定の下で、FLのグローバルモデルをパーソナライズするための、新しい循環蒸留ガイドチャネルデカップリングフレームワークであるCD^2-pFedを提案します。異なるクライアント間で非IIDデータを克服するためにレイヤーごとのパーソナライズを確立する以前の作業とは異なり、チャネルデカップリングと呼ばれるモデルパーソナライズのためのチャネルごとの割り当てを最初に試みます。プライベートウェイトと共有ウェイト間のコラボレーションをさらに促進するために、フェデレーション中にローカルモデル表現とグローバルモデル表現の間に一貫した正則化を課す新しい循環蒸留スキームを提案します。循環蒸留によって導かれる、チャネルデカップリングフレームワークは、機能スキュー、ラベル分布スキュー、コンセプトシフトなどのさまざまな種類の不均一性に対してより正確で一般化された結果を提供できます。自然画像と医療画像分析タスクを含む4つのベンチマークでの包括的な実験は、ローカルと外部の両方の検証で私たちの方法の一貫した有効性を示しています。
Federated learning (FL) is a distributed learning paradigm that enables multiple clients to collaboratively learn a shared global model. Despite the recent progress, it remains challenging to deal with heterogeneous data clients, as the discrepant data distributions usually prevent the global model from delivering good generalization ability on each participating client. In this paper, we propose CD^2-pFed, a novel Cyclic Distillation-guided Channel Decoupling framework, to personalize the global model in FL, under various settings of data heterogeneity. Different from previous works which establish layer-wise personalization to overcome the non-IID data across different clients, we make the first attempt at channel-wise assignment for model personalization, referred to as channel decoupling. To further facilitate the collaboration between private and shared weights, we propose a novel cyclic distillation scheme to impose a consistent regularization between the local and global model representations during the federation. Guided by the cyclical distillation, our channel decoupling framework can deliver more accurate and generalized results for different kinds of heterogeneity, such as feature skew, label distribution skew, and concept shift. Comprehensive experiments on four benchmarks, including natural image and medical image analysis tasks, demonstrate the consistent effectiveness of our method on both local and external validations.
updated: Fri Apr 08 2022 07:13:30 GMT+0000 (UTC)
published: Fri Apr 08 2022 07:13:30 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アソシエイト