Applied Mathematics

Kyle Loh

I am a fourth-year joint B.S.–M.S. student at Florida Atlantic University studying applied mathematics, where I work with Jason Mireles-James and Yu Xiang.

My research focuses on the mathematical foundations of machine learning, and on using those foundations to build better algorithms.

Research

Rigorous numerics for optimizers

I use computer-assisted proofs to study quasiperiodic and chaotic structures, and routes to chaos, in adaptive optimizers such as RMSProp and Adam. My goal is to understand which mechanisms in an adaptive optimizer drive chaotic behavior.

Our recent RMSProp paper proves that, for positive-definite quadratic losses, trajectories eventually enter a compact trapping region even when the learning rate makes the minimizer unstable. We derive explicit bounds on this region and use numerical continuation to study invariant circles, tori, and chaotic motion within it.

RMSProp bifurcation diagram and finite-time Lyapunov exponent as the learning rate increases, with two enlarged parameter windows.
RMSProp dynamics as the learning rate increases.

Conformal inference

I am developing conformal inference frameworks that bound the false discovery rate under practical constraints: decentralization, limited communication bandwidth, and non-exchangeability.

Publications

Google Scholar

Awards