An interpretable framework for gastric cancer classification using multi-channel attention mechanisms and transfer learning approach on histopathology images.

Journal: Scientific reports
PMID:

Abstract

The importance of gastric cancer (GC) and the role of deep learning techniques in categorizing GC histopathology images have recently increased. Identifying the drawbacks of traditional deep learning models, including lack of interpretability, inability to capture complex patterns, lack of adaptability, and sensitivity to noise. A multi-channel attention mechanism-based framework is proposed that can overcome the limitations of conventional deep learning models by dynamically focusing on relevant features, enhancing extraction, and capturing complex relationships in medical data. The proposed framework uses three different attention mechanism channels and convolutional neural networks to extract multichannel features during the classification process. The proposed framework's strong performance is confirmed by competitive experiments conducted on a publicly available Gastric Histopathology Sub-size Image Database, which yielded remarkable classification accuracies of 99.07% and 98.48% on the validation and testing sets, respectively. Additionally, on the HCRF dataset, the framework achieved high classification accuracy of 99.84% and 99.65% on the validation and testing sets, respectively. The effectiveness and interchangeability of the three channels are further confirmed by ablation and interchangeability experiments, highlighting the remarkable performance of the framework in GC histopathological image classification tasks. This offers an advanced and pragmatic artificial intelligence solution that addresses challenges posed by unique medical image characteristics for intricate image analysis. The proposed approach in artificial intelligence medical engineering demonstrates significant potential for enhancing diagnostic precision by achieving high classification accuracy and treatment outcomes.

Authors

  • Muhammad Zubair
    Swedish University of Agricultural Sciences, Department of Plant Breeding and Biotechnology Balsgård, Fjälkestadsvägen 459, SE-291 94 Kristianstad, Sweden.
  • Muhammad Owais
    Division of Electronics and Electrical Engineering, Dongguk University, 30 Pildong-ro 1-gil, Jung-gu, Seoul 04620, Korea.
  • Taimur Hassan
    Department of Computer Engineering, National University of Sciences and Technology, Islamabad, Pakistan.
  • Malika Bendechache
    School of Computing, Faculty of Engineering and Computing, Dublin City University, Ireland.
  • Muzammil Hussain
    Department of Computer Science, School of Systems and Technology, University of Management and Technology, Lahore 54000, Pakistan.
  • Irfan Hussain
  • Naoufel Werghi
    Khalifa University of Science and Technology, PO Box 127788, Abu Dhabi, United Arab Emirates.