An approach to optimizing abstaining area for small sample data classification

Blaise Hanczar (1) , Jean-Daniel Zucker (2)

1 IBISC - Informatique, Biologie Intégrative et Systèmes Complexes
2 UMMISCO - Unité de modélisation mathématique et informatique des systèmes complexes [Bondy]

Given a classification task, an approach to improve accuracy relies on the use of abstaining classifiers. These classifiers are trained to reject observations for which predicted values are not reliable enough: these rejected observations belong to an abstaining area in the feature space. Two equivalent methods exist to theoretically compute the optimal abstaining area for a given classification problem. The first one is based on the posterior probability computed by the model and the other is based on the derivative of the ROC function of the model. Although the second method has proved to give the best results, in small-sample settings such as the one found in omics data, the estimation of posterior probabilities and derivative of ROC curve are both lacking of precision leading to far from optimal abstaining areas. As a consequence none of the two methods bring the expected improvements in accuracy. We propose five alternative algorithms to compute the abstaining area adapted to small-sample problems. The idea of these algorithms is to compute an accurate and robust estimation of the ROC curve and its derivatives. These estimation are mainly based on the assumption that the distribution of the output of the classifier for each class is normal or mixture of normal distributions. These distributions are estimated by a kernel density estimator or Bayesian semiparametric estimator. Another method works on the approximation of the convex hull of the ROC curve. Once the derivative of the ROC curve are estimated, the optimal abstaining area can be directly computed. The performance of our algorithms are directly related to their capacity to compute an accurate estimation of the ROC curve. A sensitivity analysis of our methods to the dataset size and rejection cost has been done on a set of experiments. We show that our methods improve the performances of the abstaining classifiers on several real datasets and for different learning algorithms.

Mots clés: Abstaining classifier; Reject option; ROC curve estimation; Small-sample setting; Supervised learning

Domaines: Apprentissage [cs.LG]

Blaise Hanczar, Jean-Daniel Zucker. An approach to optimizing abstaining area for small sample data classification. Expert Systems with Applications, 2018, 95, pp.153--161. ⟨10.1016/j.eswa.2017.11.013⟩. ⟨hal-01658150⟩