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

被引:5
作者
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 条
[71]  
Yan Xiaoli, 2021, Bio-Inspired Computing: Theories and Applications: 15th International Conference, BIC-TA 2020. Communications in Computer and Information Science (1363), P406, DOI 10.1007/978-981-16-1354-8_29
[72]  
Yang LX, 2021, PR MACH LEARN RES, V139
[73]  
Yi L, 2020, LECT NOTES COMPUT SC, V12306, P602, DOI 10.1007/978-3-030-60639-8_50
[74]   DCCL: A Benchmark for Cervical Cytology Analysis [J].
Zhang, Changzheng ;
Liu, Dong ;
Wang, Lanjun ;
Li, Yaoxin ;
Chen, Xiaoshi ;
Luo, Rui ;
Che, Shuanlong ;
Liang, Hehua ;
Li, Yinghua ;
Liu, Si ;
Tu, Dandan ;
Qi, Guojun ;
Luo, Pifu ;
Luo, Jiebo .
MACHINE LEARNING IN MEDICAL IMAGING (MLMI 2019), 2019, 11861 :63-72
[75]   Auxiliary classification of cervical cells based on multi-domain hybrid deep learning framework [J].
Zhang, Chuanwang ;
Jia, Dongyao ;
Li, Ziqi ;
Wu, Nengkai .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 77
[76]   DeepPap: Deep Convolutional Networks for Cervical Cell Classification [J].
Zhang, Ling ;
Lu, Le ;
Nogues, Isabella ;
Summers, Ronald M. ;
Liu, Shaoxiong ;
Yao, Jianhua .
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2017, 21 (06) :1633-1643
[77]   Classification and Localization Consistency Regularized Student-Teacher Network for Semi-supervised Cervical Cell Detection [J].
Zhang, Menglu ;
Li, Xuechen ;
Shen, Linlin .
2021 IEEE 34TH INTERNATIONAL SYMPOSIUM ON COMPUTER-BASED MEDICAL SYSTEMS (CBMS), 2021, :289-294
[78]   ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices [J].
Zhang, Xiangyu ;
Zhou, Xinyu ;
Lin, Mengxiao ;
Sun, Ran .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :6848-6856
[79]   DC-SSL: Addressing Mismatched Class Distribution in Semi-supervised Learning [J].
Zhao, Zhen ;
Zhou, Luping ;
Duan, Yue ;
Wang, Lei ;
Qi, Lei ;
Shi, Yinghuan .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, :9747-9755
[80]   Learning Deep Features for Discriminative Localization [J].
Zhou, Bolei ;
Khosla, Aditya ;
Lapedriza, Agata ;
Oliva, Aude ;
Torralba, Antonio .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :2921-2929