Center for Data Science, New York University and Center for Computational Mathematics, Flatiron Institute; IHPC, Agency for Science, Technology and Research, Singapore and CFAR, Agency for Science, Technology and Research, Singapore; Department of Mathematical Informatics, the University of Tokyo and Center for Advanced Intelligence Project, RIKEN
Atsushi Nitanda, Kazusato Oko, Taiji Suzuki, Denny Wu
neural network optimization;feature learning;mean-field Langevin dynamics
Improved statistical and computational complexity of the mean-field Langevin dynamics under structured data