计算机工程与应用 ›› 2022, Vol. 58 ›› Issue (2): 193-200.DOI: 10.3778/j.issn.1002-8331.2008-0084

• 模式识别与人工智能 • 上一篇    下一篇

改进的M2det内窥镜息肉检测方法

王博,张丽媛,师为礼,杨华民,蒋振刚   

  1. 长春理工大学 计算机科学技术学院,长春 130022
  • 出版日期:2022-01-15 发布日期:2022-01-18

Improved M2det Endoscopic Polyp Detection Method

WANG Bo, ZHANG Liyuan, SHI Weili, YANG Huamin, JIANG Zhengang   

  1. College of Computer Science and Technology, Changchun University of Science and Technology, Changchun 130022, China
  • Online:2022-01-15 Published:2022-01-18

摘要: 结直肠癌是一种致命的疾病,作为息肉的肠腺瘤被认为是结直肠癌的主要病因,因此在临床诊断中发现肠息肉是一项至关重要的任务。息肉检测通常由医生操作内窥镜来实现,由于肠道环境复杂,息肉影像数据量大,小尺度息肉不易辨识,息肉检查过程除了极其依赖医生经验之外,工作压力和强度也给医生带来了极大的负担,因此需要借助计算机辅助诊断技术来检测息肉,该技术可以有效地处理大量的息肉影像数据、发现早期息肉、提高息肉检测的准确率。目前的一些方法对小型息肉存在漏检,因此提出了一种改进的M2det方法用于息肉检测,通过FFMs模块融合主干网络特征,使图像特征得到了充分利用,在SFAM模块中增加scSENet注意力机制,保留了有效特征,抑制无用特征,采用Focal loss计算分类损失,解决了正负样本不平衡问题。大量实验表明,该方法可以有效地检测出息肉且优于前沿的息肉检测方法,在CVC15数据集上mAP、F1-score、F2-score分别提升到了98.25%、97.30%、97.98%。

关键词: 息肉检测, 计算机辅助诊断, M2det, 特征融合模块(FFMs), 空间和通道上的压缩激励网络(scSENet), Focal loss

Abstract: Colorectal cancer is a fatal disease. Intestinal adenoma, which is a polyp, is considered to be the main cause of colorectal cancer. Therefore, finding intestinal polyps in clinical diagnosis is a vital task. Polyp detection is usually achieved by doctors operating an endoscope. Due to the complex intestinal environment, the large amount of polyp imaging data, and the difficulty of identifying small-scale polyps, the polyp examination process is extremely dependent on the doctor’s experience, and the work pressure and intensity also bring the doctors. Therefore, it is necessary to use CAD to detect polyps. This technology can effectively process a large amount of polyp imaging data, find early polyps, and improve the accuracy of polyp detection. Some current methods fail to detect small polyps. Therefore, it proposes an improved M2det method for polyp detection. First, the FFMs module is used to fuse the backbone network features to make full use of the image features. Secondly, scSENet is added to the SFAM module. The attention mechanism retains effective features and suppresses useless features. Finally, Focal loss is used to calculate the classification loss, which solves the problem of imbalance between positive and negative samples. A large number of experiments show that this method can effectively detect polyps and is better than cutting-edge polyp detection methods. On the CVC15 data set, mAP, F1-score, and F2-score have been increased to 98.25%, 97.30%, and 97.98%, respectively.

Key words: polyp detection, computer-aided diagnosis, M2det, feature fusion module(FFMs), spatial and channel squeeze-and-excitation networks(scSENet), Focal loss