Interpretable detector for cervical cytology using self-attention and cell origin group guidance

被引:3
作者
Jiang, Peng [1 ]
Liu, Juan [1 ]
Feng, Jing [1 ]
Chen, Hua [1 ]
Chen, Yuqi [1 ]
Li, Cheng [2 ]
Pang, Baochuan [2 ]
Cao, Dehua [2 ]
机构
[1] Wuhan Univ, Inst Artificial Intelligence, Sch Comp Sci, Wuhan 430072, Peoples R China
[2] Landing Artificial Intelligence Ctr Pathol Diag, Wuhan 430072, Peoples R China
关键词
Cervical cell detection; Cervical cytology screening; Object detection; Self-attention; Feature fusion; Deep learning; CLASSIFICATION; SEGMENTATION;
D O I
10.1016/j.engappai.2024.108661
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Deep learning has advanced the development of automated cervical cytology, yet limited studies have delved into methods for incorporating medical domain knowledge, and model interpretability has not been thoroughly investigated. To address this issue, this paper proposes a novel, explainable detection method for abnormal cervical cells, called dual -stream self -attention based feature fusion and origin grouping network (DSAFFOGNet). To encourage the model to focus more on lesion cells and cell nuclei of diagnostic significance, the dual -stream self -attention (DSA) module is introduced to enhance the learning of lesion -specific features. In view of the complex background, cell dense distribution, cell overlap, or clumps existing in the actual cervical cytology images, multi -scale features are extracted and fused by using the path aggregation network (PAN) to enhance the feature representation ability. By integrating biomedical insights regarding cell provenance and formulating an origin grouping loss, DSA-FFOGNet adjusts the penalties for cervical cells originating from different groups, thereby enhancing the optimization of the model training process. To further improve the detection performance, the classification and localization tasks are decoupled via the use of double detection heads. Extensive experiments validate the robustness of the proposed DSA-FFOGNet. The visualization of class activation maps (CAMs) showcases the model's interpretability. The proposed approach advances the application and development of explainable artificial intelligence (XAI) models in cervical cytology and inspires further research in automated cervical cytology.
引用
收藏
页数:18
相关论文
共 80 条
[1]  
Basak H., 2021, SN Comput. Sci, V2, P369, DOI [10.1007/s42979-021-00741-2, DOI 10.1007/S42979-021-00741-2]
[2]  
Chai Zhizhong, 2022, 2022 IEEE 19th International Symposium on Biomedical Imaging (ISBI), DOI 10.1109/ISBI52829.2022.9761652
[3]   Automatic cervical cell segmentation and classification in Pap smears [J].
Chankong, Thanatip ;
Theera-Umpon, Nipon ;
Auephanwiriyakul, Sansanee .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2014, 113 (02) :539-556
[4]  
Chen H., 2023, 2023 IEEE INT C BIOI, P1842, DOI 10.1109/BIBM58861.2023.10385911
[5]   CytoBrain: Cervical Cancer Screening System Based on Deep Learning Technology [J].
Chen, Hua ;
Liu, Juan ;
Wen, Qing-Man ;
Zuo, Zhi-Qun ;
Liu, Jia-Sheng ;
Feng, Jing ;
Pang, Bao-Chuan ;
Xiao, Di .
JOURNAL OF COMPUTER SCIENCE AND TECHNOLOGY, 2021, 36 (02) :347-360
[6]  
Chen K, 2019, Arxiv, DOI [arXiv:1906.07155, DOI 10.48550/ARXIV.1906.07155]
[7]   An unsupervised style normalization method for cytopathology images [J].
Chen, Xihao ;
Yu, Jingya ;
Cheng, Shenghua ;
Geng, Xiebo ;
Liu, Sibo ;
Han, Wei ;
Hu, Junbo ;
Chen, Li ;
Liu, Xiuli ;
Zeng, Shaoqun .
COMPUTATIONAL AND STRUCTURAL BIOTECHNOLOGY JOURNAL, 2021, 19 :3852-3863
[8]   Semi-Automatic Segmentation and Classification of Pap Smear Cells [J].
Chen, Yung-Fu ;
Huang, Po-Chi ;
Lin, Ker-Cheng ;
Lin, Hsuan-Hung ;
Wang, Li-En ;
Cheng, Chung-Chuan ;
Chen, Tsung-Po ;
Chan, Yung-Kuan ;
Chiang, John Y. .
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2014, 18 (01) :94-108
[9]   Recent advancement in cervical cancer diagnosis for automated screening: a detailed review [J].
Chitra, B. ;
Kumar, S. S. .
JOURNAL OF AMBIENT INTELLIGENCE AND HUMANIZED COMPUTING, 2022, 13 (01) :251-269
[10]   Inception v3 based cervical cell classification combined with artificially extracted features [J].
Dong, N. ;
Zhao, L. ;
Wu, C. H. ;
Chang, J. F. .
APPLIED SOFT COMPUTING, 2020, 93