Brain Network Diffusion-Driven fMRI Connectivity Augmentation for Enhanced Autism Spectrum Disorder Diagnosis

Haokai Zhao,Haowei Lou,Lina Yao,Yu Zhang
2024-09-11
Abstract:Functional magnetic resonance imaging (fMRI) is an emerging neuroimaging modality that is commonly modeled as networks of Regions of Interest (ROIs) and their connections, named functional connectivity, for understanding the brain functions and mental disorders. However, due to the high cost of fMRI data acquisition and labeling, the amount of fMRI data is usually small, which largely limits the performance of recognition models. With the rise of generative models, especially diffusion models, the ability to generate realistic samples close to the real data distribution has been widely used for data augmentations. In this work, we present a transformer-based latent diffusion model for functional connectivity generation and demonstrate the effectiveness of the diffusion model as an augmentation tool for fMRI functional connectivity. Furthermore, extended experiments are conducted to provide detailed analysis of the generation quality and interpretations for the learned feature pattern. Our code will be made public upon acceptance.
Neurons and Cognition,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper attempts to address the issue of improving the accuracy of Autism Spectrum Disorder (ASD) diagnosis in the context of scarce functional Magnetic Resonance Imaging (fMRI) data. Specifically, due to the high cost of fMRI data acquisition and the need for a specialized operational environment, the amount of data available for training is limited. This limitation restricts the performance and generalization ability of machine learning models in ASD diagnosis. To solve this problem, the research team proposed a Transformer-based latent diffusion model—Brain-Net-Diffusion, which is used to generate realistic functional connectivity matrices to augment the dataset. This method aims to increase the number and diversity of training samples through data augmentation techniques, thereby improving the performance of ASD diagnosis models. Additionally, the paper discusses how to overcome challenges such as the non-Euclidean nature of functional connectivity data, distribution mismatch, and small sample sizes to ensure that the generated data can effectively enhance the performance of downstream classification tasks.