改进YOLOv8模型实现多类型肺结节检测

doi:10.3778/j.issn.1673-9418.2406018

计算机科学与探索 ›› 2025, Vol. 19 ›› Issue (2): 429-442.DOI: 10.3778/j.issn.1673-9418.2406018

改进YOLOv8模型实现多类型肺结节检测

包强强，唐思源，李擎乾，王乃钰，杨敏，谷宇，赵金亮，高婧博，王嘉欣，曲禹涵

1. 内蒙古科技大学数智产业学院，内蒙古包头 014010
2. 内蒙古科技大学包头医学院计算机科学与技术学院，内蒙古包头 014040
3. 内蒙古科技大学包头医学院图书馆，内蒙古包头 014040
4. 内蒙古科技大学数智产业学院内蒙古自治区模式识别与智能图像处理重点实验室，内蒙古包头 014010

出版日期:2025-02-01 发布日期:2025-01-23

Improved YOLOv8 Model for Multi-type Lung Nodule Detection

BAO Qiangqiang, TANG Siyuan, LI Qingqian, WANG Naiyu, YANG Min, GU Yu, ZHAO Jinliang, GAO Jingbo, WANG Jiaxin, QU Yuhan

1. School of Digtial and Intelligence Industry, Inner Mongolia University of Science and Technology, Baotou, Inner Mongolia 014010, China
2. School of Computer Science and Technology, Inner Mongolia University of Science and Technology Baotou Medical College, Baotou, Inner Mongolia 014040, China
3. Library, Inner Mongolia University of Science and Technology Baotou Medical College, Baotou, Inner Mongolia 014040, China
4. Inner Mongolia Key Laboratory of Pattern Recognition and Intelligent Image Processing, School of Digtial and Intelligence Industry, Inner Mongolia University of Science and Technology, Baotou, Inner Mongolia 014010, China

Online:2025-02-01 Published:2025-01-23

摘要/Abstract

摘要： 目前，肺结节检测通常是对实性肺结节的单一型检测，不同类型肺结节对应多种肺癌类型，多类型检测有助于提高肺癌的整体检出率，提升治愈率。为实现对实性、混合型、磨玻璃型多类型肺结节检测，对YOLOv8模型进行针对性改进。提出RepViTCAA模块对主干部分的C2f模块进行改进，提升微小肺结节检测精度并对模型进行轻量化设计。提出ECLA-HSFPN模块，重建模型特征融合部分，提升尺度不固定肺结节检测精度。将KAN网络融入模型当中，基于KAN网络非线性特征学习能力强的特性，进一步提升微小肺结节检测精度，增强模型泛化能力。基于Inner-IoU辅助框思想，对CIoU损失函数进行改进，进一步解决肺结节尺度不固定问题，提升模型检测精度。在LUNA16数据集中进行测试，改进模型相比原模型及YOLOv9、RT-DETR等主流模型各项评价指标均有提升。在4种类型（实性、磨玻璃型、混合型、微小型）肺结节的专项数据集中测试，改进模型检测效果优于原模型。在LUNA16与本地医院的混合数据集中进行泛化性测试，改进模型具有较强的泛化能力。针对多类型肺结节检测任务，模型的改进较为有效，可以准确检测不同类型的肺结节。

关键词: 多类型肺结节检测, YOLOv8, RepViTCAA, ECLA-HSFPN, KAN, Inner-IoU

Abstract: Currently, lung nodule testing is usually a single type of testing for solid lung nodules, and different kinds of lung nodules correspond to multiple types of lung cancer. Multi-type testing can help improve the overall detection rate of lung cancer and enhance the cure rate. Targeted improvements to the YOLOv8 model are made to enable the detection of multiple types of lung nodules, including solid, mixed, and ground glass. Firstly, the RepViTCAA module is proposed to improve the C2f module of the main part to enhance the accuracy of tiny lung nodule detection and the lightweight design of the model. Secondly, the ECLA-HSFPN module is proposed to reconstruct the feature fusion part of the model to improve the scale-invariant lung nodule detection accuracy. Then, the KAN network is integrated into the model further to improve the detection accuracy of tiny lung nodules and enhance the generalization ability of the model based on the strong learning ability of nonlinear features of the KAN network. Finally, based on the Inner-IoU auxiliary frame idea, the CIoU loss function is improved to solve the problem of unfixed lung nodule scale and enhance the model detection accuracy. Tested on the LUNA16 dataset, the improved model has improved all evaluation indices compared with the original model and mainstream models such as YOLOv9 and RT-DETR. The improved model has better detection performance than the original model when tested on a specialized dataset of four types (solid, ground glass, mixed, and microminiature) of lung nodules. The generalizability is tested on a mixed dataset of LUNA16 and local hospitals, and the improved model has strong generalization ability. The improvement of the model is more effective for the task of detecting multiple types of lung nodules and it can accurately detect different types of lung nodules.

