BrainNetGAN: Data augmentation of brain connectivity using generative adversarial network for dementia classification

Chao Li,Yiran Wei,Xi Chen,Carola-Bibiane Schonlieb
DOI: https://doi.org/10.48550/arXiv.2103.08494
2021-07-23
Abstract:Alzheimer's disease (AD) is the most common age-related dementia. It remains a challenge to identify the individuals at risk of dementia for precise management. Brain MRI offers a noninvasive biomarker to detect brain aging. Previous evidence shows that the brain structural change detected by diffusion MRI is associated with dementia. Mounting studies has conceptualised the brain as a complex network, which has shown the utility of this approach in characterising various neurological and psychiatric disorders. Therefore, the structural connectivity shows promise in dementia classification. The proposed BrainNetGAN is a generative adversarial network variant to augment the brain structural connectivity matrices for binary dementia classification tasks. Structural connectivity matrices between separated brain regions are constructed using tractography on diffusion MRI data. The BrainNetGAN model is trained to generate fake brain connectivity matrices, which are expected to reflect latent distribution of the real brain network data. Finally, a convolutional neural network classifier is proposed for binary dementia classification. Numerical results show that the binary classification performance in the testing set was improved using the BrainNetGAN augmented dataset. The proposed methodology allows quick synthesis of an arbitrary number of augmented connectivity matrices and can be easily transferred to similar classification tasks.
Machine Learning,Artificial Intelligence,Image and Video Processing
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: **the problem of insufficient training of deep - learning models in Alzheimer's disease (AD) classification due to the limited diffusion magnetic resonance imaging (dMRI) data**. Specifically, Alzheimer's disease is the most common age - related dementia, which significantly affects the quality of life of patients and the economy and society. Although the brain structural network (generated by dMRI) performs well in dementia classification tasks, the limited availability of dMRI data restricts the training effect of deep - learning models. To solve this problem, the author proposes a new variant based on the generative adversarial network (GAN) - BrainNetGAN, which is used to efficiently enhance the brain structural network matrix, thereby improving the performance of dementia classification. ### Main problems: 1. **Insufficient data volume**: Deep - learning models usually require a large amount of training data and balanced class labels, but it is not realistic to obtain sufficient dMRI data in clinical practice. 2. **Traditional data - enhancement methods are not applicable**: Traditional image - enhancement methods (such as rotation, flipping) cannot be directly applied to the brain - network matrix because these operations will change the order of brain regions and introduce noise. ### Solutions: - **BrainNetGAN**: A GAN variant specifically designed to generate realistic brain - connection matrices, which can reflect the potential distribution and topological features of real - brain - network data. - **Wasserstein GAN with Gradient Penalty (WGAN - GP)**: Ensure that the difference between the generated fake samples and the real samples is minimized, and stabilize the training process through gradient penalty. - **Graph neural network (GNN) as a classifier**: Used to evaluate whether the generated data has captured the topological differences between AD and CN brain networks. Through experimental verification, the data generated by BrainNetGAN significantly improves the performance of dementia classification, and the generated fake - brain - network matrices have a high similarity to the real data.