Assessment of Machine Learning Pipelines for Prediction of Behavioral Deficits from Brain Disconnectomes
| Autor: | Marco Zorzi, Michele De Filippo De Grazia, Elvio Blini, Alberto Testolin |
|---|---|
| Rok vydání: | 2021 |
| Předmět: |
Disconnections
Computer science business.industry Pipeline (computing) Dimensionality reduction Supervised learning Feature extraction Machine learning computer.software_genre Independent component analysis Predictive modeling Stroke Structural connectome Lasso (statistics) Principal component analysis Artificial intelligence business computer Interpretability |
| Zdroj: | Brain Informatics ISBN: 9783030869922 BI |
| Popis: | Recent studies have shown that brain lesions following stroke can be probabilistically mapped onto disconnections of white matter tracts, and that the resulting “disconnectome” is predictive of the patient’s behavioral deficits. Disconnectome maps are sparse, high-dimensional 3D matrices that require unsupervised dimensionality reduction followed by supervised learning for prediction of the associated behavioral data. However, the optimal machine learning pipeline for disconnectome data still needs to be identified. We examined four dimensionality reduction methods at varying levels of compression and used the extracted features as input for cross-validated regularized regression to predict the associated language and motor deficits. Features extracted by Principal Component Analysis and Non-Negative Matrix Factorization were found to be the best predictors, followed by Independent Component Analysis and Dictionary Learning. Optimizing the number of extracted features improved predictive accuracy and greatly reduced model complexity. Moreover, the choice of dimensionality reduction technique was found to optimally combine with a specific type of regularized regression (ridge vs. LASSO). Overall, our findings represent an important step towards an optimal pipeline that yields high prediction accuracy with a small number of features, which can also improve model interpretability. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|