适应多机型的乳腺X射线影像识别方法和装置制造方法及图纸

本发明专利技术提供一种适应多机型的乳腺X射线影像识别方法和装置，所述方法包括：构建网络框架，其中，网络框架包括乳腺影像特征提取网络、乳腺影像机型分类网络和乳腺影像识别网络；获取来自不同机型的多个乳腺X射线影像数据集并统计得到其数据分布，根据数据集的数据分布分别为每张待识别乳腺X射线影像确定窗宽和窗位；将多个机型的、确定了窗宽和窗位的乳腺X射线影像输入网络框架中，对乳腺影像特征提取网络和乳腺影像机型分类网络进行一致性训练，同时训练乳腺影像识别网络；通过训练后的乳腺影像特征提取网络提取确定了窗宽和窗位的待识别乳腺X射线影像的特征，该特征经过训练后的乳腺影像识别网络，以实现对待识别乳腺X射线影像的识别。

The method and device of mammography recognition for multiple models

【技术实现步骤摘要】
适应多机型的乳腺X射线影像识别方法和装置
本专利技术涉及图像识别
，具体涉及一种适应多机型的乳腺X射线影像识别方法和一种适应多机型的乳腺X射线影像识别装置。
技术介绍
乳腺X射线影像能比较全面而正确地反应出整个乳房的大体解剖结构，对于人体组织分析等研究工作具有较高参考价值。在实际应用中，乳腺X射线影像来源于不同的机型，如Philips、Siemens、GE等，这些机型拍摄的影像在风格上差异很大，人类视觉很少会被这种差异影响，而计算机则相反，容易被这种差异影响，作出错误的判断。对于目前普遍应用的一些计算机视觉算法，当训练和测试数据来自同一机型时，分布较为近似，计算机对乳腺影像的识别效果好，但针对每一种机型都收集大量的数据代价非常大，而且非常困难。
技术实现思路
本专利技术为解决上述技术问题，提供了一种适应多机型的乳腺X射线影像识别方法和装置，能够适用于对多种机型的乳腺X射线影像的识别，简单方便且准确性较高。本专利技术采用的技术方案如下：一种适应多机型的乳腺X射线影像识别方法，包括以下步骤：构建网络框架，其中，网络框架包括乳腺影像特征提取网络、乳腺影像机型分类网络和乳腺影像识别网络；获取来自不同机型的多个乳腺X射线影像数据集并统计得到其数据分布，根据数据集的数据分布分别为每张待识别乳腺X射线影像确定窗宽和窗位；将多个机型的、确定了窗宽和窗位的乳腺X射线影像输入网络框架中，对乳腺影像特征提取网络和乳腺影像机型分类网络进行一致性训练，同时训练乳腺影像识别网络；通过训练后的乳腺影像特征提取网络提取确定了窗宽和窗位的待识别乳腺X射线影像的特征，该特征经过训练后的乳腺影像识别网络，以实现对待识别乳腺X射线影像的识别。所述乳腺影像特征提取网络包括若干个卷积模块和若干个下采样模块。所述乳腺影像机型分类网络包括若干个卷积模块，或者包括若干个卷积模块和若干个完全连接模块。所述数据分布包括影像像素值的均值和方差。根据所述数据集的数据分布分别确定每张待识别乳腺X射线影像的窗宽和窗位，具体包括：自适应地选取所述待识别乳腺X射线影像的窗宽和窗位，使得所述待识别乳腺X射线影像的数据分布尽可能地接近所述数据集的数据分布。利用多个机型的乳腺X射线影像对所述乳腺影像特征提取网络和所述乳腺影像机型分类网络进行一致性训练，同时训练乳腺影像识别网络，具体包括：一致性训练，通过所述乳腺影像特征提取网络和所述乳腺影像机型分类网络的共同训练，约束同一腺体类型不同机型的乳腺影像特征提取网络输出的特征图分布差异最小化、特征空间上的距离最小化，或分布差异和特征空间上的距离加权和最小化；通过对输入的多个机型的乳腺X射线影像采样或对所述乳腺影像机型分类网络的损失采样，限制所述乳腺影像机型分类网络每次梯度回传更新权重对应的多个乳腺X射线影像的腺体类型相同；计算每个乳腺X射线影像与其他腺体类型相同的乳腺X射线影像在特征层面的平均距离，并根据所述平均距离得到置信分数，作为所述乳腺影像机型分类网络的置信度；网络的总损失，由乳腺影像识别网络的损失以及上述置信度加权后的乳腺影像机型分类网络的损失两部分组成。其中，乳腺X射线影像之间的距离为cosine距离或L2距离。一种适应多机型的乳腺X射线影像识别装置，包括：网络框架，所述网络框架包括乳腺影像特征提取网络、乳腺影像机型分类网络和乳腺影像识别网络；预处理模块，所述预处理模块对来自不同机型的多个乳腺X射线影像数据集进行统计得到其数据分布，根据数据集的数据分布分别为每张待识别乳腺X射线影像确定窗宽和窗位；训练模块，所述训练模块将多个机型的、确定了窗宽和窗位的乳腺X射线影像输入网络框架中，对乳腺影像特征提取网络和乳腺影像机型分类网络进行一致性训练，同时训练乳腺影像识别网络；测试模块，所述测试模块通过训练后的乳腺影像特征提取网络提取确定了窗宽和窗位的待识别乳腺X射线影像的特征，该特征经过训练后的乳腺影像识别网络，以实现对待识别乳腺X射线影像的识别。本专利技术的有益效果：本专利技术的适应多机型的乳腺X射线影像识别方法和装置，在图像层面上，根据数据集的数据分布确定待识别乳腺X射线影像的窗宽和窗位，使图像层面上不同机型的乳腺X线影像风格尽可能相似，在特征层面上，通过乳腺影像特征提取网络与乳腺影像机型分类网络的一致性训练，使特征层面上不同机型的乳腺X射线影像风格尽可能相似，同时减小腺体类型和影像语义信息等对乳腺影像机型分类网络训练的影响，从而能够适用于对多种机型的乳腺X射线影像的识别，简单方便且准确性较高。附图说明图1为本专利技术实施例的适应多机型的乳腺X射线影像识别方法的流程图；图2为本专利技术实施例的适应多机型的乳腺X射线影像识别装置的方框示意图。具体实施方式下面将结合本专利技术实施例中的附图，对本专利技术实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本专利技术保护的范围。如图1所示，本专利技术实施例的适应多机型的乳腺X射线影像识别方法包括以下步骤：S1，构建网络框架，其中，网络框架包括乳腺影像特征提取网络、乳腺影像机型分类网络和乳腺影像识别网络。在本专利技术的一个实施例中，乳腺影像特征提取网络可包括若干个卷积模块和若干个下采样模块，能够对输入的乳腺X射线影像进行特征提取。乳腺影像机型分类网络可包括若干个卷积模块，或者包括若干个卷积模块和若干个完全连接模块，能够根据乳腺影像特征提取网络输出的特征估计出乳腺X射线影像的来源机型。乳腺影像识别网络能够根据乳腺影像特征提取网络输出的特征进行影像识别，得到乳腺病灶检出、乳腺病灶良恶性分类等识别结果。S2，获取来自不同机型的多个乳腺X射线影像数据集并统计得到其数据分布，根据数据集的数据分布分别为每张待识别乳腺X射线影像确定窗宽和窗位。其中，数据分布可包括但不限于像素值的均值、方差。首先可统计得到不同机型的多个乳腺X射线影像数据集的上述数据分布，然后根据数据集的上述数据分布，自适应地选取待识别乳腺X射线影像的窗宽和窗位，使得待识别乳腺X射线影像的数据分布尽可能地接近数据集的数据分布。此处的尽可能接近，可以是各项数据分布差值的均值或加权均值最小。S3，将多个机型的、确定了窗宽和窗位的乳腺X射线影像输入网络框架中，对乳腺影像特征提取网络和乳腺影像机型分类网络进行一致性训练，同时训练乳腺影像识别网络。在本专利技术的一个实施例中，可通过乳腺X射线影像的识别任务相关标签和机型标签来训练网络。机型标签表明乳腺X射线影像来自哪个机型，假设某个输入的乳腺X射线影像来自某个机型，则对该乳腺X射线影像，该机型的标签为1，其他机型的标签均为0。如果数据集中部分乳腺X射线影像的识别任务相关标签缺失，则该部分乳腺X射线影像只有机型标签。腺体类型和影像的语义信息对乳腺影像机型分类网络的学习有较大影响，其中，腺体类型包本文档来自技高网...

