MultiSCCHisto-Net-KD: A deep network for multi-organ explainable squamous cell carcinoma diagnosis with knowledge distillation

被引:0
|
作者
Prabhu, Swathi [1 ]
Prasad, Keerthana [2 ]
Hoang, Thuong [3 ]
Lu, Xuequan [4 ]
机构
[1] Department of Information and Communication Technology, Manipal Institute of Technology, Manipal Academy of Higher Education, Karnataka, Manipal
[2] Manipal School of Information Sciences, Manipal Academy of Higher Education, Karnataka, Manipal
[3] School of Information Technology, Faculty of Science Engineering and Built Environment, Deakin University, Geelong, VIC 3220, VIC
[4] Department of Computer Science and IT, La Trobe University, Melbourne, VIC 3086, VIC
关键词
Attention module; Explainable deep learning; Histopathological images; Image classification; Knowledge distillation; Squamous cell carcinoma;
D O I
10.1016/j.compbiomed.2024.109469
中图分类号
学科分类号
摘要
Squamous cell carcinoma is a prevalent cancer type that affects various organs in the human body. Manual analysis for detecting squamous cell carcinoma in histopathological images is time-consuming and may be subjective. Squamous cell carcinoma diagnosis is typically based on the differences in the architectural arrangement of squamous epithelial layers and the presence of keratinization. However, the existing literature has predominantly concentrated on identifying cellular irregularities with high magnification images and considering specific organs of squamous cell carcinoma origin. In contrast, relatively little attention has been given to recognizing structural abnormalities observable at low magnification images. In this paper, we consider squamous cell carcinoma histopathological images across different organs of origin captured at low magnification and these images are gathered from various centers to develop a robust model. We propose a novel deep neural network model (MultiSCCHisto-Net) that can detect squamous cell carcinoma of any organ irrespective of organ of origin. In addition, deep neural networks used for histopathological image analysis typically have many parameters, making them computationally expensive. To address this research gap, we incorporate knowledge distillation, which compresses knowledge from a complex teacher model (MultiSCCHisto-Net) into a smaller student model (MultiSCCHisto-Net-KD) while preserving performance and enhancing the generalization of the student model by learning from the teacher's intermediate layer features. Moreover, an explainable deep learning technique called gradient-weighted class activation mapping is incorporated to highlight the image areas that help to classify the sample into particular classes. This explainability significantly enhances our confidence in the proposed model outcomes. We evaluate the model's robustness using private multi-centric and publicly available datasets. Our results show that accuracy rates of 97% and 93% are achieved on private and public datasets, respectively, surpassing the performance of state-of-the-art models. © 2024 The Authors
引用
收藏
相关论文
共 3 条
  • [1] Multi-organ squamous cell carcinoma classification using feature interpretation technique for explainability
    Prabhu, Swathi
    Prasad, Keerthana
    Hoang, Thuong
    Lu, Xuequan
    Sandhya, I
    BIOCYBERNETICS AND BIOMEDICAL ENGINEERING, 2024, 44 (02) : 312 - 326
  • [2] Abdominal multi-organ segmentation in Multi-sequence MRIs based on visual attention guided network and knowledge distillation
    Fu, Hao
    Zhang, Jian
    Li, Bin
    Chen, Lanlan
    Zou, Junzhong
    Zhang, ZhuiYang
    Zou, Hao
    PHYSICA MEDICA-EUROPEAN JOURNAL OF MEDICAL PHYSICS, 2024, 122
  • [3] Efficient Multi-Organ Segmentation From 3D Abdominal CT Images With Lightweight Network and Knowledge Distillation
    Zhao, Qianfei
    Zhong, Lanfeng
    Xiao, Jianghong
    Zhang, Jingbo
    Chen, Yinan
    Liao, Wenjun
    Zhang, Shaoting
    Wang, Guotai
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2023, 42 (09) : 2513 - 2523