Dec, 2023

MG-Skip:用于非光滑分布式优化的随机多八卦跳跃方法

TL;DR基于概率局部更新的分布式优化方法在通信加速方面引起了人们的关注,然而,这种能力只在损失函数平滑且网络连接充分的情况下有效。本文提出了第一种具有概率局部更新的线性收敛方法MG-Skip用于非平滑分布式优化,无需额外条件对网络连接性能,大多数迭代可以跳过多轮“传闲话”通信,其迭代复杂度为O(κ log(1/ε)),通信复杂度仅为O(√(κ/(1-ρ)) log(1/ε)),其中κ是损失函数的条件数,ρ反映了网络拓扑的连通性。据我们所知,当损失函数具有平滑(强凸)+非平滑(凸)复合形式时,MG-Skip实现了最佳的通信复杂度。