Krishnakumar Balasubramanian, Ph.D.

Krishna Balasubramanian

Associate Professor

  • Statistics
4109 Mathematical Sciences Building
One Shields Avenue, Davis CA 95616
Research Interests & Expertise
  • Deep learning Theory
  • Sampling and Stochastic Optimization
  • Geometric, Topological and Nonparametric statistics
Publications
  • Functional linear and single-index models: A unified approach via Gaussian Stein identity
    Journal Article | Balasubramanian, K., Müller, H., Sriperumbudur, B. Bernoulli, 31, 973-1006. 10.3150/24-bej1755
  • IMPROVED FINITE-PARTICLE CONVERGENCE RATES FOR STEIN VARIATIONAL GRADIENT DESCENT
    Conference Paper | Banerjee, S., Balasubramanian, K., Ghosal, P. 13th International Conference on Learning Representations Iclr 2025, 76764-76785.
  • Online Covariance Estimation in Nonsmooth Stochastic Approximation
    Conference Paper | Jiang, L., Roy, A., Balasubramanian, K., Davis, D., Drusvyatskiy, D., Na, S. Proceedings of Machine Learning Research, 291,
  • Regularized Stein Variational Gradient Flow
    Journal Article | He, Y., Balasubramanian, K., Sriperumbudur, B., Lu, J. Foundations of Computational Mathematics, 25, 1199-1257. 10.1007/s10208-024-09663-w
  • TRANSFORMERS HANDLE ENDOGENEITY IN IN-CONTEXT LINEAR REGRESSION
    Conference Paper | Liang, H., Balasubramanian, K., Lai, L. 13th International Conference on Learning Representations Iclr 2025, 70972-71010.
  • See publications and more on Aggie Experts: https://experts.ucdavis.edu/expert/3Fvvdh2f