The Geometry of Statistical Feature Learning in Mean-Field Langevin Dynamics

Zong Shang, Tomoya Wakayama, Guillaume Lecué +1

We introduce a geometric formulation of statistical feature learning for supervised regression. Feature learning is defined through a base--fiber decomposition: the base is the fea…