Machine learning classification meets migraine: recommendations for study evaluation.

Journal: The journal of headache and pain
Published Date:

Abstract

The integration of machine learning (ML) classification techniques into migraine research has offered new insights into the pathophysiology and classification of migraine types and subtypes. However, inconsistencies in study design, lack of methodological transparency, and the absence of external validation limit the impact and reproducibility of such studies. This paper presents a framework of six essential recommendations for evaluating ML-based classification in migraine research: (1) group homogenization by clinical phenotype, attack frequency, comorbidity, therapy, and demographics; (2) defining adequate sample size; (3) quality control of collected and preprocessed data; (4) transparent training, testing, and performance evaluation of ML models, including strategies for data splitting, overfitting control, and feature selection; (5) interpretability of results with clinical relevance; and (6) open data and code sharing to facilitate reproducibility. These recommendations aim to balance the trade-off between model generalization and precision while encouraging collaborative standardization across the ML and headache communities. Furthermore, this framework intends to stimulate discussion toward forming a consortium to establish definitive guidelines for ML-based classification research in migraine field.

Authors

  • Igor Petrušić
    Faculty of Physical Chemistry, Laboratory for Advanced Analysis of Neuroimages, University of Belgrade, Belgrade, Serbia.
  • Andrej Savić
    Science and Research Centre, School of Electrical Engineering, University of Belgrade, University of Belgrade, Belgrade, Serbia.
  • Katarina Mitrović
    Department of Information Technologies, Faculty of Technical Sciences Čačak, University of Kragujevac, Čačak, Serbia.
  • Nebojsa Bacanin
    Department of Informatics and Computing, Singidunum University, Belgrade, Serbia.
  • Gabriele Sebastianelli
    Department of Medico-Surgical Sciences and Biotechnologies, Sapienza University of Rome Polo Pontino ICOT, Latina, Italy.
  • Daniele Secci
    Department of Engineering and Architecture, University of Parma, Parma, Italy.
  • Gianluca Coppola
    Department of Medico-Surgical Sciences and Biotechnologies, University of Rome Sapienza, Latina, Italy.