Specialized gray matter segmentation via a generative adversarial network: application on brain white matter hyperintensities classification

Abstract

Background: White matter hyperintensities (WMH) observed in T2 fluid-attenuated inversion recovery (FLAIR) images have emerged as potential markers of neurodegenerative diseases like Multiple Sclerosis (MS). Lacking comprehensive automated WMH classification systems in current research, there is a need to develop accurate detection and classification methods for WMH that will benefit the diagnosis and monitoring of brain diseases.

Objective: Juxtacortical WMH (JCWMH) is a less explored subtype of WMH, primarily due to the hard definition of the cortex in FLAIR images, which is escalated by the presence of lesions to obtain appropriate gray matter (GM) masks.

Methods: In this study, we present a method to perform a specialized GM segmentation developed for the classification of WMH, especially JCWMH. Using T1 and FLAIR images, we propose a pipeline to integrate masks of white matter, cerebrospinal fluid, ventricles, and WMH to create a unique mask to refine the primary GM map. Subsequently, we utilize this pipeline to generate paired data for training a conditional generative adversarial network (cGAN) to substitute the pipeline and reduce the inputs to only FLAIR images. The classification of WMH is then based on the distances between WMH and ventricular and GM masks. Due to the lack of multi-class labeled WMH datasets and the need for extensive data for training deep learning models, we attempted to collect a large local dataset and manually segment and label some data for WMH and ventricles.

Results: In JCWMH classification, the proposed method exhibited a Dice similarity coefficient, precision, and sensitivity of 0.76, 0.69, and 0.84, respectively. With values of 0.66, 0.55, and 0.81, the proposed method clearly outperformed the approach commonly used in the literature, which uses extracted GM masks from registered T1 images on FLAIR.

Conclusion: After training, the method proves its efficiency by providing results in less than one second. In contrast, the usual approach would require at least two minutes for registration and segmentation alone. The proposed method is automated and fast and requires no initialization as it works exclusively with FLAIR images. Such innovative methods will undoubtedly facilitate accurate and meaningful analysis of WMH in clinical practice by reducing complexity and increasing efficiency.