Chinese Journal of Schistosomiasis Control ›› 2021, Vol. 33 ›› Issue (5): 445-.

    Next Articles

Establishment of a deep learning⁃visual model for intelligent recognition of Oncomelania hupensis

SHI Liang1, XIONG Chun⁃Rong1, LIU Mao⁃Mao2, WEI Xiu⁃Shen3, WANG Xin⁃Yao1, WANG Tao1, HUANG Yi⁃Xin1, HONG Qing⁃Biao1, LI WEI1, YANG Hai⁃Tao1, ZHANG Jian⁃Feng1*, YANG Kun1,2*   

  1. 1 Key Laboratory of National Health Commission on Parasitic Disease Control and Prevention, Jiangsu Provincial Key Laboratory on Parasite and Vector Control Technology, Jiangsu Institute of Parasitic Diseases, Public Health Research Center of Jiangnan University, Wuxi 214064, China; 2 School of Public Health, Nanjing Medical University, China; 3 School of Computer Science and Engineering, Nanjing University of Science and Technology, Key Laboratory of Intelligent Perception and Systems for High⁃Dimensional Information of Ministry of Education, Jiangsu Provincial Key Laboratory of Image and Video Understanding for Social Safety, China
  • Online:2021-11-02 Published:2021-11-02

基于深度学习技术的湖北钉螺视觉智能识别模型的建立

施亮1,熊春蓉1,刘毛毛2,魏秀参3,王鑫瑶1,王涛1,黄轶昕1,洪青标1, 李伟1,羊海涛1,张键锋1*,杨坤1, 2*   

  1. 1国家卫生健康委员会寄生虫病预防与控制技术重点实验室、江苏省寄生虫与媒介控制技术重点实验室、江苏省血吸虫病防治研究所、江南大学公共卫生研究中心(无锡214064);2 南京医科大学公共卫生学院;3 南京理工大学计算机科学与工程学院、高维信息智能感知与系统教育部重点实验室、江苏省社会安全图像与视频理解重点实验室
  • 作者简介:施亮,男,硕士,主管医师。研究方向:空间流行病学与机器学习
  • 基金资助:
    国家自然科学基金(82173586);江苏省国际科技合作项目(BZ2020003);江苏省省属公益院所能力提升项目(BM2018020-3);江苏省卫生健康委科研项目(X201805);江南大学公共卫生研究中心项目(JUPH201837、JUPH202008)

Abstract: Objective To establish a deep learning⁃based visual model for intelligent recognition of Oncomelania hupensis, the intermediate host of Schistosoma japonicum, and evaluate the effects of different training strategies for O. hupensis image recognition. Methods A total of 2 614 datasets of O. hupensis snails and 4 similar snails were generated through field sampling and internet capture, and were divided into training sets and test sets. An intelligent recognition model was created based on deep learning, and was trained and tested. The precision, sensitivity, specificity, accuracy, F1 score and Youden index were calculated. In addition, the receiver operating characteristic (ROC) curve of the model for snail recognition was plotted to evaluate the effects of “new learning”, “transfer learning” and “transfer learning + data enhancement” training strategies on the accuracy of the model for snail recognition. Results Under the “transfer learning + data enhancement” strategy, the precision, sensitivity, specificity, accuracy, Youden index and F1 score of the model were 90.10%, 91.00%, 97.50%, 96.20%, 88.50% and 90.51% for snail recognition, which were all higher than those under both “new learning” and “transfer learning” strategies. There were significant differences in the sensitivity, specificity and accuracy of the model for snail recognition under “new learning”, “transfer learning” and “transfer learning + data enhancement” training strategies (all P values < 0.001). In addition, the area under the ROC curve of the model was highest (0.94) under the “transfer learning + data enhancement” training strategy. Conclusions  This is the first visual model for intelligent recognition of O. hupensis based on deep learning, which shows a high accuracy for snail image recognition. The “transfer learning + data enhancement” training strategy is helpful to improve the accuracy of the model for snail recognition.

Key words: Oncomelania hupensis, Deep learning, Intelligent recognition, Computer vision, Machine learning, Artificial intelligence

摘要: 目的 建立一种基于深度学习技术的日本血吸虫中间宿主湖北钉螺视觉智能识别模型,评价不同训练策略用于钉螺图像识别的效果。方法 通过现场采集和互联网抓取构建钉螺及4种相似螺类数据集2 614幅,将其分为训练集和测试集。基于深度学习技术建立智能识别模型,并对模型进行训练及测试,计算模型识别钉螺的精确率、敏感性、特异性、准确率、F1值、约登指数;绘制受试者工作特征(receiver operating characteristic, ROC)曲线,分析“全新学习”、“迁移学习”、“迁移学习+数据增强”等3种不同训练策略对模型识别钉螺准确性的影响。结果 “迁移学习+数据增强”训练策略下,模型识别钉螺的精确率、敏感性、特异性、准确率、约登指数和F1值分别为90.10%、91.00%、97.50%、96.20%、88.50%、90.51%,均高于“全新学习”、“迁移学习”策略;“全新学习”、“迁移学习”、“迁移学习+数据增强”训练策略下,模型识别钉螺的敏感性、特异性和准确率差异均有统计学意义(P均 < 0.001)。“迁移学习+数据增强”训练策略下,模型ROC曲线下面积最大(0.94)。结论 首次建立了基于深度学习技术的湖北钉螺视觉智能识别模型,钉螺图像识别准确性较高。“数据增强和迁移学习”训练策略有助于提高模型识别钉螺的准确性。  

关键词: 湖北钉螺, 深度学习, 智能识别, 计算机视觉, 机器学习, 人工智能

CLC Number: