A robust transfer learning approach for colorectal cancer identification based on histopathology images

Toto Haryanto, Helmi Al Farel, Heru Suhartanto, Kusmardi Kusmardi, Marina Yusoff, Jasni Mohamad Zain, Ari Wibisono

Research output: Contribution to journalArticlepeer-review

Abstract

Objective: Diagnosis of cancer at the benign stage is crucial. Recently, pathologists have been using computer-aided diagnostics with machine learning to diagnose patients from medical images. The limitation of the medical image dataset is a challenge to obtain a robust model for cancer identification. This research was conducted to address this issue. The weight of the pre-trained model can be learned more efficiently with a limited amount of data. However, the training process is conducted by customizing the freeze rate so that feature extraction is preserved besides tuning some hyperparameters. Methods: Transfer learning is a technique that can handle data limitation issues in the medical field. Transfer learning will reuse pre-trained model for different or specific task. Choosing an optimum architecture and hyperparameters in machine learning is very important to improve model performance. In our experiment, we carried out a hyperparameter optimization of various deep learning architectures that classifies images containing healthy and cancer tissue. Result: The research concludes that CNN with architecture DenseNet121, freeze rate 75%, zero hidden layers on the classifier, learning rate 0.001, and optimizer RMSProp have the best performance with 98% accuracy and 19.5 s training time using NVIDIA A100 GPU accelerator. Testing with the real dataset for future direction will be an achievement for the model’s success. Conclusion: This research has successfully optimized the Densnet121 deep learning architecture by tuning parameters. With a harmonic means value of 0.98, DenseNet121 outperforms the others. Compared to Camelyon, ImageNet still becomes the baseline of the transfer learning dataset because it has a rich amount of data.

Original languageEnglish
Pages (from-to)753-767
Number of pages15
JournalResearch on Biomedical Engineering
Volume40
Issue number3-4
DOIs
Publication statusPublished - Oct 2024

Keywords

  • Colorectal cancer
  • Deep learning
  • Diagnostic accuracy
  • Histopathology image
  • Transfer learning

Fingerprint

Dive into the research topics of 'A robust transfer learning approach for colorectal cancer identification based on histopathology images'. Together they form a unique fingerprint.

Cite this