Distributed Sketching on Data Partitions for OLS Regression
Luyuan Yang, Brayden Garner, Shayan Shafaei, Chao Lan
cs.LG
Abstract
This paper studies distributed sketching for ordinary least squares (OLS) regression, an approach that distributes small sketches of a large data set over multiple machines to separately construct OLS estimators and average them. Unlike prior studies that consider sketching on the whole data set, we consider sketching on partitioned subsets to further reduce computational cost. Under the fixed design setting, we characterize the exact excess loss of the averaged OLS estimator. Results show that this loss is comparable to the established loss for sketching on the whole data set when the divergence among subset covariances is small.
Topics
Classified with taxonomy v2 on Sat, 5 Sept 2026.