White Matter Infarct Detection with Transformer and Auto-ML-Derived Models
Abstract
1. Introduction
2. Materials and Methods
2.1. T1-w Imaging Data for Chronic Stroke Detection
2.2. T1-w Data Augmentation with the Diffusion Model
2.3. Hierarchical Transformer Model for Stroke Detection
2.4. Calculating Auto-ML Features from rs-fMRI
2.5. Combining UNesT Lesion Segmentation with Auto-ML Features
2.6. Correlation with Motor and Neuropsychological Assessment
3. Results
3.1. Participant Characteristics
3.2. Data Augmentation with the Diffusion Model
3.3. T1-w Infarct Detection with the Transformer Model
3.4. Selecting Subacute Infarct-Specific rs-fMRI Metrics
3.5. Selecting Chronic Infarct-Specific rs-fMRI Metrics
3.6. rs-fMRI Association with Motor and Neuropsychological Assessment
4. Discussion
5. Conclusions
Author Contributions
Funding
Data Availability Statement
Acknowledgments
Conflicts of Interest
Abbreviations
| AIC | Akaike Information Criteria |
| ALFF | Amplitude of low frequency fluctuations |
| Auto-ML | Automated machine learning |
| BET | Brain extraction tool |
| BOLD | Blood oxygen level dependent |
| CBF | Cerebral blood flow |
| CDC | U.S. Centers for Disease Control and Prevention |
| DL | Deep learning |
| DWI | Diffusion-weighted imaging |
| FC | Functional connectivity |
| FIM | Functional independent measure |
| FN | False negative |
| FP | False positive |
| GLM | General linear model |
| GM | Gray matter |
| MED-DDPM | Medical denoising diffusion probabilistic model |
| MONAI | Medical Open Network for Artificial Intelligence |
| ReHo | Regional homogeneity |
| ROI | Region of interest |
| rs-fMRI | Resting-state fMRI |
| TP | True positive |
| T1-w | T1-weighted |
| UNesT | U-shape nested transformers |
| WM | White matter |
References
- Ahmed, M.; Bin Shafiq, S.; Razzak, J.; Mansoor, K.T.; Naveed, M.A.; Ali, A.; Bin Faheem, M.S.; Samadi, S.; Chigurupati, H.D.; Neppala, S. Trends and Disparities in Stroke Mortality Among Adults with Hyperlipidemia in the United States, 1999–2023. J. Epidemiol. Glob. Health 2025, 15, 116. [Google Scholar] [CrossRef]
- Jones, C.H.; Dolsten, M. Healthcare on the brink: Navigating the challenges of an aging society in the United States. npj Aging 2024, 10, 22. [Google Scholar] [CrossRef]
- FastStats. Available online: https://www.cdc.gov/nchs/fastats/leading-causes-of-death.htm (accessed on 9 March 2026).
- Vermeer, S.E.; Prins, N.D.; Heijer, T.D.; Hofman, A.; Koudstaal, P.J.; Breteler, M.M. Silent Brain Infarcts and the Risk of Dementia and Cognitive Decline. N. Engl. J. Med. 2003, 348, 1215–1222. [Google Scholar] [CrossRef] [PubMed]
- Winzeck, S.; Hakim, A.; McKinley, R.; Pinto, J.A.A.D.S.R.; Alves, V.; Silva, C.; Pisov, M.; Krivov, E.; Belyaev, M.; Monteiro, M.; et al. ISLES 2016 and 2017-Benchmarking Ischemic Stroke Lesion Outcome Prediction Based on Multispectral MRI. Front. Neurol. 2018, 9, 679. [Google Scholar] [CrossRef]
- Bridge, C.P.; Bizzo, B.C.; Hillis, J.M.; Chin, J.K.; Comeau, D.S.; Gauriau, R.; Macruz, F.; Pawar, J.; Noro, F.T.C.; Sharaf, E.; et al. Development and clinical application of a deep learning model to identify acute infarct on magnetic resonance imaging. Sci. Rep. 2022, 12, 2154. [Google Scholar] [CrossRef]
- Eastwood, J.D.; Engelter, S.T.; MacFall, J.F.; Delong, D.M.; Provenzale, J.M. Quantitative assessment of the time course of infarct signal intensity on diffusion-weighted images. AJNR Am. J. Neuroradiol. 2003, 24, 680–687. [Google Scholar]
- Kanda, T.; Ishii, K.; Kawaguchi, H.; Kitajima, K.; Takenaka, D. High signal intensity in the dentate nucleus and globus pallidus on unenhanced T1-weighted MR images: Relationship with increasing cumulative dose of a gadolinium-based contrast material. Radiology 2014, 270, 834–841. [Google Scholar] [CrossRef]
- Levit, A.; Hachinski, V.; Whitehead, S.N. Neurovascular unit dysregulation, white matter disease, and executive dysfunction: The shared triad of vascular cognitive impairment and Alzheimer disease. GeroScience 2020, 42, 445–465. [Google Scholar] [CrossRef]
- Wang, Y.; Liu, G.; Hong, D.; Chen, F.; Ji, X.; Cao, G. White matter injury in ischemic stroke. Prog. Neurobiol. 2016, 141, 45–60. [Google Scholar] [CrossRef] [PubMed]
- Bonanno, L.; Cannuli, A.; Pignolo, L.; Marino, S.; Quartarone, A.; Calabrò, R.S.; Cerasa, A. Neural Plasticity Changes Induced by Motor Robotic Rehabilitation in Stroke Patients: The Contribution of Functional Neuroimaging. Bioengineering 2023, 10, 990. [Google Scholar] [CrossRef] [PubMed]
- Christidi, F.; Orgianelis, I.; Merkouris, E.; Koutsokostas, C.; Tsiptsios, D.; Karavasilis, E.; Psatha, E.A.; Tsiakiri, A.; Serdari, A.; Aggelousis, N.; et al. A Comprehensive Review on the Role of Resting-State Functional Magnetic Resonance Imaging in Predicting Post-Stroke Motor and Sensory Outcomes. Neurol. Int. 2024, 16, 189–201. [Google Scholar] [CrossRef] [PubMed]
- Kielar, A.; Deschamps, T.; Chu, R.K.O.; Jokel, R.; Khatamian, Y.B.; Chen, J.J.; Meltzer, J.A. Identifying Dysfunctional Cortex: Dissociable Effects of Stroke and Aging on Resting State Dynamics in MEG and fMRI. Front. Aging Neurosci. 2016, 8, 40. [Google Scholar] [CrossRef]
- Zhang, Z. Resting-state functional abnormalities in ischemic stroke: A meta-analysis of fMRI studies. Brain Imaging Behav. 2024, 18, 1569–1581. [Google Scholar] [CrossRef] [PubMed]
- Tsai, Y.-H.; Yuan, R.; Huang, Y.-C.; Weng, H.-H.; Yeh, M.-Y.; Lin, C.-P.; Biswal, B.B. Altered resting-state FMRI signals in acute stroke patients with ischemic penumbra. PLoS ONE 2014, 9, e105117. [Google Scholar] [CrossRef][Green Version]
- Liew, S.-L.; Lo, B.P.; Donnelly, M.R.; Zavaliangos-Petropulu, A.; Jeong, J.N.; Barisano, G.; Hutton, A.; Simon, J.P.; Juliano, J.M.; Suri, A.; et al. A large, curated, open-source stroke neuroimaging dataset to improve lesion segmentation algorithms. Sci. Data 2022, 9, 320. [Google Scholar] [CrossRef]
- Smith, S.M. Fast robust automated brain extraction. Hum. Brain Mapp. 2002, 17, 143–155. [Google Scholar] [CrossRef]
- Jenkinson, M.; Smith, S. A global optimisation method for robust affine registration of brain images. Med. Image Anal. 2001, 5, 143–156. [Google Scholar] [CrossRef]
- Dorjsembe, Z.; Pao, H.-K.; Odonchimed, S.; Xiao, F. Conditional Diffusion Models for Semantic 3D Brain MRI Synthesis. IEEE J. Biomed. Health Inform. 2024, 28, 4084–4093. [Google Scholar] [CrossRef]
- Yu, X.; Yang, Q.; Zhou, Y.; Cai, L.Y.; Gao, R.; Lee, H.H.; Li, T.; Bao, S.; Xu, Z.; Lasko, T.A.; et al. UNesT: Local spatial representation learning with hierarchical transformer for efficient medical segmentation. Med. Image Anal. 2023, 90, 102939. [Google Scholar] [CrossRef]
- Cardoso, M.J.; Li, W.; Brown, R.; Ma, N.; Kerfoot, E.; Wang, Y.; Murrey, B.; Myronenko, A.; Zhao, C.; Yang, D. MONAI: An open-source framework for deep learning in healthcare. arXiv 2022, arXiv:2211.02701. [Google Scholar] [CrossRef]
- Siegel, J.S.; Shulman, G.L.; Corbetta, M. Measuring functional connectivity in stroke: Approaches and considerations. J. Cereb. Blood Flow Metab. 2017, 37, 2665–2678. [Google Scholar] [CrossRef]
- Smith, S.M.; Jenkinson, M.; Woolrich, M.W.; Beckmann, C.F.; Behrens, T.E.; Johansen-Berg, H.; Bannister, P.R.; De Luca, M.; Drobnjak, I.; Flitney, D.E.; et al. Advances in functional and structural MR image analysis and implementation as FSL. Neuroimage 2004, 23, S208–S219. [Google Scholar] [CrossRef]
- Fischl, B. FreeSurfer. NeuroImage 2012, 62, 774–781. [Google Scholar] [CrossRef]
- Jenkinson, M.; Beckmann, C.F.; Behrens, T.E.; Woolrich, M.W.; Smith, S.M. FSL. NeuroImage 2012, 62, 782–790. [Google Scholar] [CrossRef]
- Jenkinson, M.; Bannister, P.; Brady, M.; Smith, S. Improved optimization for the robust and accurate linear registration and motion correction of brain images. NeuroImage 2002, 17, 825–841. [Google Scholar] [CrossRef]
- Predictive Maintenance Toolbox. Available online: https://www.mathworks.com/products/predictive-maintenance.html (accessed on 17 March 2026).
- Dobromyslin, V.I.; Zhou, W.; For the Alzheimer’s Disease Neuroimaging Initiative. Enhancing cerebral infarct classification by automatically extracting relevant fMRI features. Brain Inform. 2025, 12, 16. [Google Scholar] [CrossRef]
- Granger, C.V.; Hamilton, B.B.; Keith, R.A.; Zielezny, M.; Sherwin, F.S. Advances in functional assessment for medical rehabilitation. Top. Geriatr. Rehabil. 1986, 1, 59–74. [Google Scholar] [CrossRef]
- Goodglass, H.; Kaplan, E.; Barresi, B. BDAE3 Boston Diagnostic Aphasia Examination, 3rd ed.; PRO-ED, Inc.: Austin, TX, USA; Available online: https://www.alimed.com/products/boston-diagnostic-aphasia-examination-3rd-ed-bdae-3-complete-kit?srsltid=AfmBOopDAlWC1tNrzd13MhkfHRSnjU8fsGMI5DA6yzA8LeVJLQGJqzel (accessed on 25 March 2026).
- Posner, M.I. Orienting of attention. Q. J. Exp. Psychol. 1980, 32, 3–25. [Google Scholar] [CrossRef]
- Gryska, E.; Björkman-Burtscher, I.; Jakola, A.S.; Dunås, T.; Schneiderman, J.; Heckemann, R.A. Deep learning for automatic brain tumour segmentation on MRI: Evaluation of recommended reporting criteria via a reproduction and replication study. BMJ Open 2022, 12, e059000. [Google Scholar] [CrossRef] [PubMed]
- Yan, W.; Huang, L.; Xia, L.; Gu, S.; Yan, F.; Wang, Y.; Tao, Q. MRI Manufacturer Shift and Adaptation: Increasing the Generalizability of Deep Learning Segmentation for MR Images Acquired with Different Scanners. Radiol. Artif. Intell. 2020, 2, e190195. [Google Scholar] [CrossRef] [PubMed]
- Puzio, T.; Matera, K.; Karwowski, J.; Piwnik, J.; Białkowski, S.; Podyma, M.; Dunikowski, K.; Siger, M.; Stasiołek, M.; Grzelak, P.; et al. Deep learning-based automatic segmentation of brain structures on MRI: A test-retest reproducibility analysis. Comput. Struct. Biotechnol. J. 2025, 28, 128–140. [Google Scholar] [CrossRef]
- Kondrateva, E.; Barg, S.; Vasiliev, M. Benchmarking the Reproducibility of Brain MRI Segmentation Across Scanners and Time. arXiv 2025, arXiv:2504.15931. [Google Scholar] [CrossRef]
- Russo, S.; Pigorini, A.; Mikulan, E.; Sarasso, S.; Rubino, A.; Zauli, F.; Parmigiani, S.; D’ORio, P.; Cattani, A.; Francione, S.; et al. Focal lesions induce large-scale percolation of sleep-like intracerebral activity in awake humans. NeuroImage 2021, 234, 117964. [Google Scholar] [CrossRef]
- Cassidy, J.M.; Wodeyar, A.; Wu, J.; Kaur, K.; Masuda, A.K.; Srinivasan, R.; Cramer, S.C. Low-Frequency Oscillations Are a Biomarker of Injury and Recovery After Stroke. Stroke 2020, 51, 1442–1450. [Google Scholar] [CrossRef] [PubMed]
- Facchin, L.; Schöne, C.; Mensen, A.; Bandarabadi, M.; Pilotto, F.; Saxena, S.; Libourel, P.A.; Bassetti, C.L.; Adamantidis, A.R. Slow Waves Promote Sleep-Dependent Plasticity and Functional Recovery after Stroke. J. Neurosci. 2020, 40, 8637–8651. [Google Scholar] [CrossRef] [PubMed]
- Fruhwirth, V.; Berger, L.; Gattringer, T.; Fandler-Höfler, S.; Kneihsl, M.; Eppinger, S.; Ropele, S.; Fink, A.; Deutschmann, H.; Reishofer, G.; et al. White matter integrity and functional connectivity of the default mode network in acute stroke are associated with cognitive outcome three months post-stroke. J. Neurol. Sci. 2024, 462, 123071. [Google Scholar] [CrossRef] [PubMed]
- Schellekens, M.M.I.; Li, H.; Boot, E.M.; Verhoeven, J.I.; Ekker, M.S.; Meijer, F.J.; Kessels, R.P.; de Leeuw, F.-E.; Tuladhar, A.M. White matter integrity and cognitive performance in the subacute phase after ischemic stroke in young adults. NeuroImage Clin. 2024, 45, 103711. [Google Scholar] [CrossRef]






