End-to-End Stroke Imaging Analysis Using Effective Connectivity and Interpretable Artificial Intelligence

Field: Computer Science (Physical Sciences)

In this paper, we propose a reservoir computing-based and directed graph analysis pipeline. The goal of this pipeline is to define an efficient brain representation for connectivity in stroke data derived from magnetic resonance imaging. Ultimately, this representation is used within a directed graph convolutional architecture and investigated with explainable artificial intelligence (AI) tools, offering a more detailed understanding of how stroke alters communication within the brain. Stroke is one of the leading causes of mortality and morbidity worldwide, and it demands precise diagnostic tools for timely intervention and improved patient outcomes. Neuroimaging data, with their rich structural and functional information, provide a fertile ground for biomarker discovery. However, the complexity and variability of information flow in the brain require advanced analysis, especially if we consider the case of disrupted networks as those given by the brain connectome of stroke patients. To address the needs given by this complex scenario we proposed an end-to-end pipeline. This pipeline begins with defining the effective connectivity of the brain. This allows directed graph network representations that have not been fully investigated so far by graph convolutional network classifiers. To have a complete overview, the analysis with reservoir computing-based causality is compared to other two effective connectivity approaches: one linear (Granger causality) and one non-linear method (transfer entropy). Then, the pipeline subsequently incorporates a classification module to categorize the effective connectivity (directed graphs) of brain networks of patients versus matched healthy control. The graph convolutional architecture is also compared to legacy methods such as random forest and support vector machine providing a complete benchmark. While the pipeline includes a classification module for distinguishing between stroke patients and healthy controls, the focus is on the interpretation of these directed graphs, which reveal critical disruptions in connectivity. Indeed, the classification led to an area under the curve of 0.69 by using graph convolutional networks, 0.72 by using local topological profiling random forest, and 0.71 by using support vector machine with the given heterogeneous dataset. More importantly, thanks to explainable tools, an interpretation of disrupted networks across the brain networks was possible. This elucidates the effective connectivity biomarker’s contribution to stroke classification, fostering insights into disease mechanisms and treatment responses. This transparent analytical framework not only enhances clinical interpretability but also instills confidence in decision-making processes, crucial for translating research findings into clinical practice. Our proposed machine learning pipeline showcases the potential of reservoir computing to define causality and therefore directed graph networks, which can in turn be used in a directed graph classifier and explainable analysis of neuroimaging data. This method prioritizes uncovering miscommunication in brain networks, with the potential to improve our understanding of stroke and other brain diseases.INDEX TERMS Effective connectivity, explainable AI, reservoir computing, stroke, graph convolutional networks, GCN, GNN.