Abstract
Brain network classification has attracted increasing attention with the widespread application in the automatic diagnosis of brain diseases. However, limited by the higher cost of detecting and marking for medical imaging, the amount of brain network data is usually small, which largely restricts the performance of current brain network classification models. In this paper, we propose a new sparse data augmentation model (SDAM) based on EncoderForest to effectively enhance the brain network data and improve the classification performance. The EncoderForest based SDAM uses a generator which innovatively encodes the rules of a set of parallel decision trees to generate sparse data with only discriminative connections. The generated data expands the original data set effectively by utilizing the advantages of EncoderForest in learning data feature sparsely and constructing a feature association generation model compactly. In addition, the SDAM is flexible to combine with different classification models, such as random forest, support vector machine, deep neural network, etc. The experimental results on three common brain disease data sets show that our model is able to reasonably augment the brain network data and remarkably improve the performance of various classifiers.
Similar content being viewed by others
Notes
http://fcon_1000.projects.nitrc.org/indi/adhd200/
References
Glasser MF, Smith SM, Marcus DS, Andersson JL, Auerbach EJ, Behrens TE, Coalson TS, Harms MP, Jenkinson M, Moeller S et al (2016) The human connectome project’s neuroimaging approach. Nature Neurosci 19(9):1175
LIANG X, WANG J (2010) Human connectome: Structural and functional brain networks. Chin Sci Bull 55(16):1565– 1583
Kim J, Calhoun VD, Shim E, Lee JH (2016) Deep neural network with weight sparsity control and pre-training extracts hierarchical features and enhances classification performance: Evidence from whole-brain resting-state functional connectivity patterns of schizophrenia. Neuroimage 124:127–146
Zhu D, Li X, Jiang X, Chen H, Shen D, Liu T (2013) Exploring high-order functional interactions via structurally-weighted lasso models. In: International conference on information processing in medical imaging. Springer, pp 13–24
Xiao X, Fang H, Wu J, Xiao C, Xiao T, Qian L, Liang F, Xiao Z, Chu KK, Ke X (2017) Diagnostic model generated by mri-derived brain features in toddlers with autism spectrum disorder. Autism Res 10(4):620–630
Heinsfeld AS, Franco AR, Craddock RC, Buchweitz A, Meneguzzi F (2018) Identification of autism spectrum disorder using deep learning and the abide dataset. NeuroImage Clin 17:16–23
Kawahara J, Brown CJ, Miller SP, Booth BG, Chau V, Grunau RE, Zwicker JG, Hamarneh G (2017) Brainnetcnn: Convolutional neural networks for brain networks; towards predicting neurodevelopment. NeuroImage 146:1038–1049
Xing X, Ji J, Yao Y (2018) Convolutional neural network with element-wise filters to extract hierarchical topological features for brain networks. In: 2018 IEEE international conference on bioinformatics and biomedicine (BIBM). IEEE, pp 780– 783
Frid-Adar M, Klang E, Amitai M, Goldberger J, Greenspan H (2018) Synthetic data augmentation using gan for improved liver lesion classification. In: 2018 IEEE 15th international symposium on biomedical imaging (ISBI 2018), IEEE, pp 289–293
Zhuang P, Schwing AG, Koyejo O (2019) Fmri data augmentation via synthesis. In: 2019 IEEE 16th International symposium on biomedical imaging (ISBI 2019). IEEE, pp 1783–1787
Gao R, Peng J, Nguyen L, Liang Y, Thng S, Lin Z (2019) Classification of non-tumorous facial pigmentation disorders using deep learning and smote. In: 2019 IEEE International symposium on circuits and systems (ISCAS). IEEE, pp 1–5
Meszlényi R J, Buza K, Vidnyánszky Z (2017) Resting state fmri functional connectivity-based classification using a convolutional neural network architecture. Front Neuroinform 11:61
Wee CY, Yap PT, Zhang D, Wang L, Shen D (2014) Group-constrained sparse fmri connectivity modeling for mild cognitive impairment identification. Brain Struct Funct 219(2):641–656
Rasmussen PM, Hansen LK, Madsen KH, Churchill NW, Strother SC (2012) Model sparsity and brain pattern interpretation of classification models in neuroimaging. Pattern Recogn 45(6):2085–2100
Feng J, Zhou ZH (2018) Autoencoder by forest. In: Thirty-Second AAAI conference on artificial intelligence
Steinberg D, Colla P (2009) Cart: classification and regression trees. Top Ten Alg Data Mining 9:179
Di Martino A, Yan CG, Li Q, Denio E, Castellanos FX, Alaerts K, Anderson JS, Assaf M, Bookheimer SY, Dapretto M et al (2014) The autism brain imaging data exchange: towards a large-scale evaluation of the intrinsic brain architecture in autism. Mol Psychiatry 19(6):659
Craddock C, Benhajali Y, Chu C, Chouinard F, Evans A, Jakab A, Khundrakpam BS, Lewis JD, Li Q, Milham M et al (2013) The neuro bureau preprocessing initiative: open sharing of preprocessed neuroimaging data and derivatives. Front Neuroinform 7
Jilka SR, Scott G, Ham T, Pickering A, Bonnelle V, Braga RM, Leech R, Sharp DJ (2014) Damage to the salience network and interactions with the default mode network. J Neurosci 34(33):10798–10807
Chen T, Cai W, Ryali S, Supekar K, Menon V (2016) Distinct global brain dynamics and spatiotemporal organization of the salience network. PLoS Biol 14(6):e1002469
Beaty RE, Benedek M, Silvia PJ, Schacter DL (2016) Creative cognition and brain network dynamics. Trends Cognit Sci 20(2):87–95
Raichle ME (2015) The brain’s default mode network. Ann Rev Neurosci 38:433–447
Elton A, Di Martino A, Hazlett HC, Gao W (2016) Neural connectivity evidence for a categorical-dimensional hybrid model of autism spectrum disorder. Biol Psychiatry 80(2):120–128
Yamada T, Itahashi T, Nakamura M, Watanabe H, Kuroda M, Ohta H, Kanai C, Kato N, Ri H (2016) Altered functional organization within the insular cortex in adult males with high-functioning autism spectrum disorder: evidence from connectivity-based parcellation. Mol Autism 7(1):41
Bigler ED, Mortensen S, Neeley ES, Ozonoff S, Krasny L, Johnson M, Lu J, Provencal SL, McMahon W, Lainhart JE (2007) Superior temporal gyrus, language function, and autism. Dev Neuropsychol 31(2):217–238
Gebauer L, Skewes J, Hørlyck L, Vuust P (2014) Atypical perception of affective prosody in autism spectrum disorder. Neuroimage Clin 6:370–378
Acknowledgment
This work is partly supported by National Natural Science Foundation of China Research Program 61672065.
Author information
Authors and Affiliations
Corresponding author
Additional information
Publisher’s note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Ji, J., Wang, Z., Zhang, X. et al. Sparse data augmentation based on encoderforest for brain network classification. Appl Intell 52, 4317–4329 (2022). https://doi.org/10.1007/s10489-021-02579-w
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10489-021-02579-w