Optimization of CNN Model for Breast Cancer Classification

Nikolay Mikhailov, Mariam Shakeel, Abdybek Urmanov, Min Ho Lee, M. Fatih Demirci

Research output: Chapter in Book/Report/Conference proceedingConference contribution

3 Citations (Scopus)

Abstract

Application of deep learning techniques for breast cancer classification using histopathology images has gained interest during recent years. In this study, an open-source convolutional neural network (CNN) model developed for breast cancer classification model is optimized by performing sensitivities on various CNN parameters such as data balancing, activation functions and adding/removing CNN layers. Some of the parameters are less-sensitive in affecting model's performance. The results show that by balancing the number of positive and negative samples, accuracy of the model can be improved. However, some additional work is required to reach to that point. Furthermore, the computation time is reduced by almost 30% by increasing the learning rate from 0.01 to 0.05 while the training and validation accuracy and loss are comparable to that of the original CNN model.

Original languageEnglish
Title of host publicationProceedings - 2021 16th International Conference on Electronics Computer and Computation, ICECCO 2021
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781665409452
DOIs
Publication statusPublished - 2021
Event16th International Conference on Electronics Computer and Computation, ICECCO 2021 - Kaskelen, Kazakhstan
Duration: Nov 25 2021Nov 26 2021

Publication series

NameProceedings - 2021 16th International Conference on Electronics Computer and Computation, ICECCO 2021

Conference

Conference16th International Conference on Electronics Computer and Computation, ICECCO 2021
Country/TerritoryKazakhstan
CityKaskelen
Period11/25/2111/26/21

Keywords

  • activation function
  • breast cancer
  • convolutional neural network
  • data balancing
  • deep learning

ASJC Scopus subject areas

  • Computer Science Applications
  • Electrical and Electronic Engineering
  • Control and Optimization
  • Modelling and Simulation
  • Education
  • Artificial Intelligence
  • Computer Networks and Communications

Fingerprint

Dive into the research topics of 'Optimization of CNN Model for Breast Cancer Classification'. Together they form a unique fingerprint.

Cite this