Improving model robustness via enhanced feature representation and sample distribution based on cascaded classifiers for computer-aided diagnosis of brain disease

Chunying Pang,Yu Zhang,Zeyu Xue,Jian Bao,Bing Keong Li,Yan Liu,Yuqi Liu,Mao Sheng,Bo Peng,Yakang Dai
DOI: https://doi.org/10.1016/j.bspc.2022.104047
IF: 5.1
2023-01-01
Biomedical Signal Processing and Control
Abstract:Autism spectrum disorders (ASD) and Parkinson’s Diseases (PD) are both neurological brain diseases that can affect patient’s neural activity. The significance of computer-aided diagnosis is to assist doctors in diagnosis and intervention of the brain diseases in time. At present, machine learning approaches have been studied for computer-aided diagnosis of brain diseases, among which cascaded multi-column classifier is one of the most effective methods. However, influence of feature representation and sample distribution limits the improvement of the classification robustness and accuracy. In this work, we proposed optimiazed cascaded classifiers via enhanced feature representation and sample distribution for improving model robusteness in computer-aided diagnosis of brain disease. Specifically, we firstly extract brain connectivity features from structural and functional magnetic resonance images. The extracted features are enhanced using empirical kernel mapping to make them have better distribution in feature space. Then the enhanced features are sent into the bottommost classifier, and the predicted labels together with the enhanced features are put into the upper layer classifier to form a cascaded classification framework. During the training process, self-paced learning is used to improve model robustness by adjusting sample distribution. The experimental results on ASD and PD datasets show that the proposed method has an outstanding classification performance compared with existing methods. It suggests that enhanced feature expression and sample distrubution can help improve the model robustness and classification performance, which is meaningful for diagnosis of brain diseases.
engineering, biomedical
What problem does this paper attempt to address?