Events

DMS Applied and Computational Mathematics Seminar

Time: Apr 04, 2025 (02:00 PM)
Location: 328 Parker Hall

Details:

moleitao

Speaker: Molei Tao (Georgia Tech)  
 
Title: Optimization, Sampling, and Generative Modeling in Non-Euclidean Spaces


Abstract: Machine learning in non-Euclidean spaces have been rapidly attracting attention in recent years, and this talk will give some examples of progress on its mathematical and algorithmic foundations. A sequence of developments that eventually leads to the generative modeling of data on Lie groups will be reported. Such a problem occurs, for example, in the Gen-AI design of molecules.

More precisely, I will begin with variational optimization, which, together with delicate interplays between continuous- and discrete-time dynamics, enables the construction of momentum-accelerated algorithms that optimize functions defined on manifolds. Selected applications, such as a generic improvement of Transformer, and a low-dim. approximation of high-dim. optimal transport distance, will be described. Then I will turn the optimization dynamics into an algorithm that samples from probability distributions on Lie groups. This sampler provably converges, even without log-concavity condition or its common relaxations. Finally, I will describe how this sampler can lead to a structurally-pleasant diffusion generative model that allows users to, given training data that follow any latent statistical distribution on a Lie group manifold, generate more data exactly on the same manifold that follow the same distribution. If time permits, applications such as molecule design and generative innovation of quantum processes will be briefly discussed.

Short bio: 
Molei Tao is a full professor in School of Math at Georgia Tech, working on the mathematical foundations of machine learning. He received B.S. from Tsinghua Univ. and Ph.D. from Caltech, and worked as a Courant Instructor at NYU before starting at Georgia Tech. He serves as an Area Chair for NeurIPS, ICLR and ICML, and he is a recipient of W.P. Carey Ph.D. Prize in Applied Mathematics (2011), American Control Conference Best Student Paper Finalist (2013), NSF CAREER Award (2019), AISTATS best paper award (2020), IEEE EFTF-IFCS Best Student Paper Finalist (2021), Cullen-Peck Scholar Award (2022), GT-Emory AI.Humanity Award (2023), SONY Faculty Innovation Award (2024), Best Poster Award at an international conference “Recent Advances and Future Directions for Sampling” held at Yale (2024), as well as several other recognitions.