Distributed machine learning has recently become a critical paradigm for training large models on vast datasets. We examine the stochastic optimization problem for deep learning within synchronous parallel computing environments under communication constraints. Although averaging distributed gradients is the default method for gradient estimation, whether this is the optimal strategy remains an open question. In this work, we analyze the distributed gradient aggregation process through the lens of subspace optimization. By formulating the aggregation problem as an objective-aware subspace optimization problem, we derive an efficient gradient weighting scheme. Our method maintains statistical unbiasedness in the aggregation and is hyperparameter-free. We demonstrate improved performance of our method over ubiquitous gradient averaging on multiple MLPerf tasks while remaining efficient in both communicational and computational complexity. The code is provided as supplementary material.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

AdaCons: Adaptive Consensus Gradient Aggregation for Distributed Training

  • Yoni Choukroun,
  • Shlomi Azoulay,
  • Pavel Kisilev

摘要

Distributed machine learning has recently become a critical paradigm for training large models on vast datasets. We examine the stochastic optimization problem for deep learning within synchronous parallel computing environments under communication constraints. Although averaging distributed gradients is the default method for gradient estimation, whether this is the optimal strategy remains an open question. In this work, we analyze the distributed gradient aggregation process through the lens of subspace optimization. By formulating the aggregation problem as an objective-aware subspace optimization problem, we derive an efficient gradient weighting scheme. Our method maintains statistical unbiasedness in the aggregation and is hyperparameter-free. We demonstrate improved performance of our method over ubiquitous gradient averaging on multiple MLPerf tasks while remaining efficient in both communicational and computational complexity. The code is provided as supplementary material.