LoRDO让大模型分布式训练通信量减少10倍,同时保持高精度。
LoRDO: Distributed Low-Rank Optimization with Infrequent Communication
- 用低秩优化+稀疏同步,减少通信开销。
- 在125M到720M模型上达到接近低秩DDP的性能。
- 特别适合内存极小、批量小的设备部署。
通过数据并行(DDP)进行基础模型的分布式训练受限于互连带宽。尽管稀疏通信策略能降低同步频率,但仍受优化器状态带来的内存与通信开销制约。低秩优化器可缓解此问题;然而,在局部更新模式下,工作节点缺乏全批量梯度以计算低秩投影,导致性能下降。我们提出LoRDO,一个将低秩优化与稀疏同步有机结合的框架。我们首先证明:基于伪梯度的全局投影虽理论更优,但会永久限制优化轨迹至低秩子空间。为恢复子空间探索能力,引入全秩准双曲更新。LoRDO在125M至720M模型的语言建模与下游任务中表现接近低秩DDP,同时通信量减少约10倍。此外,在极低内存环境下,小秩/小批量设置下性能提升更为显著。
原文摘要 · Abstract (English)
Distributed training of foundation models via $\texttt{DDP}$ is limited by interconnect bandwidth. While infrequent communication strategies reduce synchronization frequency, they remain bottlenecked by the memory and communication requirements of optimizer states. Low-rank optimizers can alleviate these constraints; however, in the local-update regime, workers lack access to the full-batch gradients required to compute low-rank projections, which degrades performance. We propose $\texttt{LoRDO}$, a principled framework unifying low-rank optimization with infrequent synchronization. We first demonstrate that, while global projections based on pseudo-gradients are theoretically superior, they permanently restrict the optimization trajectory to a low-rank subspace. To restore subspace exploration, we introduce a full-rank quasi-hyperbolic update. $\texttt{LoRDO}$ achieves near-parity with low-rank $\texttt{DDP}$ in language modeling and downstream tasks at model scales of $125$M--$720$M, while reducing communication by $\approx 10 \times$. Finally, we show that $\texttt{LoRDO}$ improves performance even more in very low-memory settings with small rank/batch size.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。