中文

分布式与联邦学习中通信压缩的不确定性原理及最优压缩器探寻

机器学习 2021-05-04 v3 分布式、并行与集群计算 信息论 math.IT 最优化与控制 机器学习

摘要

为了缓解分布式与联邦学习中的高通信成本,各种向量压缩方案(如量化、稀疏化和抖动)已变得非常流行。在设计压缩方法时,人们旨在以尽可能少的比特进行通信,从而最小化每轮通信的成本,同时试图对通信消息施加尽可能小的失真(方差),从而最小化压缩对总体通信轮数的不利影响。然而,直观来看,这两个目标根本上是冲突的:我们允许的压缩越多,消息失真就越严重。我们将这一直觉形式化,并证明了随机压缩算子的不确定性原理,从而从数学上量化了这一局限,并有效地给出了通信压缩可实现性的渐近紧下界。受这些进展的推动,我们呼吁探寻最优压缩算子。作为朝此方向迈出的第一步,我们考虑一种受向量Kashin表示启发的无偏压缩方法,称之为Kashin压缩(KC)。与所有先前提出的压缩机制不同,KC享有维度无关的方差界,即使在对每个向量元素仅需通信极少比特的情形下,我们也推导出了其显式公式。

关键词

引用

@article{arxiv.2002.08958,
  title  = {Uncertainty Principle for Communication Compression in Distributed and Federated Learning and the Search for an Optimal Compressor},
  author = {Mher Safaryan and Egor Shulgin and Peter Richtárik},
  journal= {arXiv preprint arXiv:2002.08958},
  year   = {2021}
}

备注

23 pages, 6 figures, 2 tables