A hybrid model using teaching-learning-based optimization and Salp swarm algorithm for feature selection and classification in digital mammography

摘要

Feature selection is the most important step in the design of a breast cancer diagnosis system. The basic objective of the proposed methodology is to reduce the size of the feature space to improve the performance of the classification system. In this article, a hybrid teaching-learning based optimization (TLBO) with a Salp swarm algorithm (SSA) is presented to select the features with an artificial neural network as a fitness evaluator. The features selected by TLBO-SSA are evaluated using an adaptive neuro-fuzzy inference system. The performance of the proposed methodology is tested over 651 mammograms. The experimental results show that TLBO-SSA appears to be the best when compared with the basic TLBO algorithm. TLBO-SSA archived an accuracy of 98.46% with 98.81% sensitivity, 98.08% specificity, 0.9852 F-score, 0.9692 Cohen's kappa coefficient, and area under curve A(Z) = 0.997 +/- 0.001. Again the robustness of the proposed TLBO-SSA method is tested using a benchmark dataset obtained from the UCI repository. The result obtained by TLBO-SSA is compared with the Genetic Algorithm. The results show that TLBO-SSA is better than the Genetic Algorithm.