English

Optimal Convergence for Distributed Learning with Stochastic Gradient Methods and Spectral Algorithms

Machine Learning 2018-11-06 v2 Artificial Intelligence Machine Learning Functional Analysis

Abstract

We study generalization properties of distributed algorithms in the setting of nonparametric regression over a reproducing kernel Hilbert space (RKHS). We first investigate distributed stochastic gradient methods (SGM), with mini-batches and multi-passes over the data. We show that optimal generalization error bounds can be retained for distributed SGM provided that the partition level is not too large. We then extend our results to spectral-regularization algorithms (SRA), including kernel ridge regression (KRR), kernel principal component analysis, and gradient methods. Our results are superior to the state-of-the-art theory. Particularly, our results show that distributed SGM has a smaller theoretical computational complexity, compared with distributed KRR and classic SGM. Moreover, even for non-distributed SRA, they provide the first optimal, capacity-dependent convergence rates, considering the case that the regression function may not be in the RKHS.

Keywords

Cite

@article{arxiv.1801.07226,
  title  = {Optimal Convergence for Distributed Learning with Stochastic Gradient Methods and Spectral Algorithms},
  author = {Junhong Lin and Volkan Cevher},
  journal= {arXiv preprint arXiv:1801.07226},
  year   = {2018}
}

Comments

53 pages

R2 v1 2026-06-22T23:52:16.192Z