Brain Atlas Guided Attention U-Net for White Matter Hyperintensity Segmentation
This addresses a practical limitation in clinical settings where T1-weighted images are often unavailable, particularly for acute stroke patients, by enabling accurate WMH segmentation with standard scans.
The paper tackles the problem of White Matter Hyperintensity (WMH) segmentation in brain MRI by proposing a novel method that uses only FLAIR images and a white matter brain atlas, achieving competitive performance that outperforms state-of-the-art methods requiring additional T1-weighted images.
White Matter Hyperintensities (WMH) are the most common manifestation of cerebral small vessel disease (cSVD) on the brain MRI. Accurate WMH segmentation algorithms are important to determine cSVD burden and its clinical consequences. Most of existing WMH segmentation algorithms require both fluid attenuated inversion recovery (FLAIR) images and T1-weighted images as inputs. However, T1-weighted images are typically not part of standard clinicalscans which are acquired for patients with acute stroke. In this paper, we propose a novel brain atlas guided attention U-Net (BAGAU-Net) that leverages only FLAIR images with a spatially-registered white matter (WM) brain atlas to yield competitive WMH segmentation performance. Specifically, we designed a dual-path segmentation model with two novel connecting mechanisms, namely multi-input attention module (MAM) and attention fusion module (AFM) to fuse the information from two paths for accurate results. Experiments on two publicly available datasets show the effectiveness of the proposed BAGAU-Net. With only FLAIR images and WM brain atlas, BAGAU-Net outperforms the state-of-the-art method with T1-weighted images, paving the way for effective development of WMH segmentation. Availability:https://github.com/Ericzhang1/BAGAU-Net