Balaji Lakshminarayanan

Hello! I am a research scientist at Google DeepMind. I am interested in Machine Learning and its cool applications.

Quick links

Brief Bio

Before joining DeepMind, I was a PhD student at Gatsby Unit, CSML, University College London, working with Yee Whye Teh. Between 2008-10, I was a Masters student at Oregon State University, working with Raviv Raich as part of the Bioacoustics research group. During my undergrad, I was part of the Integrated Systems laboratory, where I worked on communication subsystems for the Anna University Microsatellite (ANUSAT) project.

Publications

  1. Simple and scalable predictive uncertainty estimation using deep ensembles
    Balaji Lakshminarayanan, Alexander Pritzel and Charles Blundell
    [arXiv]
    Note: A version of this work was presented at the NIPS workshop on Bayesian deep learning, 2016.

  2. Learning in implicit generative models
    Shakir Mohamed and Balaji Lakshminarayanan
    [arXiv]
    Note: A version of this work was presented at the NIPS workshop on adversarial training, 2016.

  3. Learning deep nearest neighbor representations using differentiable boundary trees
    Daniel Zoran, Balaji Lakshminarayanan and Charles Blundell
    [arXiv]

  4. Distributed Bayesian learning with stochastic natural-gradient expectation propagation and the posterior server
    Leonard Hasenclever, Stefan Webb, Thibaut Lienart, Sebastian Vollmer, Balaji Lakshminarayanan, Charles Blundell and Yee Whye Teh
    [arXiv] [code]

  5. Decision trees and forests: a probabilistic perspective
    Balaji Lakshminarayanan
    Ph.D. thesis, University College London, 2016
    [pdf]

  6. The Mondrian kernel
    Matej Balog, Balaji Lakshminarayanan, Zoubin Ghahramani, Daniel M. Roy and Yee Whye Teh
    UAI, 2016
    [arXiv] [code] [slides] [poster]

  7. Approximate inference with the variational Holder bound
    Guillaume Bouchard and Balaji Lakshminarayanan
    [arXiv]

  8. Mondrian forests for large-scale regression when uncertainty matters
    Balaji Lakshminarayanan, Daniel M. Roy and Yee Whye Teh
    AISTATS, 2016
    [pdf] [code] [slides]

  9. Kernel-based just-in-time learning for passing expectation propagation messages
    Wittawat Jitkrittum, Arthur Gretton, Nicolas Heess, Ali Eslami, Balaji Lakshminarayanan, Dino Sejdinovic and Zoltan Szabo
    UAI, 2015
    [arXiv] [code]

  10. Particle Gibbs for Bayesian additive regression trees
    Balaji Lakshminarayanan, Daniel M. Roy and Yee Whye Teh
    AISTATS, 2015
    [pdf] [code]

  11. Mondrian forests: Efficient online random forests
    Balaji Lakshminarayanan, Daniel M. Roy and Yee Whye Teh
    NIPS, 2014
    [pdf] [code] [slides]

  12. Distributed Bayesian posterior sampling via moment sharing
    Minjie Xu, Balaji Lakshminarayanan, Yee Whye Teh, Jun Zhu and Bo Zhang
    NIPS, 2014
    [pdf] [code]

  13. Latent IBP compound Dirichlet allocation
    Cedric Archambeau, Balaji Lakshminarayanan, and Guillaume Bouchard
    TPAMI special issue on Bayesian Nonparametrics, 2015
    [pdf] [code available upon request] [IEEE link]
    Note: A short version of this work appeared at the NIPS workshop on Bayesian nonparametrics, 2011.

  14. Inferring ground truth from multi-annotator ordinal data: a probabilistic approach
    Balaji Lakshminarayanan and Yee Whye Teh
    [arXiv] [code]

  15. Top-down particle filtering for Bayesian decision trees
    Balaji Lakshminarayanan, Daniel M. Roy and Yee Whye Teh
    ICML, 2013
    [pdf] [code] [slides]

  16. Acoustic classification of multiple simultaneous bird species: A multi-instance multi-label approach
    Forrest Briggs, Balaji Lakshminarayanan, Lawrence Neal, Xiaoli Z. Fern, Raviv Raich, Sarah J. K. Hadley, Adam S. Hadley, and Matthew G. Betts
    JASA, 2012
    [link]

  17. Inference in supervised latent Dirichlet allocation
    Balaji Lakshminarayanan and Raviv Raich
    MLSP, 2011
    [pdf] [code available upon request] [IEEE link]

  18. Robust Bayesian matrix factorization
    Balaji Lakshminarayanan, Guillaume Bouchard, and Cedric Archambeau
    AISTATS, 2011
    [revised pdf] [code available upon request]
    Note: the updates for a_n, c_m were wrong in the original version of the pdf. The increments ought to be ell_n and ell_m respectively instead of 1.

  19. Probabilistic models for classification of bioacoustic data
    Balaji Lakshminarayanan
    M.S. thesis, Oregon State University, 2010
    [pdf]

  20. Non-negative matrix factorization for parameter estimation in Hidden Markov models
    Balaji Lakshminarayanan and Raviv Raich
    MLSP, 2010
    [pdf] [IEEE link]

  21. A syllable-level probabilistic framework for bird species identification
    Balaji Lakshminarayanan, Raviv Raich and Xiaoli Z. Fern
    ICMLA, 2009
    [pdf] [IEEE link]

Copyright notice: This material is presented to ensure timely dissemination of scholarly and technical work. Copyright and all rights therein are retained by authors or by other copyright holders. Personal use of this material is permitted. Permission must be obtained from the copyright holder for all other uses.

About me

Education

Industry Experience

Teaching