Provides functions that performs popular stochastic gradient Markov chain Monte Carlo (SGMCMC) methods on user specified models. The required gradients are automatically calculated using 'TensorFlow' , an efficient library for numerical computation. This means only the log likelihood and log prior functions need to be specified. The methods implemented include stochastic gradient Langevin dynamics (SGLD), stochastic gradient Hamiltonian Monte Carlo (SGHMC), stochastic gradient Nose-Hoover thermostat (SGNHT) and their respective control variate versions for increased efficiency. References: M. Welling, Y. W. Teh (2011) ; T. Chen, E. B. Fox, C. E. Guestrin (2014) ; N. Ding, Y. Fang, R. Babbush, C. Chen, R. D. Skeel, H. Neven (2014) ; J. Baker, P. Fearnhead, E. B. Fox, C. Nemeth (2017) . For more details see doi:10.18637/jss.v091.i03.
Citation | sgmcmc citation info |
github.com/STOR-i/sgmcmc | |
System requirements | TensorFlow (https://www.tensorflow.org/), TensorFlow Probability (https://www.tensorflow.org/probability/) |
Bug report | File report |