Benchmarking machine learning models in lesion-symptom mapping for predicting language outcomes in stroke survivors.

Journal: Frontiers in neuroimaging
Published Date:

Abstract

Several decades of research have investigated the neural connections between stroke-induced brain damage and language difficulties. Typically, lesion-symptom mapping (LSM) studies that address this connection have relied on mass univariate statistics, which do not account for multidimensional relationships between variables. Machine learning (ML) techniques, which can capture these intricate connections, offer a promising complement to LSM methods. To test this promise, we benchmarked ML models on structural and functional MRI to predict aphasia severity ( = 238) and naming impairment ( = 191) for a cohort of chronic-stage stroke survivors. We used nested cross-validation to examine performance along three dimensions: (1) parcellation schemes (JHU, AAL, BRO, and AICHA atlases), (2) neuroimaging modalities (resting-state functional connectivity, structural connectivity, mean diffusivity, fractional anisotropy, and lesion location) and (3) ML methods (Random Forest, Support Vector Regression, Decision Tree, K Nearest Neighbors, and Gradient Boosting). The best results were obtained by combining the JHU atlas, lesion location, and the Random Forest model. This combination yielded moderate to high correlations with the two different behavioral scores. Key regions identified included several perisylvian areas and pathways within the language network. This work complements existing LSM methods with new tools for improving the prediction of language outcomes in stroke survivors.

Authors

  • Deepa Tilwani
    Artificial Intelligence Institute, University of South Carolina, Columbia, SC, United States.
  • Christian O'Reilly
    Blue Brain Project, EPFL, Campus Biotech, Ch. des Mines 9, CH-1202, Geneva, Switzerland.
  • Nicholas Riccardi
    Department of Communication Sciences and Disorders, University of South Carolina, United States. Electronic address: riccardn@email.sc.edu.
  • Valerie L Shalin
    Knoesis Center, Wright State University, Dayton, Ohio, United States of America.
  • Dirk-Bart den Ouden
    Department of Communication Sciences and Disorders, University of South Carolina, Columbia, SC, United States.
  • Julius Fridriksson
    Center for the Study of Aphasia Recovery, University of South Carolina, Columbia, South Carolina, USA.
  • Svetlana V Shinkareva
    Institute for Mind and Brain, University of South Carolina, Columbia, SC, United States.
  • Amit P Sheth
    Artificial Intelligence Institute, University of South Carolina, Columbia, SC, United States.
  • Rutvik H Desai
    Department of Psychology, University of South Carolina, United States.

Keywords

No keywords available for this article.