MSPO: A machine learning hyperparameter optimization method for enhanced breast cancer image classification.

Journal: Digital health
Published Date:

Abstract

As one of the major threats to women's health worldwide, breast cancer requires early diagnosis and accurate classification, since they are key to optimizing therapeutic interventions and ensuring precise prognosis. Recently, deep learning has demonstrated notable advantages in breast cancer image classification. However, their performance heavily relies on the proper configuration of hyperparameters. To overcome the inefficiencies and weaknesses of conventional hyperparameter optimization methods, like limited effectiveness and vulnerability to premature convergence, this research proposes a Multi-Strategy Parrot Optimizer (MSPO) and applies it to breast cancer image classification tasks. Based on the original Parrot Optimizer, MSPO integrates several strategies, including Sobol sequence initialization, nonlinear decreasing inertia weight, and a chaotic parameter to enhance global exploration ability and convergence steadiness. Tests using the CEC 2022 benchmark functions reveal that MSPO surpasses leading algorithms regarding optimization precision and convergence rate. An ablation study was conducted on three variants of MSPO through CEC 2022 to further validate the effectiveness of each key strategy. Furthermore, MSPO is combined with the ResNet18 model and applied to the BreaKHis breast cancer image dataset. Results indicate that the model optimized by MSPO notably surpasses both the non-optimized version and other alternative optimization algorithms using four assessment indicators: accuracy, precision, recall, and F1-score. This validates the promising application potential and practical significance of MSPO in medical image classification tasks.

Authors

  • Haonan Li
    Department of Biotechnology, College of Basic Medical Sciences, Dalian Medical University, China (H.L., J.W.).
  • Vijay Govindarajan
    Distribution and Supply Technology, Expedia Group, Seattle, WA, USA.
  • Tan Fong Ang
    Center of Research for Cyber Security and Network (CSNET), Faculty of Computer Science and Information Technology, Universiti Malaya, Kuala Lumpur, Wilayar Persekutuan, Malaysia.
  • Zaffar Ahmed Shaikh
    Department of Computer Science and Information Technology, Benazir Bhutto Shaheed University Lyari, Karachi, 75660, Pakistan. zashaikh@bbsul.edu.pk.
  • Amel Ksibi
    Department of Information Systems, College of Computer and Information Sciences, Princess Nourah bint Abdulrahman University, P.O.Box 84428, Riyadh 11671, Saudi Arabia.
  • Yen-Lin Chen
    Department of Neuromuscular Research, National Institute of Neuroscience, National Center of Neurology and Psychiatry, Tokyo, Japan.
  • Chin Soon Ku
    Department of Computer Science, Universiti Tunku Abdul Rahman, 31900, Kampar, Malaysia. kucs@utar.edu.my.
  • Ming Chern Leong
    Paediatric & Congenital Heart Centre, Institut Jantung Negara, 145, Jalan Tun Razak, 51200, Kuala Lumpur, Malaysia. mcleong@ijn.com.my.
  • Fatiha Hana Shabaruddin
    Faculty of Pharmacy, University Malaya, Kuala Lumpur, Malaysia.
  • Wan Zamaniah Wan Ishak
    Clinical Oncology Unit, Faculty of Medicine, University of Malaya, Kuala Lumpur, Malaysia.
  • Lip Yee Por
    Department of Computer System and Technology, Faculty of Computer Science and Information Technology, University of Malaya, Kuala Lumpur 50603, Malaysia.

Keywords

No keywords available for this article.