基于机器学习的肝癌分类方法和装置与流程

本发明涉及人工智能技术，尤其涉及一种基于机器学习的肝癌分类处理方法和装置。

背景技术：

1、根据最新统计数据表明，全球每5人中就会有1人在一生之中罹患癌症，尤其是肝癌，由于其高发病率和高死亡率而成为全世界癌症死亡的主要原因之一，早期肝癌的检测与治疗是有效降低肝癌死亡率的重要途径。

2、目前，对于肝癌的诊断方式主要是以肿瘤的医学影像分析的方式来确定，具体的，目前的，肿瘤医学影像是癌症诊断的基础，可以采用x射线、电子计算机断层扫描(computedtomography，ct)、磁共振、超声波等手段获取相应的医学影像，可以通过机器学习的方式训练对不同类型和阶段的肝癌进行识别的模型，然后通过模型识别的方式对拍摄得到的肝脏的医学影像进行识别，确定具体的癌症的类型和阶段。

3、然而，由于目前的肝脏上产生的各种肿瘤病变的形状，大小，边界区别都非常大，并且肝脏上肿瘤的大小多样，结构位置复杂，导致目前采用的简单的机器学习的方式进行识别，可能导致误诊或者分类不准确，从而导致治疗效果不佳。

技术实现思路

1、本发明实施例提供一种基于机器学习的肝癌分类处理方法和装置，用于解决目前采用的简单的机器学习的方式进行识别，可能导致误诊或者分类不准确，从而导致治疗效果不佳的问题。

2、本发明实施例的第一方面，提供一种基于机器学习的肝癌分类处理方法，其特征在于，包括：

3、获取待进行肝癌分类检测的医学影像和病理图像，所述医学影像为针对肝脏部分扫描得到的增强ct图像或者核磁共振图像；

4、将所述医学影像输入第一分类模型得到第一分类结果，并将所述病理图像输入第二分类模型得到第二分类结果；其中，所述第一分类模型是基于加权fisher构造的神经网络模型训练得到的对肝癌进行识别分类的模型，所述第二分类模型是基于在vgg模型中增加bn算法之后进行训练得到的对肝癌病理图像进行分类的模型；

5、根据所述第一分类结果和所述第二分结果确定最终的肝癌分类结果。

6、在一种具体实现方式中，所述根据所述第一分类结果和所述第二分结果确定最终的肝癌分类结果，包括；

7、若所述第一分类结果和所述第二分类结果相同，则将其中任一结果确定为最终的肝癌分类结果；

8、若所述第一分类结果和所述第二分类结果不同，则将所述病理图像通过随机裁剪和随机颜色失真的数据方法进行扩充，得到增强处理后的病理图像；

9、将所述增强处理后的病理图像输入所述第二分类模型中进行处理，得到最终的肝癌分类结果。

10、在一种具体实现方式中，所述方法还包括：

11、将获取到的医学影像集合中的影像分为训练集和验证集，所述医学影像集合中包括多个已经标定的正常肝脏的医学影像以及每个分类的肝癌对应的多个医学影像；

12、针对所述训练集中的医学影像，将医学影像输入预先构造的神经网络模型的卷积层中进行卷积处理，得到第一卷积特征集合；

13、将所述第一卷积特征集合进行均值采样的方法进行处理，得到第二卷积特征集合；

14、将所述第二卷积特征集合输入所述神经网络模型的全连接层进行处理，最终输出所述神经网络模型的预测结果；

15、运用加权fisher准则对应的函数作为损失函数，计算医学影像的标定结果和所述预测结果之间的损失值，并基于所述损失值经过反向传播传递残差修正模型参数，重复前述步骤直至所述损失值收敛，采用所述验证集对训练的模型进行验证，得到所述第一分类模型。

16、在一种具体实现方式中，初始构造的神经网络模型的卷积层的公式为：

17、

18、其中，me表示输入的图像集合，表示神经网络模型卷积中的第1层中的第e个特征图像，kl表示第1层的卷积核，*表示卷积运算，表示与第l-1层第f个特征图像进行相连的滤波器，表示偏置，f(·)表示神经元非线性激活函数。

19、在一种具体实现方式中，所述加权fisher准则对应的损失函数为：

