The data representativeness criterion: Predicting the performance of supervised classification based on data set similarity

Evelien Schat (Corresponding author), Rens van de Schoot, Wouter M. Kouw, Duco Veen, Adriënne M. Mendrik

Research output: Contribution to journalArticleAcademicpeer-review

16 Citations (Scopus)

Abstract

In a broad range of fields it may be desirable to reuse a supervised classification algorithm and apply it to a new data set. However, generalization of such an algorithm and thus achieving a similar classification performance is only possible when the training data used to build the algorithm is similar to new unseen data one wishes to apply it to. It is often unknown in advance how an algorithm will perform on new unseen data, being a crucial reason for not deploying an algorithm at all. Therefore, tools are needed to measure the similarity of data sets. In this paper, we propose the Data Representativeness Criterion (DRC) to determine how representative a training data set is of a new unseen data set. We present a proof of principle, to see whether the DRC can quantify the similarity of data sets and whether the DRC relates to the performance of a supervised classification algorithm. We compared a number of magnetic resonance imaging (MRI) data sets, ranging from subtle to severe difference is acquisition parameters. Results indicate that, based on the similarity of data sets, the DRC is able to give an indication as to when the performance of a supervised classifier decreases. The strictness of the DRC can be set by the user, depending on what one considers to be an acceptable underperformance.

Original languageEnglish
Article numbere0237009
Number of pages16
JournalPLoS ONE
Volume15
Issue number8
DOIs
Publication statusPublished - 11 Aug 2020

Keywords

  • Machine learning
  • Bayesian inference
  • Transfer learning
  • Data set similarity

Fingerprint

Dive into the research topics of 'The data representativeness criterion: Predicting the performance of supervised classification based on data set similarity'. Together they form a unique fingerprint.

Cite this