← Search

Anant Raj

19 accepted papers

2024

Small steps no more: Global convergence of stochastic gradient bandits for arbitrary learning rates

NeurIPS 2024poster

We provide a new understanding of the stochastic gradient bandit algorithm by showing that it converges to a globally optimal policy almost surely using \emph{any} constant learning rate. This result demonstrates that the stochastic gradient algorithm continues to balance exploration and exploitatio…

Cited by 1SourcePDFScholar
2023

Algorithmic Stability of Heavy-Tailed SGD with General Loss Functions

ICML 2023poster

Heavy-tail phenomena in stochastic gradient descent (SGD) have been reported in several empirical studies. Experimental evidence in previous works suggests a strong interplay between the heaviness of the tails and generalization behavior of SGD. To address this empirical phenomena theoretically, sev…

Cited by 26SourcePDFScholar
2023

Efficient Sampling of Stochastic Differential Equations with Positive Semi-Definite Models

NeurIPS 2023poster

This paper deals with the problem of efficient sampling from a stochastic differential equation, given the drift function and the diffusion matrix. The proposed approach leverages a recent model for probabilities (Rudi and Ciliberto, 2021) (the positive semi-definite -- PSD model) from which it is p…

Cited by 1SourcePDFScholar
2023

Explicit Regularization in Overparametrized Models via Noise Injection

AISTATS 2023poster

Injecting noise within gradient descent has several desirable features, such as smoothing and regularizing properties. In this paper, we investigate the effects of injecting noise before computing a gradient step. We demonstrate that small perturbations can induce explicit regularization for simple…

2023

Uniform-in-Time Wasserstein Stability Bounds for (Noisy) Stochastic Gradient Descent

NeurIPS 2023poster

Algorithmic stability is an important notion that has proven powerful for deriving generalization bounds for practical algorithms. The last decade has witnessed an increasing number of stability bounds for different algorithms applied on different classes of loss functions. While these bounds have i…

Cited by 9SourcePDFScholar
2022

Faster Rates, Adaptive Algorithms, and Finite-Time Bounds for Linear Composition Optimization and Gradient TD Learning

AISTATS 2022poster

Gradient temporal difference (GTD) algorithms are provably convergent policy evaluation methods for off-policy reinforcement learning. Despite much progress, proper tuning of the stochastic approximation methods used to solve the resulting saddle point optimization problem requires the knowledge of…

Cited by 1SourcePDFScholar
2020

A simpler approach to accelerated optimization: iterative averaging meets optimism

ICML 2020poster

Recently there have been several attempts to extend Nesterov’s accelerated algorithm to smooth stochastic and variance-reduced optimization. In this paper, we show that there is a simpler approach to acceleration: applying optimistic online learning algorithms and querying the gradient oracle at the…

Cited by 40SourcePDFScholar
2019

Sobolev Descent

AISTATS 2019poster

We study a simplification of GAN training: the problem of transporting particles from a source to a target distribution. Starting from the Sobolev GAN critic, part of the gradient regularized GAN family, we show a strong relation with Optimal Transport (OT). Specifically with the less popular *dyna…

Cited by 48SourcePDFScholar
2018

On Matching Pursuit and Coordinate Descent

ICML 2018oral

Two popular examples of first-order optimization methods over linear spaces are coordinate descent and matching pursuit algorithms, with their randomized variants. While the former targets the optimization by moving along coordinates, the latter considers a generalized notion of directions. Exploiti…

Cited by 29SourcePDFScholar
2017

Local Group Invariant Representations via Orbit Embeddings

AISTATS 2017poster

Invariance to nuisance transformations is one of the desirable properties of effective representations. We consider transformations that form a group and propose an approach based on kernel methods to derive local group invariant representations. Locality is achieved by defining a suitable probabili…

Cited by 40SourcePDFScholar