arXiv reaDer
CosSGD:単純なコサインベースの量子化による通信効率の高い連合学習
CosSGD: Communication-Efficient Federated Learning with a Simple Cosine-Based Quantization
連合学習は、データのプライバシーと計算の懸念を軽減するための有望なフレームワークです。ただし、サーバーとクライアント間の通信コストは、展開を成功させるための主要なボトルネックになっています。勾配圧縮の目覚ましい進歩にもかかわらず、既存の量子化方法は、低ビット圧縮が適用される場合、特に、モデルの重みと勾配を圧縮するために量子化が二重方向に適用される場合、システム全体が大幅に縮退することがよくあります。この作業では、単純なコサインベースの非線形量子化を提案し、ラウンドトリップ通信コストの圧縮で印象的な結果を達成します。以前の方法よりも高い比率でモデルの重みと勾配を圧縮できるだけでなく、同時に競合するモデルのパフォーマンスを実現できます。さらに、私たちのアプローチは、計算の複雑さが低く、圧縮された情報を回復するためにわずかな追加データしか必要としないため、連合学習の問題に非常に適しています。 CIFAR-10とBraTSデータセットを使用して、画像分類と脳腫瘍のセマンティックセグメンテーションについて広範な実験が行われ、最先端の有効性と優れた通信効率が示されています。
Federated learning is a promising framework to mitigate data privacy and computation concerns. However, the communication cost between the server and clients has become the major bottleneck for successful deployment. Despite notable progress in gradient compression, the existing quantization methods require further improvement when low-bits compression is applied, especially the overall systems often degenerate a lot when quantization are applied in double directions to compress model weights and gradients. In this work, we propose a simple cosine-based nonlinear quantization and achieve impressive results in compressing round-trip communication costs. We are not only able to compress model weights and gradients at higher ratios than previous methods, but also achieve competing model performance at the same time. Further, our approach is highly suitable for federated learning problems since it has low computational complexity and requires only a little additional data to recover the compressed information. Extensive experiments have been conducted on image classification and brain tumor semantic segmentation using the CIFAR-10, and BraTS datasets where we show state-of-the-art effectiveness and impressive communication efficiency.
updated: Mon May 02 2022 00:28:53 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アソシエイト