← Search

Khashayar Gatmiry

11 accepted papers

2025

Rethinking Invariance in In-context Learning

ICLR 2025poster

In-Context Learning (ICL) has emerged as a pivotal capability of auto-regressive large language models, yet it is hindered by a notable sensitivity to the ordering of context examples regardless of their mutual independence. To address this issue, recent studies have introduced several variant algor…

2024

Can Looped Transformers Learn to Implement Multi-step Gradient Descent for In-context Learning?

ICML 2024poster

Transformers to do reasoning and few-shot learning, without any fine-tuning, is widely conjectured to stem from their ability to implicitly simulate a multi-step algorithms -- such as gradient descent -- with their weights in a single forward pass. Recently, there has been progress in understanding…

Cited by 18SourcePDFScholar
2024

EM for Mixture of Linear Regression with Clustered Data

AISTATS 2024poster

Modern data-driven and distributed learning frameworks deal with diverse massive data generated by clients spread across heterogeneous environments. Indeed, data heterogeneity is a major bottleneck in scaling up many distributed learning paradigms. In many settings however, heterogeneous data may be…

Cited by 1SourcePDFScholar
2024

Simplicity Bias via Global Convergence of Sharpness Minimization

ICML 2024poster

The remarkable generalization ability of neural networks is usually attributed to the implicit bias of SGD, which often yields models with lower complexity using simpler (e.g. linear) and low-rank features. Recent works have provided empirical and theoretical evidence for the bias of particular vari…

Cited by 1SourcePDFScholar
2024

What does guidance do? A fine-grained analysis in a simple setting

NeurIPS 2024poster

The use of guidance in diffusion models was originally motivated by the premise that the guidance-modified score is that of the data distribution tilted by a conditional likelihood raised to some power. In this work we clarify this misconception by rigorously proving that guidance fails to sample fr…

Cited by 10SourcePDFScholar
2023

Projection-Free Online Convex Optimization via Efficient Newton Iterations

NeurIPS 2023poster

This paper presents new projection-free algorithms for Online Convex Optimization (OCO) over a convex domain $\mathcal{K} \subset \mathbb{R}^d$. Classical OCO algorithms (such as Online Gradient Descent) typically need to perform Euclidean projections onto the convex set $\mathcal{K}$ to ensure feas…

Cited by 1SourcePDFScholar
2023

What is the Inductive Bias of Flatness Regularization? A Study of Deep Matrix Factorization Models

NeurIPS 2023poster

Recent works on over-parameterized neural networks have shown that the stochasticity in optimizers has the implicit regularization effect of minimizing the sharpness of the loss function (in particular, the trace of its Hessian) over the family zero-loss solutions. More explicit forms of flatness r…

Cited by 16SourcePDFScholar
2022

On the generalization of learning algorithms that do not converge

NeurIPS 2022accept

Generalization analyses of deep learning typically assume that the training converges to a fixed point. But, recent results indicate that in practice, the weights of deep neural networks optimized with stochastic gradient descent often oscillate indefinitely. To reduce this discrepancy between theor…

Cited by 15SourcePDFScholar
2022

Optimization and Adaptive Generalization of Three layer Neural Networks

ICLR 2022poster

While there has been substantial recent work studying generalization of neural networks, the ability of deep nets in automating the process of feature extraction still evades a thorough mathematical understanding. As a step toward this goal, we analyze learning and generalization of a three-laye…

Cited by 3SourcePDFScholar