TY - JOUR
T1 - The data representativeness criterion
T2 - Predicting the performance of supervised classification based on data set similarity
AU - Schat, Evelien
AU - van de Schoot, Rens
AU - Kouw, Wouter M.
AU - Veen, Duco
AU - Mendrik, Adriënne M.
PY - 2020/8/11
Y1 - 2020/8/11
N2 - 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.
AB - 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.
KW - Machine learning
KW - Bayesian inference
KW - Transfer learning
KW - Data set similarity
UR - http://www.scopus.com/inward/record.url?scp=85089360065&partnerID=8YFLogxK
U2 - 10.1371/journal.pone.0237009
DO - 10.1371/journal.pone.0237009
M3 - Article
C2 - 32780738
SN - 1932-6203
VL - 15
JO - PLoS ONE
JF - PLoS ONE
IS - 8
M1 - e0237009
ER -