【技术保护点】
1.一种适应多机型的乳腺X射线影像识别方法，其特征在于，包括以下步骤：/n构建网络框架，其中，所述网络框架包括乳腺影像特征提取网络、乳腺影像机型分类网络和乳腺影像识别网络；/n获取来自不同机型的多个乳腺X射线影像数据集并统计得到其数据分布，根据数据集的数据分布分别为每张待识别乳腺X射线影像确定窗宽和窗位；/n将多个机型的、确定了窗宽和窗位的乳腺X射线影像输入网络框架中，对乳腺影像特征提取网络和乳腺影像机型分类网络进行一致性训练，同时训练乳腺影像识别网络；/n通过训练后的乳腺影像特征提取网络提取确定了窗宽和窗位的待识别乳腺X射线影像的特征，该特征经过训练后的乳腺影像识别网络，以实现对待识别乳腺X射线影像的识别。/n

【技术特征摘要】
1.一种适应多机型的乳腺X射线影像识别方法，其特征在于，包括以下步骤：
构建网络框架，其中，所述网络框架包括乳腺影像特征提取网络、乳腺影像机型分类网络和乳腺影像识别网络；
获取来自不同机型的多个乳腺X射线影像数据集并统计得到其数据分布，根据数据集的数据分布分别为每张待识别乳腺X射线影像确定窗宽和窗位；
将多个机型的、确定了窗宽和窗位的乳腺X射线影像输入网络框架中，对乳腺影像特征提取网络和乳腺影像机型分类网络进行一致性训练，同时训练乳腺影像识别网络；
通过训练后的乳腺影像特征提取网络提取确定了窗宽和窗位的待识别乳腺X射线影像的特征，该特征经过训练后的乳腺影像识别网络，以实现对待识别乳腺X射线影像的识别。


