Publications

Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization

Yuhang Cai, Jingfeng Wu, Song Mei, Michael Lindsey, Peter L. Bartlett

Advances in Neural Information Processing Systems 2025. arXiv:2406.08654

---

Toward optimal-scaling DFT: stochastic Hartree theory in the thermodynamic and complete basis set limits at arbitrary temperature

Yuhang Cai, Michael Lindsey

arXiv preprint arXiv:2504.15816 2025. arXiv:2504.15816

---

Implicit Bias of Gradient Descent for Non-Homogeneous Deep Networks

Yuhang Cai, Kangjie Zhou, Jingfeng Wu, Song Mei, Michael Lindsey, Peter L Bartlett

arXiv preprint arXiv:2502.16075 2025. arXiv:2502.16075

---

A Statistical Theory of Contrastive Pre-training and Multimodal Generative AI

Kazusato Oko, Licong Lin, Yuhang Cai, Song Mei

arXiv preprint arXiv:2501.04641 2025. arXiv:2501.04641

---

Learning spiking neuronal networks with artificial neural networks: neural oscillations

Ruilin Zhang, Zhongyi Wang, Tianyi Wu, Yuhang Cai, Louis Tao, Zhuo-Cheng Xiao, Yao Li

Journal of Mathematical Biology 2024. arXiv:2211.11169

---

Multi-band oscillations emerge from a simple spiking network

Tianyi Wu, Yuhang Cai, Ruilin Zhang, Zhongyi Wang, Louis Tao, Zhuo-Cheng Xiao

Chaos: An Interdisciplinary Journal of Nonlinear Science 2023. arXiv:2206.14942

---

Distances between probability distributions of different dimensions

Yuhang Cai, Lek-Heng Lim

IEEE Transactions on Information Theory 2022. arXiv:2011.00629

---

Model reduction captures stochastic gamma oscillations on low-dimensional manifolds

Yuhang Cai, Tianyi Wu, Louis Tao, Zhuo-Cheng Xiao

Frontiers in Computational Neuroscience 2021. arXiv:2101.01699

---