Exploiting the Structure: Stochastic Gradient Methods Using Raw Clusters

by   Zeyuan Allen-Zhu, et al.

The amount of data available in the world is growing faster than our ability to deal with it. However, if we take advantage of the internal structure, data may become much smaller for machine learning purposes. In this paper we focus on one of the fundamental machine learning tasks, empirical risk minimization (ERM), and provide faster algorithms with the help from the clustering structure of the data. We introduce a simple notion of raw clustering that can be efficiently computed from the data, and propose two algorithms based on clustering information. Our accelerated algorithm ClusterACDM is built on a novel Haar transformation applied to the dual space of the ERM problem, and our variance-reduction based algorithm ClusterSVRG introduces a new gradient estimator using clustering. Our algorithms outperform their classical counterparts ACDM and SVRG respectively.


page 1

page 2

page 3

page 4


The Practicality of Stochastic Optimization in Imaging Inverse Problems

In this work we investigate the practicality of stochastic gradient desc...

Gradient tracking and variance reduction for decentralized optimization and machine learning

Decentralized methods to solve finite-sum minimization problems are impo...

Stochastic Approximation Beyond Gradient for Signal Processing and Machine Learning

Stochastic approximation (SA) is a classical algorithm that has had sinc...

Network Distance Based on Laplacian Flows on Graphs

Distance plays a fundamental role in measuring similarity between object...

COCO Denoiser: Using Co-Coercivity for Variance Reduction in Stochastic Convex Optimization

First-order methods for stochastic optimization have undeniable relevanc...

Approximate Inference via Clustering

In recent years, large-scale Bayesian learning draws a great deal of att...

Code Repositories


This project collects the different accepted papers and their link to Arxiv or Gitxiv

view repo

Please sign up or login with your details

Forgot password? Click here to reset