publications

Journal Articles

2025

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

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

   Advances in Neural Information Processing Systems, 2025

   arXiv Bib

   @article{cai2024large,
     title = {Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization},
     author = {Cai, Yuhang and Wu, Jingfeng and Mei, Song and Lindsey, Michael and Bartlett, Peter L.},
     year = {2025},
     journal = {Advances in Neural Information Processing Systems},
   }

2024

1. Learning spiking neuronal networks with artificial neural networks: neural oscillations

   Ruilin Zhang, Zhongyi Wang, Tianyi Wu, Yuhang Cai, Louis Tao, and 2 more authors

   Journal of Mathematical Biology, 2024

   arXiv Bib

   @article{zhang2024learning,
     title = {Learning spiking neuronal networks with artificial neural networks: neural oscillations},
     author = {Zhang, Ruilin and Wang, Zhongyi and Wu, Tianyi and Cai, Yuhang and Tao, Louis and Xiao, Zhuo-Cheng and Li, Yao},
     journal = {Journal of Mathematical Biology},
     volume = {88},
     number = {6},
     pages = {65},
     year = {2024},
     publisher = {Springer Berlin Heidelberg Berlin/Heidelberg},
   }

2023

1. Multi-band oscillations emerge from a simple spiking network

   Tianyi Wu, Yuhang Cai, Ruilin Zhang, Zhongyi Wang, Louis Tao, and 1 more author

   Chaos: An Interdisciplinary Journal of Nonlinear Science, 2023

   arXiv Bib

   @article{wu2023multi,
     title = {Multi-band oscillations emerge from a simple spiking network},
     author = {Wu, Tianyi and Cai, Yuhang and Zhang, Ruilin and Wang, Zhongyi and Tao, Louis and Xiao, Zhuo-Cheng},
     journal = {Chaos: An Interdisciplinary Journal of Nonlinear Science},
     volume = {33},
     number = {4},
     year = {2023},
     publisher = {AIP Publishing},
   }

2022

1. Distances between probability distributions of different dimensions

   Yuhang Cai, and Lek-Heng Lim

   IEEE Transactions on Information Theory, 2022

   arXiv Bib

   @article{cai2022distances,
     title = {Distances between probability distributions of different dimensions},
     author = {Cai, Yuhang and Lim, Lek-Heng},
     journal = {IEEE Transactions on Information Theory},
     volume = {68},
     number = {6},
     pages = {4020--4031},
     year = {2022},
     publisher = {IEEE}
   }

2021

1. Model reduction captures stochastic gamma oscillations on low-dimensional manifolds

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

   Frontiers in Computational Neuroscience, 2021

   arXiv Bib

   @article{cai2021model,
     title = {Model reduction captures stochastic gamma oscillations on low-dimensional manifolds},
     author = {Cai, Yuhang and Wu, Tianyi and Tao, Louis and Xiao, Zhuo-Cheng},
     journal = {Frontiers in Computational Neuroscience},
     volume = {15},
     pages = {678688},
     year = {2021},
     publisher = {Frontiers},
   }

preprint

2025

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

   Kazusato Oko, Licong Lin, Yuhang Cai, and Song Mei

   2025

   arXiv Bib

   @preprint{oko2025multimodal,
     title = {A Statistical Theory of Contrastive Pre-training and Multimodal Generative AI},
     author = {Oko, Kazusato and Lin, Licong and Cai, Yuhang and Mei, Song},
     year = {2025},
     journal = {arXiv preprint arXiv:2501.04641},
     eprint = {2501.04641},
   }