An Efficient Minibatch Acceptance Test for Metropolis-Hastings
Seita, Daniel, Pan, Xinlei, Chen, Haoyu, Canny, John
We present a novel Metropolis-Hastings method for large datasets that uses small expected-size minibatches of data. Previous work on reducing the cost of Metropolis-Hastings tests yield variable data consumed per sample, with only constant factor reductions versus using the full dataset for each sample. Here we present a method that can be tuned to provide arbitrarily small batch sizes, by adjusting either proposal step size or temperature. Our test uses the noise-tolerant Barker acceptance test with a novel additive correction variable. The resulting test has similar cost to a normal SGD update. Our experiments demonstrate several order-of-magnitude speedups over previous work.
Jul-9-2017
- Country:
- North America > United States
- California
- Alameda County > Berkeley (0.14)
- Santa Clara County > Mountain View (0.04)
- Massachusetts > Middlesex County
- Cambridge (0.04)
- California
- North America > United States
- Genre:
- Research Report (1.00)