Machine learning based gastric cancer computer-aided diagnosis system using feature selection

초록

Gastric cancer is a kind of cancer that is difficult to detect at an early stage because it has almost no symptoms at the beginning. In this study, we propose a Computer-aided Diagnosis(CADx) system that detects gastric cancer from the endoscopy. The data set we used consist of 93 normal images and 93 gastric images. We extracted 6 features in 449 dimensions from the gastric endoscopy images and reduced them to 10 dimensions through feature selection algorithms. Algorithms that we use to dimension reduction are Pearson Correlation, Chi-Squared Test, Recursive Feature Elimination, and Model-based Feature Selection, which are provided by the Sci-kit Learn library. A method was also used to select the top 10 features with a higher number of times selected by these four algorithms. Normal images and gastric cancer images were classified using support vector machine(SVM). Recursive feature elimination algorithm has the highest performance among the five feature selection algorithms, with an accuracy of 0.92. © 2020 Korean Institute of Electrical Engineers. All rights reserved.

키워드