← Search

Kulin Shah

9 accepted papers

2025

Does Generation Require Memorization? Creative Diffusion Models using Ambient Diffusion

ICML 2025poster

There is strong empirical evidence that the stateof-the-art diffusion modeling paradigm leads to models that memorize the training set, especially when the training set is small. Prior methods to mitigate the memorization problem often lead to decrease in image quality. Is it possible to obtain stro…

Cited by 0SourcePDFScholar
2025

Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusions

ICML 2025oral

In recent years, masked diffusion models (MDMs) have emerged as a promising alternative approach for generative modeling over discrete domains. Compared to autoregressive models (ARMs), MDMs trade off complexity at training time with flexibility at inference time. At training time, they must learn t…

Cited by 4SourcePDFScholar
2024

Causal language modeling can elicit search and reasoning capabilities on logic puzzles

NeurIPS 2024poster

Causal language modeling using the Transformer architecture has yielded remarkable capabilities in Large Language Models (LLMs) over the last few years. However, the extent to which fundamental search and reasoning capabilities emerged within LLMs remains a topic of ongoing debate. In this work, we…

2024

Unrolled denoising networks provably learn to perform optimal Bayesian inference

NeurIPS 2024poster

Much of Bayesian inference centers around the design of estimators for inverse problems which are optimal assuming the data comes from a known prior. But what do these optimality guarantees mean if the prior is unknown? In recent years, algorithm unrolling has emerged as deep learning's answer to th…

Cited by 0SourcePDFScholar
2023

Ambient Diffusion: Learning Clean Distributions from Corrupted Data

NeurIPS 2023poster

We present the first diffusion-based framework that can learn an unknown distribution using only highly-corrupted samples. This problem arises in scientific applications where access to uncorrupted samples is impossible or expensive to acquire. Another benefit of our approach is the ability to train…

2022

Learning and Generalization in Overparameterized Normalizing Flows

AISTATS 2022poster

In supervised learning, it is known that overparameterized neural networks with one hidden layer provably and efficiently learn and generalize, when trained using stochastic gradient descent with a sufficiently small learning rate and suitable initialization. In contrast, the benefit of overparamete…