Key words: multi-type lung nodule detection, YOLOv8, RepViTCAA, ECLA-HSFPN, KAN, Inner-IoU

包强强, 唐思源, 李擎乾, 王乃钰, 杨敏, 谷宇, 赵金亮, 高婧博, 王嘉欣, 曲禹涵. 改进YOLOv8模型实现多类型肺结节检测[J]. 计算机科学与探索, 2025, 19(2): 429-442.

BAO Qiangqiang, TANG Siyuan, LI Qingqian, WANG Naiyu, YANG Min, GU Yu, ZHAO Jinliang, GAO Jingbo, WANG Jiaxin, QU Yuhan. Improved YOLOv8 Model for Multi-type Lung Nodule Detection[J]. Journal of Frontiers of Computer Science and Technology, 2025, 19(2): 429-442.

参考文献

[1] THAKUR S K, SINGH D P, CHOUDHARY J. Lung cancer identification: a review on detection and classification[J]. Cancer and Metastasis Reviews, 2020, 39(3): 989-998.
[2] 王新宇, 赵静文, 刘翔, 等. 融合坐标注意力机制的YOLOv3肺结节检测算法[J]. 电子科技, 2024(6): 1-7.
WANG X Y, ZHAO J W, LIU X, et al. YOLOv3 lung nodule detection based on coordinate attention[J]. Electronic Science and Technology, 2024(6): 1-7.
[3] 赵奎, 仇慧琪, 李旭, 等. 结合注意力和多路径融合的实时肺结节检测算法[J]. 计算机应用, 2024, 44(3): 945-952.
ZHAO K, QIU H Q, LI X, et al. Real-time pulmonary nodule detection algorithm combining attention and multipath fusion[J]. Journal of Computer Applications, 2024, 44(3): 945-952.
[4] 袁金丽, 赵琳琳, 郭志涛, 等. 改进U型残差网络用于肺结节检测[J]. 计算机工程与应用, 2022, 58(13): 195-203.
YUAN J L, ZHAO L L, GUO Z T, et al. Improved U-shaped residual network for lung nodule detection[J]. Computer Engineering and Applications, 2022, 58(13): 195-203.
[5] JI Z L, WU Y, ZENG X Y, et al. Lung nodule detection in medical images based on improved YOLOv5s[J]. IEEE Access, 2023, 11: 76371-76387.
[6] ZHU L, ZHU H, YANG S, et al. Pulmonary nodule detection based on hierarchical-split HRNet and feature pyramid network with atrous convolution[J]. Biomedical Signal Processing and Control, 2023, 85: 105024.
[7] XIONG Y, DENG L, WANG Y. Pulmonary nodule detection based on model fusion and adaptive false positive reduction[J]. Expert Systems with Applications, 2024, 238: 121890.
[8] HUSSAIN M. YOLO-v1 to YOLO-v8, the rise of YOLO and its complementary nature toward digital manufacturing and industrial defect detection[J]. Machines, 2023, 11(7): 677.
[9] WANG A, CHEN H, LIN Z, et al. RepViT: revisiting mobile CNN from ViT perspective[EB/OL]. [2024-04-24]. https://arxiv.org/abs/2307.09283.
[10] CAI X, LAI Q, WANG Y, et al. Poly kernel inception network for remote sensing detection[EB/OL]. [2024-04-24]. https://arxiv.org/abs/2403.06258.
[11] CHEN Y F, ZHANG C Y, CHEN B, et al. Accurate leukocyte detection based on deformable-DETR and multi-level feature fusion for aiding diagnosis of blood diseases[J]. Computers in Biology and Medicine, 2024, 170: 107917.
[12] XU W, WAN Y. ELA: efficient local attention for deep convolutional neural networks[EB/OL]. [2024-04-24]. https://arxiv.org/abs/2403.01123.
[13] LIU W, LU H, FU H, et al. Learning to upsample by learning to sample[C]//Proceedings of the 2023 IEEE/CVF International Conference on Computer Vision. Piscataway: IEEE, 2023: 6027-6037.
[14] LIU Z, WANG Y, VAIDYA S, et al. KAN: Kolmogorov-Arnold networks[EB/OL]. [2024-05-16]. https://arxiv.org/abs/2404.19756.
[15] ZHENG Z H, WANG P, LIU W, et al. Distance-IoU loss: faster and better learning for bounding box regression[J]. Proceedings of the AAAI Conference on Artificial Intelligence, 2020, 34(7): 12993-13000.
[16] ZHANG H, XU C, ZHANG S. Inner-IOU: more effective intersection over union loss with auxiliary bounding box[EB/OL]. [2024-04-24]. https://arxiv.org/abs/2311.02877.
[17] SETIO A A A, TRAVERSO A, DE BEL T, et al. Validation, comparison, and combination of algorithms for automatic detection of pulmonary nodules in computed tomography images: the LUNA16 challenge[J]. Medical Image Analysis, 2017, 42: 1-13.
[18] REN S Q, HE K M, GIRSHICK R, et al. Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39(6): 1137-1149.
[19] LIU W, ANGUELOV D, ERHAN D, et al. SSD: single shot multibox detector[C]//Proceedings of the 14th European Conference on Computer Vision. Cham: Springer, 2016: 21-37.
[20] TERVEN J, CORDOVA-ESPARZA D. A comprehensive review of YOLO architectures in computer vision: from YOLOv1 to YOLOv8 and YOLO-NAS[EB/OL]. [2024-04-24]. https://arxiv.org/abs/2304.00501.
[21] LI C, LI L, JIANG H, et al. YOLOv6: a single-stage object detection framework for industrial applications[EB/OL]. [2024-04-24]. https://arxiv.org/abs/2209.02976.
[22] WANG C Y, BOCHKOVSKIY A, LIAO H Y M. YOLOv7: trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[C]//Proceedings of the 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2023: 7464-7475.
[23] WANG C C, HE W, NIE Y, et al. Gold-YOLO: efficient object detector via gather-and-distribute mechanism[EB/OL]. [2024-04-24]. https://arxiv.org/abs/2309.11331.
[24] CHEN Y M, YUAN X B, WU R Q, et al. YOLO-MS: rethinking multi-scale representation learning for real-time object detection[EB/OL]. [2024-04-24]. https://arxiv.org/abs/2308. 05480.
[25] TANG Y, HAN K, GUO J, et al. GhostNetv2: enhance cheap operation with long-range attention[C]//Advances in Neural Information Processing Systems 35, New Orleans, Nov 28-Dec 9, 2022: 9969-9982.
[26] ZHAO Y A, LV W Y, XU S L, et al. DETRs beat YOLOs on real-time object detection[C]//Proceedings of the 2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2024: 16965-16974.
[27] WANG C Y, YEH I H, LIAO H Y M. YOLOv9: learning what you want to learn using programable gradient information[EB/OL]. [2024-04-24]. https://arxiv.org/abs/2402.13616.

