Sparse Bayesian Learning and the Relevance Vector Machine
Michael E. Tipping
Introduces the Relevance Vector Machine, a sparse Bayesian learning framework that matches the generalization accuracy of support vector machines while producing dramatically sparser models, delivering calibrated probabilistic predictions, and automatically estimating hyperparameters without cross-validation.
Supervised learning tasks such as regression and classification frequently encounter overfitting when models use as many parameters as training examples. Support vector machines address this through margin maximization but require cross-validation for key parameters, produce non-probabilistic outputs, and restrict kernels to those satisfying Mercer's condition. These constraints limit practical deployment where uncertainty estimates or computational efficiency matter.
The article presents a fully Bayesian alternative called the relevance vector machine. It places a hierarchical prior on model weights, with one hyperparameter per weight, and estimates those hyperparameters by maximizing the marginal likelihood of the data. This procedure automatically prunes most weights to zero while delivering predictions and, in regression, error bars. The same framework applies to both regression and classification without requiring separate tuning steps.
Experiments on synthetic functions and standard benchmarks show that relevance vector machines match or exceed support vector machine accuracy while using far fewer basis functions, typically 10 to 20 percent as many. The models also supply well-calibrated posterior probabilities in classification and allow arbitrary basis functions, including direct optimization of input scale parameters within the kernel. These advantages arise because the Bayesian treatment concentrates posterior mass on sparse solutions.
The resulting sparsity reduces memory and prediction cost, while probabilistic outputs enable proper handling of asymmetric losses and class imbalance. Automatic parameter estimation removes the need for cross-validation, simplifying model deployment. For very large training sets the original algorithm scales as the cube of the number of examples, though a constructive variant mitigates this.
Practitioners should evaluate the relevance vector machine when sparse probabilistic models are needed and data volume permits the cubic training cost. Additional work on scaling and on criteria for kernel selection would further broaden applicability. Results rest on consistent performance across repeated trials and multiple data sets, with the main uncertainty lying in behavior on extremely large problems where only the improved algorithm has been tested.
- Paper: Support Vector Regression Machines, Harris Drucker et al. (1996). Introduces support vector regression using insensitive loss functions and margin regularization, providing the primary non-probabilistic benchmark that the Relevance Vector Machine seeks to reformulate and improve upon.
- Paper: Support-vector networks, Corinna Cortes et al. (1995). Establishes the foundational support-vector network architecture for classification, defining the standard margin-maximization and kernel framework that the source replaces with a sparse Bayesian formulation.
- Paper: A training algorithm for optimal margin classifiers, B. Boser et al. (1992). Presents the original optimal margin classifier and dual formulation that motivate the source's pursuit of sparsity and kernel-based pattern recognition.
- Paper: Support Vector Method for Function Approximation, Regression Estimation and Signal Processing, V. Vapnik et al. (1996). Extends support vector machines to high-dimensional function approximation and regression, establishing key baseline concepts for sparse kernel estimation.
- Paper: Keeping Neural Networks Simple by Minimizing the Description Length of the Weights, Geoffrey E. Hinton et al. (1993). Demonstrates how placing adaptive Gaussian priors on parameters penalizes model complexity, foreshadowing the hierarchical prior framework utilized in sparse Bayesian learning.
- Paper: Choosing Multiple Parameters for Support Vector Machines, OLIVIER CHAPELLE et al. (2002). Extends hyperparameter optimization in kernel methods by computing gradients of generalization bounds to tune feature scaling parameters, paralleling the source's capacity for automated parameter selection.
- Paper: Learning the Kernel Matrix with Semidefinite Programming, Gert R. G. Lanckriet et al. (2004). Develops semidefinite programming techniques to automatically optimize kernel matrices directly from data, providing an alternative non-Bayesian solution to the kernel selection challenge highlighted in the source.
- Paper: Feature selection, L1 vs. L2 regularization, and rotational invariance, A. Ng (2004). Provides a theoretical comparison of sparsity-inducing L1 regularization versus L2 regularization under high dimensions, enriching the conceptual understanding of sparse learning mechanisms.
- Paper: An Introduction to Variable and Feature Selection, Isabelle M Guyon et al. (2003). Surveys systematic variable and feature selection techniques, framing the sparse parameter pruning achieved by the Relevance Vector Machine within the broader landscape of model selection.
- Paper: On Over-fitting in Model Selection and Subsequent Selection Bias in Performance Evaluation, G. Cawley et al. (2010). Analyzes the risk of overfitting during hyperparameter selection in kernel machines and Gaussian processes, critically evaluating the automated model selection properties discussed in the source.
- Paper: Random Features for Large-Scale Kernel Machines, Ali Rahimi et al. (2007). Introduces explicit random feature mappings to overcome cubic kernel scaling bottlenecks, directly addressing the computational scaling challenges associated with kernel and relevance vector methods.
- Paper: An empirical comparison of supervised learning algorithms, R. Caruana et al. (2006). Benchmarks probability calibration across diverse learning algorithms, validating the practical significance of the well-calibrated posterior probabilities generated by Bayesian kernel approaches.
