[发明专利]基于显著图约束和X光头颅定位侧位图像的年龄估计方法有效
申请号: | 202110410483.2 | 申请日: | 2021-04-13 |
公开(公告)号: | CN112950631B | 公开(公告)日: | 2023-06-30 |
发明(设计)人: | 张智勇;刘宁涛;缑水平;姚瑶;阎春霞;卢云飞;续溢男;李继超 | 申请(专利权)人: | 西安交通大学口腔医院;西安电子科技大学 |
主分类号: | G06T7/00 | 分类号: | G06T7/00;G06N3/08;G06N3/0464 |
代理公司: | 陕西电子工业专利中心 61205 | 代理人: | 陈宏社;王品华 |
地址: | 710004*** | 国省代码: | 陕西;61 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 基于 显著 约束 头颅 定位 图像 年龄 估计 方法 | ||
1.一种基于显著图约束和X光头颅定位侧位图像的年龄估计方法,其特征在于,包括如下步骤:
(1)获取训练样本集Dtrain、验证样本集Dval和测试样本集Dtest:
(1a)获取N个年龄在4-40岁的X光头颅定位侧位图像D={Dn|1n≤N}及其对应的年龄标签集合A={An|1n≤N},并对每幅X光头颅定位侧位图像Dn进行直方图均衡化,得到数据增强后X光头颅定位侧位图像集合其中,N≥10000,Dn表示第n个形状为长方形且包含有增龄性变化区域为头颅、牙齿和脊柱的X光头颅定位侧位图像,An表示Dn对应的年龄标签,表示Dn的数据增强结果;
(1b)对每幅数据增强后X光头颅定位侧位图像的短边进行补零填充,并对补零填充后的形状为正方形的X光头颅定位侧位图像的尺寸缩放至H×H,得到尺寸变换后X光头颅定位侧位图像集合其中,800≤H≤1200,表示尺寸变换后的结果;
(1c)将中的每幅尺寸变换后X光头颅定位侧位图像进行归一化处理,得到归一化后X光头颅定位侧位图像集合并按照7:1.5:1.5的比例将D3中的图像划分为训练样本集Dtrain、验证样本集Dval和测试样本集Dtest,表示的归一化结果;
(2)对自适应缩放卷积神经网络EfficientNet-B0进行迭代训练:
(2a)初始化迭代次数为e1,最大迭代次数为E1,E1≥150,学习率为η1=0.0001,第e1次迭代的学习率为提升停滞次数为C1,最优平均绝对误差为MAEbest_1,以EfficientNet-B0在自然图像数据集ImageNet上的预训练的权重和偏置参数作为EfficientNet-B0的初始权重和偏置参数,并令e1=1,C1=0,MAEbest_1=9223372036854775807,
(2b)将训练样本集Dtrain划分为S1个训练样本子集每个训练样本子集包括b1个训练样本,并将作为EfficientNet-B0的输入提取每个训练样本的增龄性变化区域的特征,通过该特征对训练样本对应的年龄进行回归预测,得到b1个年龄预测值,然后采用平均误差函数计算每个年龄预测值与其对应的年龄标签标注的年龄的平均绝对误差值L1,其中S1≥700,表示第s1个训练样本子集,5≤b1≤10;
(2c)采用自适应学习率优化算法Adam,并以为学习率,通过L1对EfficientNet-B0通过L1对EfficientNet-B0的权重W1和偏置I1参数进行更新,得到第e1次迭代后的自适应缩放卷积神经网络EfficientNet-B0-e1;
(2d)将验证样本集Dval作为EfficientNet-B0-e1的输入,逐个提取每个验证样本增龄性变化区域的特征,通过该特征对验证样本对应的年龄进行回归预测,得到Dval对应的年龄预测值并采用平均误差函数计算中的每个年龄估计值与其对应的年龄标签标注的年龄的平均绝对误差MAEv_1;
(2e)判断MAEv_1MAEbest_1是否成立,若是,令MAEbest_1=MAEv_1,C1=0,并执行步骤(2f),否则令C1=C1+1,并执行步骤(2f);
(2f)判断e1=E1或者C1=3是否成立,若是,得到训练好的自适应缩放卷积神经网络Effibest,否则令e1=e1+1,并执行步骤(2b);
(3)构建显著图约束训练样本集Dtrain、显著图约束验证样本集Dval和显著图约束测试样本集Dtest:
(3a)将训练样本集Dtrain、验证样本集Dval和测试样本集Dtest作为训练好的自适应缩放卷积神经网络Effibest的输入,并基于Grad-CAM算法进行梯度反向传播,得到Dtrain、Dval和Dtest对应的显著图集Gtrain、Gval和Gtest;
(3b)对训练样本集Dtrain中的每个训练样本与显著图集Gtrain中对应的显著图进行拼接,同时对验证样本集Dval中的每个验证样本与显著图集Gval中对应的显著图进行拼接,对测试样本集Dtest中的每个测试样本与显著图集Gtest中对应的显著图进行拼接,实现对每个训练样本、每个验证样本和每个测试样本的显著性区域约束,显著性区域约束后的每个显著图约束样本的大小为H×H×2,2表示双通道,得到显著图约束训练样本集DGtrain、显著图约束验证样本集DGval和显著图约束测试样本集DGtest;
(4)获取自复制样本集和混合训练样本集:
(4a)对训练样本集Dtrain进行复制,并对Dtrain中的每个训练样本与复制的训练样本集Dt'rain中对应的训练样本进行拼接,同时对验证样本集Dval进行复制,并对Dval中的每个验证样本与复制的验证样本集Dv'al中对应的验证样本进行拼接,对测试样本集Dtest进行复制,并对Dtest中的每个测试样本与复制的测试样本集Dt'est中对应的测试样本进行拼接,实现将大小为H×H的样本转化为双通道的大小为H×H×2的自复制样本,得到自复制训练样本集DDtrain、自复制验证样本集DDval和自复制测试样本集DDtest;
(4b)对训练样本集Dtrain进行复制,并判断复制后的训练样本集中的每一个样本对应的年龄标签值是否小于25,若是,对Dtrain小于25的训练样本通过DDtrain中对应的自复制训练样本进行替换,否则,对Dtrain大于25的训练样本通过DGtrain中对应的显著图约束训练样本进行替换,得到混合训练样本集Mtrain;
(5)构建基于显著图约束的自适应缩放卷积神经网络EfficientNet-S:
构建包含顺次连接的Stem模块、Block1、Block2、Block3、Block4、Block5、Block6、Block7和最终层的基于显著图约束的自适应缩放卷积神经网络EfficientNet-S;其中Stem模块包含顺次连接的接收双通道的显著图约束训练样本的卷积神经网络、重缩放层、归一化层、零填充层、批归一化层和激活层;Block1包含一个Module1结构;Block2和Block3均包含顺次连接的Module2结构、Module3结构和加和层;Block4、Block5和Block6均包含顺次连接的Module2结构、Module3结构、加和层、Module3结构和加和层;Block7包含一个Module2结构;最终层包含顺次连接的卷积层、批归一化层和激活层;Module1结构包含顺次连接的深度卷积层、批归一化层和激活层;Module2构包含顺次连接的深度卷积层、批归一化层、激活层、零填充层、深度卷积层、批归一化层和激活层;Module3结构包含顺次连接的全局平均池化层、重缩放层、卷积层和卷积层;
(6)对基于显著图约束的自适应缩放卷积神经网络EfficientNet-S进行迭代训练:
(6a)初始化迭代次数为e2,最大迭代次数为E2,E2≥150,学习率为η2=0.0001,第e2次迭代的学习率为提升停滞次数为C2,最优平均绝对误差为MAEbest_2,并令e2=1,C2=0,MAEbest_2=9223372036854775807,以EfficientNet-S在自然图像数据集ImageNet上的预训练的权重和偏置参数作为EfficientNet-S的初始权重和偏置参数;
(6b)将混合训练样本集Mtrain划分为S2个混合训练样本子集每个子集包括b2个混合训练样本,并将作为EfficientNet-S的输入提取每个混合训练样本的增龄性变化区域的特征,通过这些特征对年龄进行回归预测,得到b2个年龄预测值,然后采用平均误差函数计算每个年龄预测值与其对应的年龄标签标注的年龄的平均绝对误差L2,其中S2≥700,表示第s2个混合训练样本子集,5≤b2≤10;
(6c)采用自适应学习率优化算法Adam,并以为学习率,通过L2对EfficientNet-S的权重和偏置参数进行更新,得到第e2次迭代后的自适应缩放卷积神经网络EfficientNet-S-e2;
(6d)将X片自复制验证样本集DDval中的每个自复制验证样本逐个输入至EfficientNet-B0-e2,提取每个自复制验证样本的增龄性变化区域的特征,并通过这些特征对年龄进行回归预测,得到DDval对应的年龄预测值
(6e)判断中的每个年龄预测值是否小于25,若是,将小于25的年龄预测值添加至预先定义的年龄预测值集合中,否则,将其余年龄预测值对应的DGval中显著图约束验证样本的输入至EfficientNet-S-e2中,提取每个显著图约束验证样本的增龄性变化区域的特征以及显著图中的注意力约束信息,并以该特征和约束信息对每个年龄进行回归预测,然后将预测结果添加至中,中所添加的所有年龄预测值即为验证样本集Dval的年龄估计;
(6f)以平均绝对误差函数计算中的每一个年龄估计值与其对应的年龄标签标注的年龄的平均绝对误差MAEv_2;
(6g)判断MAEv_2MAEbest_2是否成立,若是,令MAEbest_2=MAEv_2,C2=0,并执行步骤(6h),否则令C2=C2+1,并执行步骤(6h);
(6h)判断e2=E2或者C2=3是否成立,若是,得到训练好的基于显著图约束的自适应缩放卷积神经网络K,否则令e2=e2+1,并执行步骤(6b);
(7)获取年龄估计结果:
(7a)将自复制测试样本集DDtest中的每个自复制验证样本逐个输入至K中,提取每个自复制测试样本的增龄性变化区域的特征,并通过该特征对每个年龄进行回归预测,得到DDtest对应的年龄预测值
(7b)判断中的每个年龄预测值是否小于25,若是,将小于25的年龄预测值添加至预先定义的年龄预测值集合中,否则,将其余年龄预测值对应的DGtest中显著图约束验证样本的输入至EfficientNet-S-e2中,提取每个显著图约束验证样本的增龄性变化区域的特征以及显著图中的注意力约束信息,并以该特征和约束信息对每个年龄进行回归预测,然后将预测结果添加至中,中所添加的所有年龄预测值即为测试样本集Dtest的年龄估计。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于西安交通大学口腔医院;西安电子科技大学,未经西安交通大学口腔医院;西安电子科技大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202110410483.2/1.html,转载请声明来源钻瓜专利网。