| Raw Signal Features | Q1 | Power Spectrum Features | Power | Maximum |
| Clearance Factor | Q3 | Peak Amplitude | Linearly De-trended Features | Q1 |
| Crest Factor | IQR | Peak Frequency | Clearance Factor | Q3 |
| Impulse Factor | Signal Envelope Features | Power | Crest Factor | IQR |
| Kurtosis | Clearance Factor | Features After Autoregressive Filtering | Impulse Factor | Shape Factor |
| Mean | Crest Factor | First Coefficient | Kurtosis | Linearly De-trended, After Autoregressive Filter, Features |
| Peak Value | Impulse Factor | First Frequency | Mean | First Coefficient |
| RMS | Kurtosis | MSE | Peak Value | First Frequency |
| SINAD | Mean | MAE | RMS | Damping Coefficient |
| SNR | Peak Value | AIC | SINAD | MSE |
| Shape Factor | RMS | Mean | SNR | MAE |
| Skewness | SINAD | Variance | Shape Factor | AIC |
| SD | SNR | RMS | Skewness | Mean |
| THD | Shape Factor | Kurtosis | SD | Variance |
| Minimum | Skewness | Signal Envelope Spectral Features | THD | RMS |
| Median | SD | Peak Amplitude | Minimum | Kurtosis |
| Maximum | THD | Peak Frequency | Median |
| Dataset | n | Median Days Post-Stroke | Median Lesion Volume [mm3] | Infarct Type Count | Age Mean ± SD | Gender % Male % Female |
|---|---|---|---|---|---|---|
| ATLAS R2 Subacute–Chronic Cohort | 654 | 501 | 3996 | * | * | * |
| WU Subacute Infarct Cohort | 20 | 12 | 37,914 | 16 ischemic 1 hemorrhagic 3 other | 52.9 ± 11.5 | 40.0% M 60.0% F |
| WU Chronic Infarct Cohort | 14 | 382 | 36,235 | 11 ischemic 1 hemorrhagic 2 other | 54.6 ± 8.3 | 22.2% M 77.8% F |
| Model | Data Split | Subj. 1 | Subj. 2 | Subj. 3 | Subj. 4 | Subj. 5 | Subj. 6 | Subj. 7 | Subj. 8 | Subj. 9 | Subj. 10 | Mean ±SD |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Original UNesT * | Original Train/Test | 0.00 | 0.55 | 0.00 | 0.67 | 0.02 | 0.08 | 0.00 | 0.50 | 0.07 | 0.54 | 0.24 ±0.28 |
| Re-Optimized UNesT * | 0.03 | 0.68 | 0.11 | 0.74 | 0.10 | 0.37 | 0.00 | 0.77 | 0.45 | 0.83 | 0.41 ±0.32 | |
| Multivariate GLM | 0.10 ±0.00 | 0.64 ±0.00 | 0.39 ±0.01 | 0.78 ±0.00 | 0.65 ±0.01 | 0.42 ±0.00 | 0.00 ±0.00 | 0.80 ±0.00 | 0.52 ±0.01 | 0.85 ±0.00 | 0.51 ±0.28 | |
| Original UNesT * | Swapped Train/Test | 0.11 | 0.49 | 0.66 | 0.01 | 0.57 | 0.00 | 0.00 | 0.01 | 0.47 | 0.63 | 0.29 ±0.29 |
| Re-Optimized UNesT * | 0.56 | 0.57 | 0.58 | 0.08 | 0.42 | 0.00 | 0.37 | 0.09 | 0.85 | 0.44 | 0.40 ±0.26 | |
| Multivariate GLM | 0.69 ±0.01 | 0.51 ±0.01 | 0.43 ±0.00 | 0.33 ±0.01 | 0.15 ±0.06 | 0.00 ±0.00 | 0.62 ±0.01 | 0.75 ±0.01 | 0.89 ±0.00 | 0.36 ±0.00 | 0.49 ±0.26 |
| Model | Data Split | Subj. 1 | Subj. 2 | Subj. 3 | Subj. 4 | Subj. 5 | Subj. 6 | Subj. 7 | Mean ±SD |
|---|---|---|---|---|---|---|---|---|---|
| UNesT * | Original Train/Test | 0.58 | 0.30 | 0.01 | 0.45 | 0.85 | 0.53 | 0.14 | 0.41 ±0.28 |
| Re-Optimized UNesT * | 0.67 | 0.44 | 0.06 | 0.53 | 0.86 | 0.51 | 0.33 | 0.49 ±0.25 | |
| Multivariate GLM | 0.67 ±0.00 | 0.43 ±0.01 | 0.06 ±0.00 | 0.54 ±0.00 | 0.86 ±0.00 | 0.51 ±0.00 | 0.37 ±0.00 | 0.48 ±0.23 | |
| UNesT * | Swapped Train/Test | 0.53 | 0.63 | 0.00 | 0.47 | 0.47 | 0.43 | 0.43 | 0.42 ±0.20 |
| Re-Optimized UNesT * | 0.65 | 0.55 | 0.00 | 0.43 | 0.72 | 0.59 | 0.58 | 0.50 ±0.24 | |
| Multivariate GLM | 0.66 ±0.00 | 0.47 ±0.00 | 0.00 ±0.00 | 0.37 ±0.01 | 0.72 ±0.00 | 0.69 ±0.00 | 0.61 ±0.00 | 0.51 ±0.23 |
| Independent Variable | Dependent Variable | Slope Estimate | p-Value | R2 |
|---|---|---|---|---|
| Subacute Timepoint Spectral Peak Amplitude | Change in Word Comprehension | 0.90 | 0.012 ** | 0.42 |
| Spectral Peak Amplitude Change | Change in FIM Walk | −0.76 | 0.031 | 0.33 |
| Spectral Peak Amplitude Change | Change in Word Comprehension | −0.95 | 0.083 | 0.23 |
| Subacute Timepoint Spectral Peak Amplitude | Change in FIM Walk | 0.41 | 0.108 | 0.20 |
| Subacute Timepoint Spectral Peak Amplitude | Change in Posner Cueing | −36.40 | 0.603 | 0.04 |
| Spectral Peak Amplitude Change | Change in Posner Cueing | 33.74 | 0.622 | 0.03 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Dobromyslin, V.; Zhou, W. White Matter Infarct Detection with Transformer and Auto-ML-Derived Models. Brain Sci. 2026, 16, 529. https://doi.org/10.3390/brainsci16050529
Dobromyslin V, Zhou W. White Matter Infarct Detection with Transformer and Auto-ML-Derived Models. Brain Sciences. 2026; 16(5):529. https://doi.org/10.3390/brainsci16050529
Chicago/Turabian StyleDobromyslin, Vitaly, and Wenjin Zhou. 2026. "White Matter Infarct Detection with Transformer and Auto-ML-Derived Models" Brain Sciences 16, no. 5: 529. https://doi.org/10.3390/brainsci16050529
APA StyleDobromyslin, V., & Zhou, W. (2026). White Matter Infarct Detection with Transformer and Auto-ML-Derived Models. Brain Sciences, 16(5), 529. https://doi.org/10.3390/brainsci16050529
