← Search

Fabian Pedregosa

23 accepted papers

2024

Stepping on the Edge: Curvature Aware Learning Rate Tuners

NeurIPS 2024poster

Curvature information -- particularly, the largest eigenvalue of the loss Hessian, known as the sharpness -- often forms the basis for learning rate tuners. However, recent work has shown that the curvature information undergoes complex dynamics during training, going from a phase of increasing shar…

Cited by 1SourcePDFScholar
2023

A Novel Stochastic Gradient Descent Algorithm for Learning Principal Subspaces

AISTATS 2023poster

Many machine learning problems encode their data as a matrix with a possibly very large number of rows and columns. In several applications like neuroscience, image compression or deep reinforcement learning, the principal subspace of such a matrix provides a useful, low-dimensional representation o…

2023

Second-order regression models exhibit progressive sharpening to the edge of stability

ICML 2023poster

Recent studies of gradient descent with large step sizes have shown that there is often a regime with an initial increase in the largest eigenvalue of the loss Hessian (progressive sharpening), followed by a stabilization of the eigenvalue near the maximum value which allows convergence (edge of sta…

Cited by 33SourcePDFScholar
2022

Efficient and Modular Implicit Differentiation

NeurIPS 2022accept

Automatic differentiation (autodiff) has revolutionized machine learning. It allows to express complex computations by composing elementary ones in creative ways and removes the burden of computing their derivatives by hand. More recently, differentiation of optimization problem solutions has attra…

2022

GradMax: Growing Neural Networks using Gradient Information

ICLR 2022poster

The architecture and the parameters of neural networks are often optimized independently, which requires costly retraining of the parameters whenever the architecture is modified. In this work we instead focus on growing the architecture without requiring costly retraining. We present a method that…

2022

On Implicit Bias in Overparameterized Bilevel Optimization

ICML 2022spotlight

Many problems in machine learning involve bilevel optimization (BLO), including hyperparameter optimization, meta-learning, and dataset distillation. Bilevel problems involve inner and outer parameters, each optimized for its own objective. Often, at least one of the two levels is underspecified and…

2022

Only tails matter: Average-Case Universality and Robustness in the Convex Regime

ICML 2022spotlight

The recently developed average-case analysis of optimization methods allows a more fine-grained and representative convergence analysis than usual worst-case results. In exchange, this analysis requires a more precise hypothesis over the data generating process, namely assuming knowledge of the expe…

Cited by 11SourcePDFScholar
2022

Super-Acceleration with Cyclical Step-sizes

AISTATS 2022poster

We develop a convergence-rate analysis of momentum with cyclical step-sizes. We show that under some assumption on the spectral gap of Hessians in machine learning, cyclical step-sizes are provably faster than constant step-sizes. More precisely, we develop a convergence rate analysis for quadratic…

2022

The Curse of Unrolling: Rate of Differentiating Through Optimization

NeurIPS 2022accept

Computing the Jacobian of the solution of an optimization problem is a central problem in machine learning, with applications in hyperparameter optimization, meta-learning, optimization as a layer, and dataset distillation, to name a few. Unrolled differentiation is a popular heuristic that approxim…

Cited by 17SourcePDFScholar
2021

Average-case Acceleration for Bilinear Games and Normal Matrices

ICLR 2021poster

Advances in generative modeling and adversarial learning have given rise to renewed interest in smooth games. However, the absence of symmetry in the matrix of second derivatives poses challenges that are not present in the classical minimization framework. While a rich theory of average-case analys…

Cited by 8SourcePDFScholar
2021

Boosting Variational Inference With Locally Adaptive Step-Sizes

IJCAI 2021poster

Variational Inference makes a trade-off between the capacity of the variational family and the tractability of finding an approximate posterior distribution. Instead, Boosting Variational Inference allows practitioners to obtain increasingly good posterior approximations by spending more compute. Th…

2020

Linearly Convergent Frank-Wolfe with Backtracking Line-Search

AISTATS 2020poster

Structured constraints in Machine Learning have recently brought the Frank-Wolfe (FW) family of algorithms back in the spotlight. While the classical FW algorithm has poor local convergence properties, the Away-steps and Pairwise FW variants have emerged as improved variants with faster convergence.…

2020

On the interplay between noise and curvature and its effect on optimization and generalization

AISTATS 2020poster

The speed at which one can minimize an expected loss using stochastic methods depends on two properties: the curvature of the loss and the variance of the gradients. While most previous works focus on one or the other of these properties, we explore how their interaction affects optimization speed.…

Cited by 72SourcePDFScholar
2020

Stochastic Frank-Wolfe for Constrained Finite-Sum Minimization

ICML 2020poster

We propose a novel Stochastic Frank-Wolfe (a. k. a. conditional gradient) algorithm for constrained smooth finite-sum minimization with a generalized linear prediction/structure. This class of problems includes empirical risk minimization with sparse, low-rank, or other structured constraints. The p…

2017

Breaking the Nonsmooth Barrier: A Scalable Parallel Method for Composite Optimization

NeurIPS 2017spotlight

Due to their simplicity and excellent performance, parallel asynchronous variants of stochastic gradient descent have become popular methods to solve a wide range of large-scale optimization problems on multi-core architectures. Yet, despite their practical success, support for nonsmooth objectives…