What can a Single Attention Layer Learn? A Study Through the Random Features Lens. - Hengyu Fu; Tianyu Guo 004; Yu Bai 017; Song Mei (NeurIPS, 2023)

Unveil Conditional Diffusion Models with Classifier-free Guidance: A Sharp Statistical Theory. - Hengyu Fu; Zhuoran Yang; Mengdi Wang; Minshuo Chen (CoRR, 2024)

Learning Hierarchical Polynomials of Multiple Nonlinear Features with Three-Layer Networks. - Hengyu Fu; Zihao Wang; Eshaan Nichani; Jason D. Lee (CoRR, 2024)

Diffusion Transformer Captures Spatial-Temporal Dependencies: A Theory for Gaussian Process Data. - Hengyu Fu; Zehao Dou; Jiawei Guo; Mengdi Wang; Minshuo Chen (ICLR, 2025)

Learning Hierarchical Polynomials of Multiple Nonlinear Features. - Hengyu Fu; Zihao Wang; Eshaan Nichani; Jason D. Lee (ICLR, 2025)