F. Kuusisto, V. Santos Costa, H. Nassif, E. Burnside, D. Page & J. Shavlik (2014).
Support Vector Machines for Differential Prediction. Proceedings of the European Conference on Machine Learning (ECML).
Poster (PDF). Slides (PDF). Code. Data.

This publication is contained in the following 3 PDF files
File 1, File 2, File 3.

The software and data associated with this publication are available online.

Abstract:

Machine learning is continually being applied to a growing set of fields, including the social sciences, business, and medicine. Some fields present problems that are not easily addressed using standard machine learning approaches and, in particular, there is growing interest in differential prediction. In this type of task we are interested in producing a classifier that specifically characterizes a subgroup of interest by maximizing the difference in predictive performance for some outcome between subgroups in a population. We discuss adapting maximum margin classifiers for differential prediction. We first introduce multiple approaches that do not affect the key properties of maximum margin classifiers, but which also do not directly attempt to optimize a standard measure of differential prediction. We next propose a model that directly optimizes a standard measure in this field, the uplift measure. We evaluate our models on real data from two medical applications and show excellent results.

Return to the publications of the Univ. of Wisconsin Machine Learning Research Group.