본문 바로가기
  • Home

Transfer Learning based DNN-SVM Hybrid Model for Breast Cancer Classification

  • Journal of The Korea Society of Computer and Information
  • Abbr : JKSCI
  • 2023, 28(11), pp.1-11
  • DOI : 10.9708/jksci.2023.28.11.001
  • Publisher : The Korean Society Of Computer And Information
  • Research Area : Engineering > Computer Science
  • Received : August 21, 2023
  • Accepted : October 24, 2023
  • Published : November 30, 2023

Gui Rae Jo 1 Beomsu Baek 1 Young-Soon Kim 1 Dong Hoon Lim 1

1경상국립대학교

Accredited

ABSTRACT

Breast cancer is the disease that affects women the most worldwide. Due to the development of computer technology, the efficiency of machine learning has increased, and thus plays an important role in cancer detection and diagnosis. Deep learning is a field of machine learning technology based on an artificial neural network, and its performance has been rapidly improved in recent years, and its application range is expanding. In this paper, we propose a DNN-SVM hybrid model that combines the structure of a deep neural network (DNN) based on transfer learning and a support vector machine (SVM) for breast cancer classification. The transfer learning-based proposed model is effective for small training data, has a fast learning speed, and can improve model performance by combining all the advantages of a single model, that is, DNN and SVM. To evaluate the performance of the proposed DNN-SVM Hybrid model, the performance test results with WOBC and WDBC breast cancer data provided by the UCI machine learning repository showed that the proposed model is superior to single models such as logistic regression, DNN, and SVM, and ensemble models such as random forest in various performance measures.

Citation status

* References for papers published after 2023 are currently being built.