基于遗传算法的肺结核检测模型的训练方法、装置及设备与流程

1.本发明涉及肺结核检测技术领域，具体涉及一种基于遗传算法的肺结核检测模型的训练方法、装置及设备。


背景技术：

2.肺结核是我国的乙类传染病。结核病是由结核杆菌感染引起的慢性传染病。结核菌可能侵入人体全身各种器官，但主要侵犯肺脏，称为肺结核病。结核病又称为痨病和“白色瘟疫”，是一种古老的传染病，自有人类以来就有结核病。肺结核患者主要分布在我国穷困地区，由于缺乏高水平影像科专家对胸片进行准确判断，因此容易导致肺结核在穷困地区的进一步扩展。医疗影像辅助诊断设备能够在大规模检测肺结核筛查任务中发挥巨大的作用，但是医疗影像辅助诊断模型训练的超参数空间巨大，采用随机调参需要足够的算力来覆盖足够多的模型。如果采用手工调参的方式，则需要丰富的经验而且不能保证得到较为不错的结果。


技术实现要素：

3.针对现有技术的不足，本发明旨在提供一种基于遗传算法的肺结核检测模型的训练方法、装置及设备。
4.为了实现上述目的，本发明采用如下技术方案：
5.一种基于遗传算法的肺结核检测模型的训练方法，具体过程如下：
6.s1、采集样本胸片的dcm文件，同时排除非正位胸片，筛选得到正位胸片；将采集到的正位胸片的dcm文件转换为png文件；
7.s2、影像科专家对采集得到的正位胸片进行判断，判断其是否属于肺结核患者，排除健康的正位胸片；
8.s3、影像科专家对筛选出来的属于肺结核患者的正位胸片进行标注，以标注框的方式将正位胸片中的病灶标注出来；
9.s4、将经过标注的正位胸片以及标注框信息输入肺结核检测模型中并行训练多个模型，各个模型开始训练前先在超参数的范围内随机选择固定数值；
10.s5、把训练好的各个模型在验证集上进行验证，计算模型的平均准确率作为该模型的适应值；
11.s6、将训练好的所有模型的超参数进行选择、交叉、变异操作，得到新一代待训练的多个模型；
12.s7、将经过标注的正位胸片以及标注框信息输入新一代待训练的多个模型中并行训练；
13.s8、计算训练得到的模型在验证集上的平均准确率作为该模型的适应值；
14.s9、如果当前的迭代轮数小于预设的迭代轮数上限，返回至步骤s6，否则选择适应值最高的模型作为最终的肺结核检测模型。
15.本发明提供一种实现上述训练方法的训练装置，包括：
16.胸片采集模块：用于采集样本胸片的dcm文件，同时排除非正位胸片，筛选得到正位胸片，以及将采集到的正位胸片的dcm文件转换为png文件；
17.胸片筛选模块：用于供影像科专家对采集得到的正位胸片进行判断，判断其是否属于肺结核患者，排除健康的正位胸片；
18.标注模块：用于供影像科专家对筛选出来的属于肺结核患者的正位胸片进行标注，以标注框的方式将正位胸片中的病灶标注出来；
19.训练模块：用于按照权利要求1所述方法的步骤s4
‑
s9进行模型训练，得到最终的肺结核检测模型。
20.本发明还提供一种具有上述训练方法训练得到的肺结核检测模型的辅助诊断设备。
21.本发明的有益效果在于：本发明能够在较少的试验次数下得到比随机调参更优的结果并且能自动化地对参数进行调整，减少人工调参对经验的依赖。
附图说明
22.图1为本发明实施例1的方法流程图。
具体实施方式
23.以下将结合附图对本发明作进一步的描述，需要说明的是，本实施例以本技术方案为前提，给出了详细的实施方式和具体的操作过程，但本发明的保护范围并不限于本实施例。
24.实施例1
25.本实施例提供一种基于遗传算法的肺结核检测模型的训练方法，如图1所示，具体过程如下：
26.s1、采集样本胸片的dcm文件，同时排除非正位胸片，筛选得到正位胸片；将采集到的正位胸片的dcm文件转换为png文件；
27.s2、影像科专家对采集得到的正位胸片进行判断，判断其是否属于肺结核患者，排除健康的正位胸片；
28.s3、影像科专家对筛选出来的属于肺结核患者的正位胸片进行标注，以标注框的方式将正位胸片中的病灶标注出来；
29.s4、将经过标注的正位胸片以及标注框信息输入肺结核检测模型中并行训练多个模型，各个模型开始训练前先在超参数的范围内随机选择固定数值；
30.s5、把训练好的各个模型在验证集上进行验证，计算模型的平均准确率作为该模型的适应值；
31.s6、将训练好的所有模型的超参数进行选择、交叉、变异操作，得到新一代待训练的多个模型；
32.s7、将经过标注的正位胸片以及标注框信息输入新一代待训练的多个模型中并行训练；
33.s8、计算训练得到的模型在验证集上的平均准确率作为该模型的适应值；
34.s9、如果当前的迭代轮数小于预设的迭代轮数上限，返回至步骤s6，否则选择适应值最高的模型作为最终的肺结核检测模型。
35.利用所述肺结核检测模型时，将待检测的胸片输入肺结核检测模型中，肺结核检测模型对胸片进行检测并输出检测预测结果。
36.实施例2
37.本实施例提供一种实现实施例1所述训练方法的训练装置，包括：
38.胸片采集模块：用于采集样本胸片的dcm文件，同时排除非正位胸片，筛选得到正位胸片，以及将采集到的正位胸片的dcm文件转换为png文件；
39.胸片筛选模块：用于供影像科专家对采集得到的正位胸片进行判断，判断其是否属于肺结核患者，排除健康的正位胸片；
40.标注模块：用于供影像科专家对筛选出来的属于肺结核患者的正位胸片进行标注，以标注框的方式将正位胸片中的病灶标注出来；
41.训练模块：用于按照实施例1方法的步骤s4
‑
s9进行模型训练，得到最终的肺结核检测模型。
42.实施例3
43.为了验证实施例1方法的优势，本实施例收集了4000张有临床验证的胸片并标注了其肺结核的病灶区域。将这4000张胸片分为3000张训练集，以及1000张验证集。同时还收集了10000张健康的胸片用于统计性能指标。在判断胸片是否正确检测为肺结核患者的任务上，在相同的100轮超参数调试的实验中，实施例1方法能够取得0.9461的auc
‑
roc值。而随机调参的方法只能取得0.9353的auc
‑
roc值。
44.其中超参数定义如表1所示。
45.表1
46.超参数参数范围学习率1e
‑
5～1e
‑
1权重衰减0～1e
‑
3图像旋转角度0～45度图像放大系数0～0.9mixup系数0～1.0分类项权重0.2～4.0定位项权重0.2～4.0图像左右翻转概率0～1.0
47.对于本领域的技术人员来说，可以根据以上的技术方案和构思，给出各种相应的改变和变形，而所有的这些改变和变形，都应该包括在本发明权利要求的保护范围之内。