← Search

Michael Lindsey

3 accepted papers

2025

Implicit Bias of Gradient Descent for Non-Homogeneous Deep Networks

ICML 2025poster

We establish the asymptotic implicit bias of gradient descent (GD) for generic non-homogeneous deep networks under exponential loss. Specifically, we characterize three key properties of GD iterates starting from a sufficiently small empirical risk, where the threshold is determined by a measure of…

Cited by 0SourcePDFScholar
2024

Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization

NeurIPS 2024poster

The typical training of neural networks using large stepsize gradient descent (GD) under the logistic loss often involves two distinct phases, where the empirical risk oscillates in the first phase but decreases monotonically in the second phase. We investigate this phenomenon in two-layer networks…

Cited by 6SourcePDFScholar
2024

Multimarginal Generative Modeling with Stochastic Interpolants

ICLR 2024poster

Given a set of $K$ probability densities, we consider the multimarginal generative modeling problem of learning a joint distribution that recovers these densities as marginals. The structure of this joint distribution should identify multi-way correspondences among the prescribed marginals. We forma…

Cited by 8SourcePDFScholar