编辑推荐 0

Metrics

阅读次数

全文

HTML			PDF

最新录用	在线预览	正式出版	最新录用	在线预览	正式出版
0	0	0	35	0	59

来源	本网站	其他网站

次数	93	1
比例	99%	1%

摘要

138

最新录用	在线预览	正式出版

51	0	87

	来源	本网站

	次数	138
	比例	100%

改进YOLOv8模型实现多类型肺结节检测

Improved YOLOv8 Model for Multi-type Lung Nodule Detection

PDF

可视化

摘要/Abstract

引用本文

使用本文

参考文献

相关文章 6

编辑推荐 0

Metrics

[1]	傅裕, 高树辉. 改进YOLOv8s-Pose多人姿态估计轻量化模型研究[J]. 计算机科学与探索, 2025, 19(3): 682-692.
[2]	朱玉敏, 孙光灵, 缪飞. 基于改进YOLOv8算法的鱼眼图像下行人检测[J]. 计算机科学与探索, 2025, 19(2): 443-453.
[3]	颜豪男, 吕伏, 冯永安. 特征级自适应增强的无人机目标检测算法[J]. 计算机科学与探索, 2024, 18(6): 1566-1578.
[4]	吕伏, 傅宇恒, 贺丽娜, 杨冬鹏. 三维多层次特征协同的无人机遥感目标检测算法[J]. 计算机科学与探索, 2024, 18(5): 1301-1317.
[5]	王春梅, 刘欢. YOLOv8-VSC：一种轻量级的带钢表面缺陷检测算法[J]. 计算机科学与探索, 2024, 18(1): 151-160.
[6]	孙悦，郭斌，欧阳逸，於志文，王柱. 移动APP演化策略研究[J]. 计算机科学与探索, 2020, 14(1): 40-50.