Due to the lack of devices and the difficulty of gathering patients, the small sample size is one of the most challenging problems in functional brain network (FBN) analysis. Previous studies have attempted to solve this problem of sample limitation through data augmentation methods, such as sample transformation and noise addition. However, these methods ignore the unique spatial-temporal information of functional magnetic resonance imaging (fMRI) data, which is essential for FBN analysis.
To address this issue, we propose a spatial-temporal data-augmentation-based classification (STDAC) scheme that can fuse the spatial-temporal information, increase the samples, while improving the classification performance. Firstly, we propose a spatial augmentation module utilizing the spatial prior knowledge, which was ignored by previous augmentation methods. Secondly, we design a temporal augmentation module by random discontinuous sampling period, which can generate more samples than former approaches. Finally, a tensor fusion method is used to combine the features from the above two modules, which can make efficient use of spatial-temporal information of fMRI simultaneously. Besides, we apply our scheme to different types of classifiers to verify the generalization performance. To evaluate the effectiveness of our proposed scheme, we conduct extensive experiments on the Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset and REST-meta-MDD Project (MDD) dataset.
Experimental results show that the proposed scheme achieves superior classification accuracy (ADNI: 82.942%, MDD: 63.406%) and feature interpretation on the benchmark datasets.
The proposed STDAC scheme, utilizing both spatial and temporal information, can generate more diverse samples than former augmentation methods for brain disorder classification and analysis.