深度学习的煤矿尘肺病图像识别
Image recognition of pneumoconiosis of miners based on deep learning
刘丹丹 1刘玉秋 1李德文 2郭胜均 2汤春瑞2
作者信息
- 1. 黑龙江科技大学电气与控制工程学院,哈尔滨 150022
- 2. 中煤科工集团重庆研究院有限公司,重庆 400037
- 折叠
摘要
针对尘肺病人工诊断率低和诊断误差大等问题,通过深度学习利用神经网络模型对煤矿尘肺病CT图进行辅助诊断,通过比较现有的ResNet原始模型、DenseNet模型、MC-CNN模型训练结果,提出一种基于改进ResNet模型的图像分类方法.将GN正则化嵌入到模型中,将残差网络模型中常规卷积convolution更改为内卷involution.结果表明,ResNet101 模型的整体性更强,其准确率达到93.2%,精确率达93.8%,召回率达到93.6%,F1 达到93.7%.该研究融合深度学习算法与正则化的优势验证图像识别模型是可行的.
Abstract
This paper is aimed at addressing the low manual diagnosis rate and big diagnosis error of pneumoconiosis in mine.The study consists of using neural network model to assist in diagnosis of pneumo-coniosis CT map;proposing an image classification method based on the improved ResNet model by compa-ring the existing ResNet original model,DenseNet model and MC-CNN model training result;embeding GN regularization into the model to change the regular convolution to involution in the residual network model.The experimental results show that the ResNet101 model has a stronger integrity,with the accuracy rate of 93.2%and the accuracy rate of 93.8%,the recall rate of 93.6%,and the F1 of 93.7%.This image rec-ognition model verified by using deep learning algorithms and fusing the advantages of regularization is fea-sible.
关键词
煤矿/深度学习/残差网络/正则化Key words
coal mine/deep learning/residual network/regularization引用本文复制引用
基金项目
国家重点研发计划(2017YFC0805208)
出版年
2024