本論文は、リーマン多様体上のフェデレーション学習のための新しいアルゴリズムRFedAGSを提案し、その収束性を分析している。
主な内容は以下の通り:
従来のフェデレーション学習アルゴリズムでは、サーバーがエージェントからアップロードされた局所パラメータの平均を取ることで新しいグローバルパラメータを生成していた。しかし、リーマン多様体上では、この平均操作が複雑になる。
本論文では、エージェントがアップロードする勾配ストリームの平均化に基づいて、サーバーが新しいグローバルパラメータを生成するRFedAGSアルゴリズムを提案する。これは、ユークリッド空間におけるフェデレーション平均の自然な一般化と考えられる。
RFedAGSの収束性を理論的に分析する。一般の非凸問題に対して、固定ステップサイズでは亜線形収束率を示し、ポリャック-ロジャシェフィッツ条件を満たす問題に対しては線形収束性を示す。
数値実験の結果は、理論的な分析と一致しており、RFedAGSが中央集権的な手法と比較して遜色ないパフォーマンスを示すことを明らかにする。
他の言語に翻訳
原文コンテンツから
arxiv.org
深掘り質問