用YOLOv11检测结肠镜图像中的息肉,提升早期癌症筛查效率。
Polyp detection in colonoscopy images using YOLOv11
- 基于YOLOv11系列模型,采用单阶段检测架构实现快速定位息肉。
- 在Kvasir数据集上,YOLOv11s精度达94.3%,增广后效果进一步提升。
- 适合临床辅助诊断系统开发,尤其关注实时性与高准确率的场景。
结直肠癌(CRC)是全球最常见的癌症之一,起源于结肠内壁的息肉。为预防CRC,需早期发现息肉。结肠镜检查通过摄像头拍摄图像,目前主要依赖专家人工分析。随着机器学习发展,传统模型逐渐被深度学习替代。深度学习在特征提取和泛化能力上表现更优。目标检测模型分为单阶段与两阶段:单阶段模型推理速度快,更适合实时应用;两阶段模型精度更高。YOLO作为代表性单阶段模型,因其低延迟受到关注。近年来各版本迭代使性能持续提升。本文评估最新发布的YOLOv11在息肉检测中的表现,测试其五个变体(YOLOv11n、s、m、l、x),使用Kvasir数据集进行训练与测试。分别采用原始数据集和经增强技术处理的数据集,对比不同配置下的检测性能。
原文摘要 · Abstract (English)
Colorectal cancer (CRC) is one of the most commonly diagnosed cancers all over the world. It starts as a polyp in the inner lining of the colon. To prevent CRC, early polyp detection is required. Colonosopy is used for the inspection of the colon. Generally, the images taken by the camera placed at the tip of the endoscope are analyzed by the experts manually. Various traditional machine learning models have been used with the rise of machine learning. Recently, deep learning models have shown more effectiveness in polyp detection due to their superiority in generalizing and learning small features. These deep learning models for object detection can be segregated into two different types: single-stage and two-stage. Generally, two stage models have higher accuracy than single stage ones but the single stage models have low inference time. Hence, single stage models are easy to use for quick object detection. YOLO is one of the singlestage models used successfully for polyp detection. It has drawn the attention of researchers because of its lower inference time. The researchers have used Different versions of YOLO so far, and with each newer version, the accuracy of the model is increasing. This paper aims to see the effectiveness of the recently released YOLOv11 to detect polyp. We analyzed the performance for all five models of YOLOv11 (YOLO11n, YOLO11s, YOLO11m, YOLO11l, YOLO11x) with Kvasir dataset for the training and testing. Two different versions of the dataset were used. The first consisted of the original dataset, and the other was created using augmentation techniques. The performance of all the models with these two versions of the dataset have been analysed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。