Proceedings

Fast Laplace Approximation for Sparse Bayesian Spike and Slab Models

Download BibTeX Cite

Proceedings of the Twenty-Fifth International Joint Conference on Artificial Intelligence. Pages 1867-1972.

Abstract

We consider the application of Bayesian spike-and-slab models in high-dimensional feature selection problems. To do so, we propose a simple yet effective fast approximate Bayesian inference algorithm based on Laplace's method. We exploit two efficient optimization methods, GIST and L-BFGS, to obtain the mode of the posterior distribution. Then we propose an ensemble Nystrom based approach to calculate the diagonal of the inverse Hessian over the mode to obtain the approximate posterior marginals in O(knp) time, k ≪ p. Furthermore, we provide the theoretical analysis about the estimation consistency and approximation error bounds. With the posterior marginals of the model weights, we use quadrature integration to estimate the marginal posteriors of selection probabilities and indicator variables for all features, which quantify the selection uncertainty. Our method not only maintains the benefits of the Bayesian treatment (e.g., uncertainty quantification) but also possesses the computational efficiency, and oracle properties of the frequentist methods. Simulation shows that our method estimates better or comparable selection probabilities and indicator variables than alternative approximate inference methods such as VB and EP, but with less running time. Extensive experiments on large real datasets demonstrate that our method often improves prediction accuracy over Bayesian automatic relevance determination, EP, and frequentist L1 type methods.

  1. [Ahmed and Xing, 2009] Amr Ahmed and Eric P Xing. Re- covering time-varying networks of dependencies in so- cial and biological studies. Proceedings of the National Academy of Sciences, 106(29):11878-11883, 2009.
  2. [Bezdek and Hathaway, 2003] James C Bezdek and Richard J Hathaway. Convergence of alternating optimization. Neu- ral, Parallel & Scientific Computations, 11(4):351-368, 2003.
  3. [Candès, 2006] Emmanuel J Candès. Compressive sampling. In Proceedings oh the International Congress of Mathe- maticians: invited lectures, pages 1433-1452, 2006.
  4. [Carbonetto et al., 2012] Peter Carbonetto, Matthew Stephens, et al. Scalable variational inference for bayesian variable selection in regression, and its accuracy in genetic association studies. Bayesian Analysis, 7(1):73-108, 2012.
  5. [Gong et al., 2013] Pinghua Gong, Changshui Zhang, Zhaosong Lu, Jianhua Huang, and Jieping Ye. A general iterative shrinkage and thresholding algorithm for non- convex regularized optimization problems. In The 30th International Conference on Machine Learning, pages 37-45, 2013.
  6. [Hernández-Lobato et al., 2010a] Daniel Hernández-Lobato, José Miguel Hernández-Lobato, Thibault Helleputte, and Pierre Dupont. Expectation propagation for bayesian multi- task feature selection. In Machine Learning and Knowledge Discovery in Databases, pages 522-537. Springer, 2010.
  7. [Hernández-Lobato et al., 2010b] Daniel Hernández-Lobato, José Miguel Hernández-Lobato, and Alberto Suárez. Ex- pectation propagation for microarray data classification. Pattern recognition letters, 31(12):1618-1626, 2010.
  8. [Hernández-Lobato, 2010] José Miguel Hernández-Lobato. Balancing flexibility and robustness in machine learning semi-parametric methods and sparse linear models. PhD thesis, Ph. D. Thesis, Universidad Autó De Madrid, 2010.
  9. [Hoerl and Kennard, 1970] Arthur E Hoerl and Robert W Kennard. Ridge regression: Biased estimation for nonorthogonal problems. Technometrics, 12(1):55-67, 1970.
  10. [Ishwaran and Rao, 2005] Hemant Ishwaran and J Sunil Rao. Spike and slab variable selection: frequentist and bayesian strategies. Annals of Statistics, pages 730-773, 2005.
  11. [Kogan et al., 2009] Shimon Kogan, Dimitry Levin, Bryan R Routledge, Jacob S Sagi, and Noah A Smith. Predicting risk from financial reports with regression. In Proceedings of Human Language Technologies: The 2009 Annual Con- ference of the North American Chapter of the Association for Computational Linguistics, pages 272-280. Association for Computational Linguistics, 2009.
  12. [Kumar et al., 2009] Sanjiv Kumar, Mehryar Mohri, and Ameet Talwalkar. Ensemble nystrom method. In Advances in Neural Information Processing Systems, pages 1060- 1068, 2009.
  13. [Minka, 2000] Thomas P Minka. Deriving quadrature rules from Gaussian processes. Technical report, Statistics De- partment, Carnegie Mellon University, 2000.
  14. [Mitchell and Beauchamp, 1988] Toby J Mitchell and John J Beauchamp. Bayesian variable selection in linear regres- sion. Journal of the American Statistical Association, 83(404):1023-1032, 1988.
  15. [Nocedal, 1980] Jorge Nocedal. Updating quasi-newton ma- trices with limited storage. Mathematics of computation, 35(151):773-782, 1980.
  16. [Qi et al., 2005] Y. Qi, T. S. Jaakkola, and D.K. Gifford. Ap- proximate expectation propagation for bayesian inference on large-scale problems. Technical report, MIT Com- puter Science and Artificial Intelligence Laboratory, Octo- ber 2005.
  17. [Rosenwald et al., 2002] Andreas Rosenwald, George Wright, Wing C Chan, Joseph M Connors, Elias Campo, Richard I Fisher, Randy D Gascoyne, H Konrad Muller- Hermelink, Erlend B Smeland, Jena M Giltnane, et al. The use of molecular profiling to predict survival after chemotherapy for diffuse large-b-cell lymphoma. New England Journal of Medicine, 346(25):1937-1947, 2002.
  18. [Rue et al., 2009] Hravard Rue, Sara Martino, and Nicolas Chopin. Approximate bayesian inference for latent gaussian models by using integrated nested laplace approximations. Journal of the royal statistical society: Series b (statistical methodology), 71(2):319-392, 2009.
  19. [Scheetz et al., 2006] Todd E Scheetz, Kwang-Youn A Kim, Ruth E Swiderski, Alisdair R Philp, Terry A Braun, Kevin L Knudtson, Anne M Dorrance, Gerald F DiBona, Jian Huang, Thomas L Casavant, et al. Regulation of gene ex- pression in the mammalian eye and its relevance to eye dis- ease. Proceedings of the National Academy of Sciences, 103(39):14429-14434, 2006.
  20. [Tibshirani, 1996] Robert Tibshirani. Regression shrinkage and selection via the lasso. Journal of the Royal Statistical Society. Series B (Methodological), pages 267-288, 1996.
  21. [Titsias and Lázaro-Gredilla, 2011] Michalis K Titsias and Miguel Lázaro-Gredilla. Spike and slab variational infer- ence for multi-task and multiple kernel learning. In NIPS, volume 24, pages 2339-2347, 2011.
  22. [Woodbury, 1950] Max A Woodbury. Inverting modified ma- trices. Memorandum report, 42:106, 1950.
  23. [Yen, 2011] Tso-Jung Yen. A majorization-minimization ap- proach to variable selection using spike and slab priors. The Annals of Statistics, 39(3):1748-1775, 2011.
  24. [Zhe et al., 2013] Shandian Zhe, Syed AZ Naqvi, Yifan Yang, and Yuan Qi. Joint network and node selection for pathway- based genomic data analysis. Bioinformatics, 2013.
  25. [Zou and Hastie, 2005] Hui Zou and Trevor Hastie. Regular- ization and variable selection via the elastic net. Journal of the Royal Statistical Society: Series B (Statistical Method- ology), 67(2):301-320, 2005.
  26. [Zou and Zhang, 2009] Hui Zou and Hao Helen Zhang. On the adaptive elastic-net with a diverging number of parame- ters. Annals of statistics, 37(4):1733, 2009.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards