← Search

Chenyu Zheng

7 accepted papers

2025

A Theory for Conditional Generative Modeling on Multiple Data Sources

ICML 2025poster

The success of large generative models has driven a paradigm shift, leveraging massive multi-source data to enhance model capabilities. However, the interaction among these sources remains theoretically underexplored. This paper provides a first attempt to fill this gap by rigorously analyzing multi…

2025

Scaling Diffusion Transformers Efficiently via $\mu$P

NeurIPS 2025poster

Diffusion Transformers have emerged as the foundation for vision generative models, but their scalability is limited by the high cost of hyperparameter (HP) tuning at large scales. Recently, Maximal Update Parametrization ($\mu$P) was proposed for vanilla Transformers, which enables stable HP transf…

Cited by 0SourceScholar
2024

Lower Bounds of Uniform Stability in Gradient-Based Bilevel Algorithms for Hyperparameter Optimization

NeurIPS 2024poster

Gradient-based bilevel programming leverages unrolling differentiation (UD) or implicit function theorem (IFT) to solve hyperparameter optimization (HO) problems, and is proven effective and scalable in practice. To understand their generalization behavior, existing works establish upper bounds on…

Cited by 0SourcePDFScholar
2024

On Mesa-Optimization in Autoregressively Trained Transformers: Emergence and Capability

NeurIPS 2024poster

Autoregressively trained transformers have brought a profound revolution to the world, especially with their in-context learning (ICL) ability to address downstream tasks. Recently, several studies suggest that transformers learn a mesa-optimizer during autoregressive (AR) pretraining to implement…

2024

The Blessing of Randomness: SDE Beats ODE in General Diffusion-based Image Editing

ICLR 2024poster

We present a unified probabilistic formulation for diffusion-based image editing, where a latent variable is edited in a task-specific manner and generally deviates from the corresponding marginal distribution induced by the original stochastic or ordinary differential equation (SDE or ODE). Instead…

2023

Revisiting Discriminative vs. Generative Classifiers: Theory and Implications

ICML 2023poster

A large-scale deep model pre-trained on massive labeled or unlabeled data transfers well to downstream tasks. Linear evaluation freezes parameters in the pre-trained model and trains a linear classifier separately, which is efficient and attractive for transfer. However, little work has investigated…