No Spurious Local Minima in a Two Hidden Unit ReLU Network. - Chenwei Wu 002; Jiajun Luo; Jason D. Lee (ICLR (Workshop), 2018)

Beyond Lazy Training for Over-parameterized Tensor Decomposition. - Xiang Wang 011; Chenwei Wu 002; Jason D. Lee; Tengyu Ma 001; Rong Ge 001 (NeurIPS, 2020)

Secure Data Sharing With Flow Model. - Chenwei Wu 002; Chenzhuang Du; Yang Yuan (CoRR, 2020)

Dissecting Hessian: Understanding Common Structure of Hessian in Neural Networks. - Yikai Wu 001; Xingyu Zhu 003; Chenwei Wu 002; Annie N. Wang; Rong Ge 001 (CoRR, 2020)

Guarantees for Tuning the Step Size using a Learning-to-Learn Approach. - Xiang Wang 011; Shuai Yuan; Chenwei Wu 002; Rong Ge 001 (ICML, 2021)

Towards Understanding the Data Dependency of Mixup-style Training. - Muthu Chidambaram; Xiang Wang 011; Yuzheng Hu; Chenwei Wu 002; Rong Ge 001 (ICLR, 2022)

The Role of Linguistic Priors in Measuring Compositional Generalization of Vision-Language Models. - Chenwei Wu 002; Li Erran Li; Stefano Ermon; Patrick Haffner; Rong Ge 001; Zaiwei Zhang (ICBINB, 2023)

Provably Learning Diverse Features in Multi-View Data with Midpoint Mixup. - Muthu Chidambaram; Xiang Wang 011; Chenwei Wu 002; Rong Ge 001 (ICML, 2023)

Hiding Data Helps: On the Benefits of Masking for Sparse Coding. - Muthu Chidambaram; Chenwei Wu 002; Yu Cheng 002; Rong Ge 001 (ICML, 2023)

Connecting Pre-trained Language Model and Downstream Task via Properties of Representation. - Chenwei Wu 002; Holden Lee; Rong Ge 001 (NeurIPS, 2023)

Theoretical Understanding of Neural Network Optimization Landscape and Self-Supervised Representation Learning. - Chenwei Wu 002 (2023)

Adam-mini: Use Fewer Learning Rates To Gain More. - Yushun Zhang; Congliang Chen; Ziniu Li; Tian Ding; Chenwei Wu 002; Yinyu Ye 001; Zhi-Quan Luo; Ruoyu Sun 001 (CoRR, 2024)

Adam-mini: Use Fewer Learning Rates To Gain More. - Yushun Zhang; Congliang Chen; Ziniu Li; Tian Ding; Chenwei Wu 002; Diederik P. Kingma; Yinyu Ye 001; Zhi-Quan Luo; Ruoyu Sun 001 (ICLR, 2025)