Beyond accuracy: Measures for assessing machine learning models, pitfalls and guidelines
Dinga, R.; Penninx, B. W. J. H.; Veltman, D. J.; Schmaal, L.; Marquand, A. F.
Show abstract
Pattern recognition predictive models have become an important tool for analysis of neuroimaging data and answering important questions from clinical and cognitive neuroscience. Regardless of the application, the most commonly used method to quantify model performance is to calculate prediction accuracy, i.e. the proportion of correctly classified samples. While simple and intuitive, other performance measures are often more appropriate with respect to many common goals of neuroimaging pattern recognition studies. In this paper, we will review alternative performance measures and focus on their interpretation and practical aspects of model evaluation. Specifically, we will focus on 4 families of performance measures: 1) categorical performance measures such as accuracy, 2) rank based performance measures such as the area under the curve, 3) probabilistic performance measures based on quadratic error such as Brier score, and 4) probabilistic performance measures based on information criteria such as logarithmic score. We will examine their statistical properties in various settings using simulated data and real neuroimaging data derived from public datasets. Results showed that accuracy had the worst performance with respect to statistical power, detecting model improvement, selecting informative features and reliability of results. Therefore in most cases, it should not be used to make statistical inference about model performance. Accuracy should also be avoided for evaluating utility of clinical models, because it does not take into account clinically relevant information, such as relative cost of false-positive and false-negative misclassification or calibration of probabilistic predictions. We recommend alternative evaluation criteria with respect to the goals of a specific machine learning model.
Matching journals
The top 4 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Data-driven Discovery of Mathematical and Physical Relations in Oncology Data using Human-understandable Machine Learning 94%
- PECLIDES Neuro - A Personalisable Clinical Decision Support System for Neurological Diseases 93%
- An Explainable Multi-Modal Neural Network Architecture for Predicting Epilepsy Comorbidities Based on Administrative Claims Data 92%
Similar papers in this journal
Similar papers in this journal
- Selecting the most important self-assessed features for predicting conversion to Mild Cognitive Impairment with Random Forest and Permutation-based methods 95%
- Machine learning for classifying chronic kidney disease and predicting creatinine levels using at-home measurements 94%
- On evaluation metrics for medical applications of artificial intelligence 93%
Similar papers in this journal
- Class imbalance should not throw you off balance: Choosing the right classifiers and performance metrics for brain decoding with imbalanced data 95%
- Parsimonious EBM: generalising the event-based model of disease progression for simultaneous events 95%
- Bootstrap aggregating improves the generalizability of Connectome Predictive Modelling 94%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.