We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

math.ST

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Mathematics > Statistics Theory

Title: Distributed linear regression by averaging

Abstract: Distributed statistical learning problems arise commonly when dealing with large datasets. In this setup, datasets are partitioned over machines, which compute locally, and communicate short messages. Communication is often the bottleneck. In this paper, we study one-step and iterative weighted parameter averaging in statistical linear models under data parallelism. We do linear regression on each machine, send the results to a central server, and take a weighted average of the parameters. Optionally, we iterate, sending back the weighted average and doing local ridge regressions centered at it. How does this work compared to doing linear regression on the full data? Here we study the performance loss in estimation, test error, and confidence interval length in high dimensions, where the number of parameters is comparable to the training data size. We find the performance loss in one-step weighted averaging, and also give results for iterative averaging. We also find that different problems are affected differently by the distributed framework. Estimation error and confidence interval length increase a lot, while prediction error increases much less. We rely on recent results from random matrix theory, where we develop a new calculus of deterministic equivalents as a tool of broader interest.
Comments: Fixing a typo
Subjects: Statistics Theory (math.ST); Computation (stat.CO); Methodology (stat.ME); Machine Learning (stat.ML)
Journal reference: Ann. Statist. 49(2): 918-943 (April 2021)
Cite as: arXiv:1810.00412 [math.ST]
  (or arXiv:1810.00412v3 [math.ST] for this version)

Submission history

From: Edgar Dobriban [view email]
[v1] Sun, 30 Sep 2018 15:59:03 GMT (1559kb,D)
[v2] Thu, 10 Oct 2019 04:33:28 GMT (2545kb,D)
[v3] Sat, 22 Oct 2022 20:59:52 GMT (2545kb,D)

Link back to: arXiv, form interface, contact.