20、y＝y(w，b)-αyb+βyw；其中，α和β为常数系数，取值范围在0-1之间，yb表示类间距离的度量函数，yw表示类内距离度量的函数；

21、

22、

23、其中，δij表示第i类样本和第j类样本之间的马氏距离，ω(δij)为权值函数，n表示第i类样本的总数，m表示第j类样本的总数，w，b分别表示所述神经网络模型中的卷积核的权值和偏置，x表示具体的样本图像。

24、在一种具体实现方式中，所述方法还包括：

25、将获取到的肝脏病理图像集合中的图像分为训练集和验证集，所述肝脏病理图像集合中包括已经标注了正常肝脏，肝细胞癌，胆管细胞癌和混合癌的病理在多种放大倍数下的多个病理图像，所述多种放大倍数包括40倍，100倍，200倍和400倍；

26、在vgg模型的卷积层的relu函数之前加入bn算法，构建出新的vgg模型，所述新的vgg模型包含16个卷积层，3个全连接层和1个输出层；所述bn算法用于对relu函数的输入数据进行标准化处理；

27、基于所述训练集和所述验证集对所述新的vgg模型进行训练和验证，得到第二分类模型。

28、在一种具体实现方式中，所述在vgg模型的卷积层的relu函数之前加入bn算法，包括：

29、在vgg模型的卷积层的relu函数：r＝max(0，y)之前加入所述bn算法对输入的数据进行标准化处理；

30、其中，y为卷积处理后的输出值；假设xi表示所述vgg模型的第1层的输入的第i个数据，x～i表示标准化处理后的数据，m为单个训练样本包含的输入数据个数，ε取值10-8，参数y和β是通过梯度下降法求得的常数，μ表示输入数据的均值，σ表示方差取值为1，表示数据的中间值，所述bn算法包括：

31、

32、

33、

34、

35、在一种具体实现方式中，所述基于所述训练集和所述验证集对所述新的vgg模型进行训练和验证，得到第二分类模型，包括：

36、根据所述训练集中标定的多个病理图像，利用can的对抗训练方式对所述新的vgg模型进行训练，并根据所述验证集中的病理图像对训练后得到的模型进行验证，得到所述第二分类模型。

37、本发明实施例的第二方面一种基于机器学习的肝癌分类处理装置，包括：

38、图像获取模块，用于获取待进行肝癌分类检测的医学影像和病理图像，所述医学影像为针对肝脏部分扫描得到的增强ct图像或者核磁共振图像；

39、算法处理模块，用于将所述医学影像输入第一分类模型得到第一分类结果，并将所述病理图像输入第二分类模型得到第二分类结果；其中，所述第一分类模型是基于加权fisher构造的神经网络模型训练得到的对肝癌进行识别分类的模型，所述第二分类模型是基于在vgg模型中增加bn算法之后进行训练得到的对肝癌病理图像进行分类的模型；

40、分类处理模块，用于根据所述第一分类结果和所述第二分结果确定最终的肝癌分类结果。

41、本发明实施例的第三方面提供一种电子设备，包括：

42、处理器，存储器以及与所述其他设备进行数据传输的传输接口；

43、所述存储器存储计算机执行指令；

44、所述处理器执行所述存储器存储的计算机执行指令，以实现第一方面任一项所述的方法。

45、本发明实施例的第四方面，提供一种计算机存储介质，其中存储有计算机程序指令，所述计算机程序执行用于实现第一方面任一项所述的方法。

46、本发明提供的一种基于机器学习的肝癌分类处理方法和装置，通过机器学习的方式分别对肝脏的医学影像和病理图像分别进行识别分类，然后结合两个分类结果，最终确定出对肝癌的诊断分类，在进行机器学习的过程中，在对核磁或者ct等医学影像的分类模型训练时，采用fisher准则，确保样本实际值与样本标签误差尽可能小，从而提高最终基于医学影像进行分类的精确度提高。并且在对病理图像的分类模型训练过程中，增加bn算法对输入数据做批量标准化处理，优化输入数据的分布结构，来提高模型对病理图像的分类精准率.从而优化最终肝癌分类结果，提高分类准确度。