Cancer detection and classification using a simplified binary state vector machine

Med Biol Eng Comput. 2024 May;62(5):1491-1501. doi: 10.1007/s11517-023-03012-9. Epub 2024 Feb 1.

Abstract

Cancer is an invasive and malignant growth of cells and is known to be one of the most fatal diseases. Its early detection is essential for decreasing the mortality rate and increasing the probability of survival. This study presents an efficient machine learning approach based on the state vector machine (SVM) to diagnose and classify tumors into malignant or benign cancer using the online lymphographic data. Further, two types of neural network architectures are also implemented to evaluate the performance of the proposed SVM-based approach. The optimal structures of the classifiers are obtained by varying the architecture, topology, learning rate, and kernel function and recording the results' accuracy. The classifiers are trained with the preprocessed data examples after noise removal and tested on the unknown cases to diagnose each example as positive or negative. Further, the positive cases are classified into different stages including metastases, malign lymph, and fibrosis. The results are evaluated against the feed-forward and generalized regression neural networks. It is found that the proposed SVM-based approach significantly improves the early detection and classification accuracy in comparison to the experienced physicians and the other machine learning approaches. The proposed approach is robust and can perform sub-class divisions for multipurpose tasks. Experimental results demonstrate that the two-class SVM gives the best results and can effectively be used for the classification of cancer. It has outperformed all other classifiers with an average accuracy of 94.90%.

Keywords: Backpropagation; Cancer detection; Generalized regression; Malign lymph; Metastases; Support vector machine.

MeSH terms

  • Algorithms
  • Machine Learning
  • Neoplasms* / diagnosis
  • Neural Networks, Computer
  • Probability
  • Support Vector Machine*