Kexin Jin's Home Page
Contact Info:
Email: kexinj at math dot princeton dot edu
Office phone: 609-258-2322
Fine Hall, Room 409, Washington Road, Princeton NJ 08544-1000 USA
I am a fifth year Ph.D. student in the Department of Mathematics at Princeton University. I am very fortunate to be advised by Professor Alexandru Ionescu and Professor Tristan Buckmaster.
My current research focuses on nonlinear dispersive equations and their wave turbulence theory. I am also interested in the continuous-time analysis of stochastic optimizations with applications in machine learning.
Previously, I completed my M.S. in mathematics in 2018 from the Courant Institute, NYU under the supervision of Professor Jalal Shatah.
Research
How to beat a Bayesian adversary (joint with Zihan Ding, Jonas Latz, Chenguang Liu), submitted, 2024.
Subsampling Error in Stochastic Gradient Langevin Diffusions (joint with Chenguang Liu and Jonas Latz), AISTATS, 2024.
AdaNPC: Exploring Non-Parametric Classifier for Test-Time Adaptation (joint with Yifan Zhang, Xue Wang, Kun Yuan, Zhang Zhang, Liang Wang, Rong Jin, Tieniu Tan), ICML, 2023.
DSGD-CECA: Decentralized SGD with Communication-Optimal Exact Consensus Algorithm (joint with Lisang Ding, Bicheng Ying, Kun Yuan, Wotao Yin), ICML, 2023.
Losing momentum in continuous-time stochastic optimisation (joint with Jonas Latz, Chenguang Liu, and Alessandro Scagliotti), submitted, 2022.
Communication-efficient topologies for decentralized learning with O(1) consensus rate (joint with Zhuoqing Song, Weijian Li, Lei Shi, Ming Yan, Wotao Yin, Kun Yuan), NeurIPS 2022.
A continuous-time stochastic gradient descent method for continuous data (joint with Jonas Latz, Chenguang Liu, and Carola-Bibiane Schönlieb), Journal of Machine Learning Research (JMLR), 2023.
Cancellations of resonances and long time dynamics of cubic Schrödinger equation on 𝕋 (joint with Xiao Ma), Communications in Mathematical Physics, 2021.
Talks
Academy of Mathematics and Systems Science (AMSS), Chinese Academy of Sciences (CAS), Beijing, December 2021.
Student-Run Analysis & PDE Seminar, Department of Mathematics, UC Davis, May 2021.
Discussion Group, Department of Physics, Princeton University, November 2019.
Clifford Lectures Fall 2019: Laure Saint Raymond, Tulane University, New Orleans, October 2019.
Teaching Experience
Teaching assistant, One Variable Calculus with Proofs, Princeton University, Fall 2024
Grader, Complex Analysis with Applications, Princeton University, Spring 2024
Grader, One Variable Calculus with Proofs, Princeton University, Fall 2023
Teaching assistant, Single Variable Analysis with an Introduction to Proofs, Princeton University, Spring 2023
Teaching assistant, Multivariable Calculus, Princeton University, Fall 2022
Lecturer, Graduate Students Seminar on Differential Equations, Tsinghua University, Spring 2022
Lecturer, Graduate Students Seminar on Differential Equations, Tsinghua University, Fall 2021
Teaching assistant, Single Variable Analysis with an Introduction to Proofs, Princeton University, Spring 2021
Teaching assistant, Introduction to Real Analysis, Princeton University, Fall 2020
Teaching assistant, Honors Analysis (Single Variable), Princeton University, Spring 2020
Teaching assistant, Analysis, NYU, Spring 2018
Teaching assistant, Partial Differential Equations (Graduate level), NYU, Spring 2018
Teaching assistant, Introduction to Partial Differential Equations (Graduate level), NYU, Fall 2017