arXiv reaDer
CosSGD:通信効率の高い連合学習のための非線形量子化
CosSGD: Nonlinear Quantization for Communication-efficient Federated Learning
統合学習は、これらのクライアントのローカルデータを中央サーバーに転送することなく、クライアント間の学習を容易にします。連合学習法の成功にもかかわらず、限られた通信条件下でモデルを更新するために最も重要な情報を通信することをさらに改善する必要があります。これは、この学習スキームを幅広いアプリケーションシナリオに役立てることができます。この作業では、フェデレーション学習で簡単に利用できる圧縮確率的勾配降下法の非線形量子化を提案します。提案された量子化に基づいて、私たちのシステムは、トレーニングプロセスの収束と精度を大幅に維持しながら、通信コストを最大3桁大幅に削減します。 MNIST、CIFAR-10、BraTSデータセットを使用して、画像分類と脳腫瘍のセマンティックセグメンテーションについて広範な実験が行われ、最先端の有効性と優れた通信効率が示されています。
Federated learning facilitates learning across clients without transferring local data on these clients to a central server. Despite the success of the federated learning method, it remains to improve further w.r.t communicating the most critical information to update a model under limited communication conditions, which can benefit this learning scheme into a wide range of application scenarios. In this work, we propose a nonlinear quantization for compressed stochastic gradient descent, which can be easily utilized in federated learning. Based on the proposed quantization, our system significantly reduces the communication cost by up to three orders of magnitude, while maintaining convergence and accuracy of the training process to a large extent. Extensive experiments are conducted on image classification and brain tumor semantic segmentation using the MNIST, CIFAR-10 and BraTS datasets where we show state-of-the-art effectiveness and impressive communication efficiency.
updated: Tue Dec 15 2020 12:20:28 GMT+0000 (UTC)
published: Tue Dec 15 2020 12:20:28 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アソシエイト