Ghosh, Avrajit

5 publications

ICLR 2025 Learning Dynamics of Deep Matrix Factorization Beyond the Edge of Stability Avrajit Ghosh, Soo Min Kwon, Rongrong Wang, Saiprasad Ravishankar, Qing Qu

NeurIPS 2025 Variational Learning Finds Flatter Solutions at the Edge of Stability Avrajit Ghosh, Bai Cong, Rio Yokota, Saiprasad Ravishankar, Rongrong Wang, Molei Tao, Mohammad Emtiyaz Khan, Thomas Möllenhoff

TMLR 2024 Improving Generalization of Complex Models Under Unbounded Loss Using PAC-Bayes Bounds Xitong Zhang, Avrajit Ghosh, Guangliang Liu, Rongrong Wang

ICML 2024 Optimal Eye Surgeon: Finding Image Priors Through Sparse Generators at Initialization Avrajit Ghosh, Xitong Zhang, Kenneth K. Sun, Qing Qu, Saiprasad Ravishankar, Rongrong Wang

ICLR 2023 Implicit Regularization in Heavy-Ball Momentum Accelerated Stochastic Gradient Descent Avrajit Ghosh, He Lyu, Xitong Zhang, Rongrong Wang