4 papers
Mixing times of Langevin dynamics for spiked matrix models
Reza Gheissari, Curtis Grant, Tianmin Yu
We investigate the Langevin dynamics for Wigner matrices with a spherical spike, in the regime where the signal-to-noise ratio is large, but order one. For large, order-, s…
An entropy formula for the Deep Linear Network
Govind Menon, Tianmin Yu
We study the Riemannian geometry of the Deep Linear Network (DLN) as a foundation for a thermodynamic description of the learning process. The main tools are the use of group actio…
DiffATS: Diffusion in Aligned Tensor Space
Jinhua Lyu, Tianmin Yu, Brian Kim +3
Direct diffusion modeling of high-resolution spatiotemporal fields is computationally challenging. Parameter-efficient primitives address this by representing high-dimensional data…
Scalable Mean-Field Variational Inference via Preconditioned Primal-Dual Optimization
Jinhua Lyu, Tianmin Yu, Ying Ma +1
In this work, we investigate the large-scale mean-field variational inference (MFVI) problem from a mini-batch primal-dual perspective. By reformulating MFVI as a constrained finit…