Generalizable and equitable automated ischaemic stroke lesion segmentation with vision transformers

Abstract

Ischaemic stroke, a leading cause of death and disability, relies on neuroimaging for characterising the anatomical pattern of injury. Diffusion-weighted MRI (DWI) provides the most anatomically specific signal in acute ischaemic stroke but poses substantial challenges for automated lesion segmentation due to susceptibility artefacts, lesion morphological heterogeneity, comorbid pathology, instrumental variability, and limited labelled data. Current U-Net-based models therefore underperform, a problem accentuated by evaluation metrics that neglect anatomical, subpopulation and acquisition-dependent variability. We train 3D vision transformer–based segmentation models on a multi-site DWI dataset comprising 3563 annotated lesion-positive and 6900 lesion-negative volumes, using balanced cross-validation splits. We compare these models with U-Net baselines and an nnU-Net configuration under harmonised augmentation and training schemes, and introduce an evaluation framework that quantifies fidelity, anatomical precision, robustness to image corruption and equity across demographic and lesion-defined subtypes. Here, we show that transformer-based models with our proposed control-image regularisation achieve higher segmentation performance than U-Net-based approaches on clinically realistic data, while substantially reducing false positives in lesion-negative images. They exhibit more stable performance across lesion sizes, anatomical territories, image quality and patient subgroups, indicating improved epistemic equity relative to convolutional architectures. This work reconciles model expressivity with domain-specific challenges and redefines performance benchmarks to prioritise equity and generalisability–critical for personalised medicine and mechanistic research. These findings establish vision transformer architectures, combined with equity-aware validation, as a powerful approach for ischaemic stroke lesion segmentation on DWI.