Bio-Inspired Feature Selection in Brain Disease Detection via an Improved Sparrow Search Algorithm

Abstract

The timely diagnosis and treatment of brain diseases have always been an essential part of saving patients with encephalopathy. At the same time, medical image analysis is a significant step in diagnosing brain diseases. This article proposes a brain disease classification approach based on an improved sparrow search algorithm (ISSA). Specifically, a set of image features is first extracted to compose a pool of features for selection from five kinds of brain hemorrhage image datasets and one of the brain tumor image dataset. Second, we design an objective function to jointly reduce the number of the selected features and improve the classification accuracy. Finally, the proposed ISSA is utilized to solve the objective function. ISSA introduces three improvement mechanisms: the tent chaotic initialization, a novel local search strategy, and adaptive crossover operation to enhance the performance of conventional SSA. Moreover, a binary operator is used to solve the discrete feature selection problems. Besides, four machine learning algorithms, which are the K-nearest neighbor (KNN), support vector machine, decision tree, and random forest, are used to calculate the classification accuracy. Experimental results show that the proposed ISSA with KNN classifier eliminates about 65.9% of useless features, while the classification accuracy rate reaches more than 85%, which indicates that the proposed algorithm is more suitable for brain disease analysis than other comparison algorithms and other classifiers. Moreover, the effectiveness of the proposed improved mechanisms and the portability of different classifiers are evaluated by tests, and convolutional neural networks are compared with ISSA to classify the brain computed tomography (CT) images.