Comparison of ELM, LSTM, and CNN Models in Breast Cancer Classification

Authors

  • Silvia Ratna Islamic University of Kalimantan Muhammad Arsyad Al-Banjari, Indonesia
  • M. Muflih Islamic University of Kalimantan Muhammad Arsyad Al-Banjari, Indonesia
  • Haldi Budiman Islamic University of Kalimantan Muhammad Arsyad Al-Banjari, Indonesia
  • Usman Syapotro Islamic University of Kalimantan Muhammad Arsyad Al-Banjari, Indonesia
  • Muhammad Hamdani Islamic University of Kalimantan Muhammad Arsyad Al-Banjari, Indonesia

Keywords:

Breast Cancer Classification, Comparison, Extreme Learning Machine (ELM), Long Short- Term Memory (LSTM), Convolutional Neural Networks (CNN)

Abstract

Classification can significantly impact treatment decisions and patient outcomes. This study evaluates and compares the performance of three machine learning models Extreme Learning Machine (ELM), Long Short-Term Memory (LSTM), and Convolutional Neural Networks (CNN) in breast cancer classification. ELM, known for its fast-learning speed and strong generalization, is compared with LSTM, which is effective in capturing long-term dependencies in sequential data, and CNN, which is renowned for its ability to automatically extract features from images and structured data. The models were trained and tested on a breast cancer dataset, focusing on accuracy and computational efficiency. The results revealed that while CNNs demonstrated better accuracy in feature-rich data, LSTMs excelled in handling sequential data patterns. On the other hand, ELM offers a good balance between training speed and classification performance. This comparative analysis provides valuable insights into the strengths and limitations of each model, contributing to the development of more effective breast cancer diagnostic tools. In this case, LSTM outperformed ELM by 0.91%, outperformed
CNN significantly by 3.72%, and outperformed Improved LSTM by 0.91%. This indicate that the LSTM model shows higher accuracy in breast cancer classification.

Published

2024-11-26

How to Cite

Ratna, S., Muflih, M., Budiman, H., Syapotro, U., & Hamdani, M. (2024). Comparison of ELM, LSTM, and CNN Models in Breast Cancer Classification. Journal of Data Science, 2024. Retrieved from https://iuojs.intimal.edu.my/index.php/jods/article/view/595