2.根据权利要求1所述的适应多机型的乳腺X射线影像识别方法，其特征在于，所述乳腺影像特征提取网络包括若干个卷积模块和若干个下采样模块。


3.根据权利要求2所述的适应多机型的乳腺X射线影像识别方法，其特征在于，所述乳腺影像机型分类网络包括若干个卷积模块，或者包括若干个卷积模块和若干个完全连接模块。


4.根据权利要求3所述的适应多机型的乳腺X射线影像识别方法，其特征在于，所述数据分布包括影像像素值的均值和方差。


5.根据权利要求4所述的适应多机型的乳腺X射线影像识别方法，其特征在于，根据所述数据集的数据分布分别确定每张待识别乳腺X射线影像的窗宽和窗位，具体包括：
自适应地选取所述待识别乳腺X射线影像的窗宽和窗位，使得所述待识别乳腺X射线影像的数据分布尽可能地接近所述数据集的数据分布。


6.根据权利要求5所述的适应多机型的乳腺X射线影像识别方法，其特征在于，利用多个机型的乳腺X射线影像对所述乳腺影像特征提取网络和所述乳腺影像机型分...

【专利技术属性】
技术研发人员：王思雯，刚亚栋，张番栋，张笑春，俞益洲，
申请(专利权)人：北京深睿博联科技有限责任公司，杭州深睿博联科技有限公司，
类型：发明
国别省市：北京;11