Publications

Papers, chapters, preprints, and my doctoral thesis, ordered from newest to oldest.

Years indicate publication or first preprint appearance.

Google Scholar ↗

Propagation of Chaos in Contextual Flow Maps

Shi Chen, Zhengjiang Lin, Kaizhao Liu, Philippe Rigollet

Accepted · NeurIPS 2026 [Conference] [arXiv]
Scaling Limits of Long-Context Transformers

Giuseppe Bruno, Shi Chen, Zhengjiang Lin, Yury Polyanskiy, Philippe Rigollet

Accepted · NeurIPS 2026 [Conference] [arXiv]
Critical Attention Scaling in Long-Context Transformers

Shi Chen, Zhengjiang Lin, Yury Polyanskiy, Philippe Rigollet

ICLR 2026 [Conference] [arXiv] [OpenReview] [Poster]
Correcting Autodifferentiation in Neural ODE Training

Yewei Xu, Shi Chen, Qin Li

SIAM Journal on Applied Mathematics 86(5), 2360–2378 · 2026 [Journal] [arXiv]
Quantitative Clustering in Mean-Field Transformer Models

Shi Chen, Zhengjiang Lin, Yury Polyanskiy, Philippe Rigollet

Accepted · Courant Journal of Pure and Applied Mathematics; arXiv:2504.14697 · 2025 [arXiv]
Residual Connections Provably Mitigate Oversmoothing in Graph Neural Networks

Ziang Chen, Zhengjiang Lin, Shi Chen, Yury Polyanskiy, Philippe Rigollet

Accepted · Journal of Machine Learning Research; arXiv:2501.00762 · 2025 [arXiv]
Accelerating Optimization over the Space of Probability Measures

Shi Chen, Qin Li, Oliver Tse, Stephen J. Wright

Journal of Machine Learning Research 26(31), 1–40 · 2025 [Journal] [arXiv]
Multiscale Numerical Methods for Elliptic and Wave-Type PDEs and Their Inverse Problems

Shi Chen

Doctoral thesis · University of Wisconsin–Madison, 2024 [Thesis]
Bayesian Sampling Using Interacting Particles

Shi Chen, Zhiyan Ding, Qin Li

Book chapter · Active Particles, Volume 4: Theory, Models, Applications, 175–215 · 2024 [Book chapter] [arXiv]
A Good Score Does Not Lead to a Good Generative Model

Sixu Li, Shi Chen, Qin Li

Preprint · arXiv:2401.04856 · 2024 [arXiv] [GitHub]
A Reduced Order Schwarz Method for Nonlinear Multiscale Elliptic Equations Based on Two-Layer Neural Networks

Shi Chen, Zhiyan Ding, Qin Li and Stephen J. Wright

Journal of Computational Mathematics 42(2), 570–596 · 2024 [Journal] [arXiv] [GitHub]
On Optimal Bases for Multiscale PDEs and Bayesian Homogenization

Shi Chen, Zhiyan Ding, Qin Li and Stephen J. Wright

Preprint · arXiv:2305.12303 · 2023 [arXiv]
Learning Harmonic Molecular Representations on Riemannian Manifold

Yiqun Wang, Yuning Shen, Shi Chen, Lihao Wang, Fei Ye, Hao Zhou

ICLR 2023, poster [Conference] [arXiv] [OpenReview] [GitHub] [Poster]
High-Frequency Limit of the Inverse Scattering Problem: Asymptotic Convergence from Inverse Helmholtz to Inverse Liouville

Shi Chen, Zhiyan Ding, Qin Li, Leonardo Zepeda-Núñez

SIAM Journal on Imaging Sciences 16(1), 111–143 · 2023 [Journal] [arXiv] [GitHub] [Poster]
Overparameterization of Deep ResNet: Zero Loss and Mean-Field Analysis

Zhiyan Ding, Shi Chen, Qin Li and Stephen J. Wright

Journal of Machine Learning Research 23(48), 1–65 · 2022 [Journal] [arXiv]
Low-Rank Approximation for Multiscale PDEs

Ke Chen, Shi Chen, Qin Li, Jianfeng Lu, and Stephen J. Wright

Notices of the American Mathematical Society 69(6) · 2022 [Journal] [arXiv]
Manifold Learning and Nonlinear Homogenization

Shi Chen, Qin Li, Jianfeng Lu, and Stephen J. Wright

Multiscale Modeling & Simulation 20(3), 1093–1126 · 2022 [Journal] [arXiv] [GitHub]
On the Global Convergence of Gradient Descent for Multi-Layer ResNets in the Mean-Field Regime

Zhiyan Ding, Shi Chen, Qin Li and Stephen J. Wright

Preprint · arXiv:2110.02926 · 2021 [arXiv]
Semiclassical Limit of an Inverse Problem for the Schrödinger Equation

Shi Chen and Qin Li

Research in the Mathematical Sciences 8, article 39 · 2021 [Journal] [arXiv]
Classical Limit for the Varying-Mass Schrödinger Equation with Random Inhomogeneities

Shi Chen, Qin Li and Xu Yang

Journal of Computational Physics 438, article 110365 · 2021 [Journal] [arXiv]
State-Specific Projection of COVID-19 Infection in the United States and Evaluation of Three Major Control Measures

Shi Chen, Qin Li, Song Gao, Yuhao Kang and Xun Shi

Scientific Reports 10, article 22429 · 2020 [Journal] [medRxiv] [GitHub] [Poster]
How Implementation of Entropy in Driving Structural Ordering of Nanoparticles Relates to Assembly Kinetics: Insight into Reaction-Induced Interfacial Assembly of Janus Nanoparticles

Ye Yang, Pengyu Chen, Yufei Cao, Zihan Huang, Guolong Zhu, Ziyang Xu, Xiaobin Dai, Shi Chen, Bing Miao, and Li-Tang Yan

Langmuir 34(32), 9477–9488 · 2018 [Journal]

Publications

Propagation of Chaos in Contextual Flow Maps

Scaling Limits of Long-Context Transformers

Critical Attention Scaling in Long-Context Transformers

Correcting Autodifferentiation in Neural ODE Training

Quantitative Clustering in Mean-Field Transformer Models

Residual Connections Provably Mitigate Oversmoothing in Graph Neural Networks

Accelerating Optimization over the Space of Probability Measures

Multiscale Numerical Methods for Elliptic and Wave-Type PDEs and Their Inverse Problems

Bayesian Sampling Using Interacting Particles

A Good Score Does Not Lead to a Good Generative Model

A Reduced Order Schwarz Method for Nonlinear Multiscale Elliptic Equations Based on Two-Layer Neural Networks

On Optimal Bases for Multiscale PDEs and Bayesian Homogenization

Learning Harmonic Molecular Representations on Riemannian Manifold

High-Frequency Limit of the Inverse Scattering Problem: Asymptotic Convergence from Inverse Helmholtz to Inverse Liouville

Overparameterization of Deep ResNet: Zero Loss and Mean-Field Analysis

Low-Rank Approximation for Multiscale PDEs

Manifold Learning and Nonlinear Homogenization

On the Global Convergence of Gradient Descent for Multi-Layer ResNets in the Mean-Field Regime

Semiclassical Limit of an Inverse Problem for the Schrödinger Equation

Classical Limit for the Varying-Mass Schrödinger Equation with Random Inhomogeneities

State-Specific Projection of COVID-19 Infection in the United States and Evaluation of Three Major Control Measures

How Implementation of Entropy in Driving Structural Ordering of Nanoparticles Relates to Assembly Kinetics: Insight into Reaction-Induced Interfacial Assembly of Janus Nanoparticles