Feature selection using metaheuristics to predict annual amyotrophic lateral sclerosis progression

Abstract

Objective: Amyotrophic lateral sclerosis (ALS), a progressive neurodegenerative disease with no curative treatment and affecting motor neurons, leads to motor weakness, atrophy, spasticity and difficulties with speech, swallowing, and breathing. Accurately predicting disease progression and survival is crucial for optimizing patient care, intervention planning, and informed decision-making.

Methods: Data were gathered from the PRO-ACT database (4659 patients), clinical trial data from ExonHit Therapeutics (384 patients) and the PULSE multicenter cohort aimed at identifying predictive factors of disease progression (198 patients). Machine learning (ML) techniques including logistic/linear regression (LR), K-nearest neighbors, decision tree, random forest, and light gradient boosting machine (LGBM) were applied to forecast ALS progression using ALS Functional Rating Scale (ALSFRS) scores and patient survival over one year. Models were validated using 10-fold cross-validation, while Kaplan-Meier estimates were employed to cluster patients according to their profiles. To enhance the predictive accuracy of our models, we performed feature selection using ANOVA and differential evolution (DE).

Results: LR with DE achieved a balanced accuracy of 76.05% on validation (ranging from 68.6% to 79.8% per fold) and 76.33% on test data, with an AUC of 0.84. With Kaplan-Meier's estimates, we identified five distinct patient clusters (C-index = 0.8; log-rank test p value ≤0.0001). Additionally, LGBM predictions for ALSFRS progression at 3 months yielded an RMSE of 3.14 and an adjusted R² of 0.764.

Conclusion: This study showcases the potential of ML models to provide significant predictive insights in ALS, enhancing the understanding of disease dynamics and supporting patient care.

Keywords: Machine learning; classification; feature selection; optimization; regression.