一种肺功能下降预测装置及其预测方法

文档序号:170979 发布日期:2021-10-29 浏览:12次 >En<

阅读说明:本技术 一种肺功能下降预测装置及其预测方法 (Lung function decline prediction device and prediction method thereof ) 是由 陈舞 孙军梅 李秀梅 于 2021-08-26 设计创作,主要内容包括:本发明公开一种肺功能下降预测装置及其预测方法。本发明提出的肺功能进展预测模型包括CT特征提取网络、多模态特征预测网络;CT特征提取网络用于对预处理后的肺部CT影像进行CT特征提取,多模态特征预测网络用于预测肺功能进展情况,输入为CT特征提取网络提取的CT特征与临床特征融合后形成的多模态特征,输出为未来不同周数时的FVC预测值。本发明构建的肺功能进展预测模型通过CT特征提取网络提高对CT特征的提取能力,并通过多模态特征预测网络利用多模态数据对肺功能进展情况进行预测,有效提高模型预测的准确性。(The invention discloses a lung function decline prediction device and a prediction method thereof. The lung function progress prediction model provided by the invention comprises a CT (computed tomography) feature extraction network and a multi-modal feature prediction network; the CT feature extraction network is used for carrying out CT feature extraction on the preprocessed lung CT images, the multi-mode feature prediction network is used for predicting the lung function progress condition, multi-mode features formed by fusing CT features extracted by the CT feature extraction network and clinical features are input, and the FVC prediction values in different weeks in the future are output. The lung function progress prediction model constructed by the method improves the extraction capability of CT characteristics through the CT characteristic extraction network, and predicts the lung function progress by multi-modal data through the multi-modal characteristic prediction network, thereby effectively improving the accuracy of model prediction.)

一种肺功能下降预测装置及其预测方法

技术领域

本发明属于人工智能领域,涉及一种基于多模态数据的肺功能下降预测装置及其非诊断目的的肺功能下降预测方法。

背景技术

特发性肺纤维化(Idiopathic Pulmonary Fibrosis,IPF)是一种起病隐匿,病因不明,组织学或影像学表现为普通型间质性肺炎,以进行性呼吸困难和肺功能下降为特征的慢性肺疾病,发病率和患病率分别在每一万人0.09-1.30和0.33-4.51的范围。由于IPF病情呈渐进式发展、诊断手段有限,最终可能导致完全性肺功能障碍。IPF患者一般的中位生存时间只有3-5年,病情的预后存在较大的困难。虽然现在还没有广泛使用的技术可以估计IPF病情的进展,但普遍认为IPF患者的肺功能下降情况可以对IPF的预后提供一定的指导建议。

发明内容

本发明的一个目的是针对现有技术的不足,提供一种非诊断目的的基于多模态数据的肺功能下降预测方法。

本发明非诊断目的的基于多模态数据的肺功能下降预测方法,包括以下步骤:

步骤(1)、获取历史肺部CT影像和对应临床文本数据;其中临床文本数据包括肺部影响因子、测量用力肺活量的周数、用力肺活量(Forced Vital Capacity,FVC)、用力肺活量与正常人标准值的百分比;所述肺部影响因子包括年龄、性别、吸烟情况;

步骤(2)、对历史肺部CT影像和对应临床文本数据进行预处理,构建数据集;

作为优选,步骤(2)具体是:

2-1对肺部CT影像进行预处理:去除无法打开的DICOM医学图像文件和不包含肺部信息的无价值CT图像数据;调整图像大小为统一的512*512。

2-2对临床文本数据进行预处理:去除临床文本数据中不完整、错误的数据;对临床文本数据进行特征工程,生成更有效的数据特征用于模型训练;进行Min-Max标准化处理;最后得到预处理后的临床特征。

2-3对数据集中不同周数及其对应的FVC值按照最小二乘法进行计算,获取FVC线性变化率,作为训练集的其中一个标签。

步骤(3)、构建肺功能进展预测模型,利用步骤(2)构建的数据集进行训练

所述肺功能进展预测模型包括CT特征提取网络、多模态特征预测网络;所述CT特征提取网络用于对预处理后的肺部CT影像进行CT特征提取,所述多模态特征预测网络用于预测肺功能进展情况,输入为CT特征提取网络提取的CT特征与临床特征融合后形成的多模态特征,输出为未来不同周数时的FVC预测值。

3-1构建CT特征提取网络

CT特征提取网络以InceptionV1为骨干网络,包括前端下采样模块和多尺度CT特征融合模块。

所述前端下采样模块包括1*1、3*3卷积层和最大池化层,通过下采样获取高维特征,降低网络参数量,加快计算速度的同时防止过拟合。

作为优选,所述前端下采样模块包括依次级联的三个串联的3*3卷积层、最大池化层、1*1卷积层、两个串联的3*3卷积层、最大池化层;

所述多尺度CT特征融合模块包括依次级联的n1个串联的多尺度CT特征融合模块A、第一最大池化层、n2个串联的多尺度CT特征融合模块B、第二最大池化层、n3个串联的多尺度CT特征融合模块C、平均池化层、全局平均池化层、第一全连接层、第二全连接层,n1≥1,n2≥1,n3≥1。

作为优选,所述多尺度CT特征融合模块包括依次级联的2个串联的多尺度CT特征融合模块A、第一最大池化层、2个串联的多尺度CT特征融合模块B、第二最大池化层、2个串联的多尺度CT特征融合模块C、平均池化层、全局平均池化层、第一全连接层、第二全连接层;

所述多尺度CT特征融合模块A包括5条并联支路、特征融合层、残差连接层、改进CBAM通道注意力模块、1*1卷积升维层;所述5条并联支路中第1支路包括一个1*1卷积层;第2支路包括1*1卷积层、3*3卷积层、空洞率为2的空洞卷积层、特征图相加层,1*1卷积层的输出端接3*3卷积层的输入端、空洞率为2的空洞卷积层的输入端,3*3卷积层的输出端、空洞率为2的空洞卷积层的输出端接特征图相加层;第3支路包括1*1卷积层、5*5卷积层、空洞率为2的空洞卷积层、特征图相加层,1*1卷积层的输出端接5*5卷积层的输入端、空洞率为2的空洞卷积层的输入端,5*5卷积层的输出端、空洞率为2的空洞卷积层的输出端接特征图相加层;第4支路包括依次级联的平均池化层、1*1卷积层;第1至第4支路的输出特征图将在特征融合层进行Concatenate连接,形成多尺度CT特征;残差连接层对多尺度CT特征与第5支路的原输入特征图进行相加,提升网络的表征能力;改进CBAM通道注意力模块用于接收残差连接层处理后的特征,为多尺度CT特征添加注意力权重,抑制无用信息;1*1卷积升维层对改进CBAM通道注意力模块输出的多尺度CT特征进行跨通道的特征融合,同时用最少的参数拓宽网络通道数。

所述多尺度CT特征融合模块B包括5条并联支路、特征融合层、残差连接层、1*1卷积升维层;所述5条并联支路中第1、2、4支路的结构与多尺度CT特征融合模块A相同;第3支路包括1*1卷积层、第一3*3卷积层、第二3*3卷积层、空洞率为2的空洞卷积层、特征图相加层,1*1卷积层的输出端接第一3*3卷积层的输入端、第二3*3卷积层的第一输入端、空洞率为2的空洞卷积层的输入端,第一3*3卷积层的输出端接第二3*3卷积层的第二输入端,第二3*3卷积层的输出端、空洞率为2的空洞卷积层的输出端接特征图相加层。第1至第4支路的输出特征图在特征融合层进行Concatenate连接,形成多尺度CT特征;残差连接层对多尺度CT特征与第5支路的原输入特征图进行相加,提升网络的表征能力;1*1卷积升维层对残差连接层处理后的特征进行跨通道的特征融合,同时用最少的参数拓宽网络通道数。

所述多尺度CT特征融合模块C包括5条并联支路、特征融合层、残差连接层、改进的CBAM通道注意力模块、1*1卷积升维层;所述5条并联支路中第1支路的结构与多尺度CT特征融合模块A相同;第2支路包括1*1卷积层、1*3卷积层、3*1卷积层、空洞率为2的空洞卷积层、特征图相加层,1*1卷积层的输出端接1*3卷积层的输入端、空洞率为2的空洞卷积层的输入端,1*3卷积层的输出端接3*1卷积层的输入端,3*1卷积层的输出端、空洞率为2的空洞卷积层的输出端接特征图相加层;第3支路包括1*1卷积层、1*3卷积层A、3*1卷积层A、空洞率为2的空洞卷积层A、特征图相加层A、1*3卷积层B、3*1卷积层B、空洞率为2的空洞卷积层B、特征图相加层B,1*1卷积层的输出端接1*3卷积层A的输入端、空洞率为2的空洞卷积层A的输入端、1*3卷积层B的输入端,1*3卷积层A的输出端接3*1卷积层A的输入端,3*1卷积层A的输出端、空洞率为2的空洞卷积层A的输出端接特征图相加层A的输入端,特征图相加层A的输出端接1*3卷积层B的输入端、空洞率为2的空洞卷积层B的输入端,1*3卷积层B的输出端接3*1卷积层B的输入端,3*1卷积层B的的输出端、空洞率为2的空洞卷积层B的输出端接特征图相加层B的输入端;第4支路包括平均池化层、最大池化层、特征图相加层、1*1卷积层,平均池化层的输出端、最大池化层的输出端接特征图相加层的输入端,特征图相加层的输出端接1*1卷积层的输入端。

所述改进CBAM的通道注意力模块(Improved Channel Attention of CBAM,CBAM-ICA)是将输入特征图分别进行全局平均池化和全局最大池化,然后经过两个1*1卷积层并通过Sigmoid激活函数生成两个不同的通道注意力特征图,最后将两个通道注意力特征图相乘形成最终的注意力权重,并将其与输入特征图F逐像素相乘得到最终输出特征F'。具体过程可用式1表示:

F'=(ε(C(Pag(F)))×ε(C(Pmx(F))))⊙F (1)

式中,F表示输入特征图,Pag表示全局平均池化,Pmx表示全局最大池化,C代表两个1*1卷积层,ε表示Sigmoid激活函数,F'表示通过改进CBAM通道注意力模块后的输出特征。

3-2构建多模态特征预测网络

所述多模态特征预测网络包括第一多模态特征模块和第二多模态特征模块;

所述第一多模态特征模块以CT特征提取网络提取的CT特征和预处理过后的肺部影响因子作为输入,FVC线性变化率作为输出,包括依次级联的Concatenate特征融合层A、全连接层;所述Concatenate特征融合层A用于将CT特征提取网络提取的CT特征和预处理过后的临床特征(年龄、性别、吸烟情况)进行Concatenate融合,得到第一多模态特征;所述全连接层用于预测FVC线性变化率。预测FVC线性变化率使用的损失函数为平均绝对误差MAE(Mean Absolute Error),它是目标值与预测值之差绝对值的和,表示了预测值的平均误差幅度,具有更好的鲁棒性。为了缓解网络训练过程中过拟合现象的出现,并使网络具有良好的泛化性,在全连接层之前添加了dropout层。

所述第二多模态特征模块以第一多模态特征模块输出的FVC线性变化率和所有临床特征作为输入,未来不同周数时的FVC预测值作为输出,包括依次级联的注意力模块、多层感知机(muti-layer perception,MLP)。

所述注意力模块计算过程用式2表示:

式中,Fx表示第一多模态特征模块输出的FVC线性变化率和所有临床特征,M表示两个全连接层,ε表示Sigmoid激活函数,Fwx表示通过注意力模块后的输出特征。多模态特征Fx经过两个全连接层,然后通过Sigmoid激活函数计算得到注意力权重,最后将注意力权重与输入特征Fx相乘再相加,得到最后的输出特征Fwx

所述多层感知机包括依次级联的第一全连接层、ELU激活函数层、第二全连接层、GELU激活函数层、第三全连接层,以注意力模块的输出特征图为输入,以FVC值为输出。

步骤(4)、利用训练好的肺功能进展预测模型,实现肺功能进展预测。

本发明的另一个目的是提供一种基于多模态数据的肺功能下降预测装置,包括:

肺部数据获取模块,用于获取肺部CT影像以及对应肺部影响因子,肺部影响因子包括年龄、性别、吸烟情况;

肺部数据预处理模块,用于对所述肺部CT影像和肺部影响因子进行预处理;

肺功能进展预测模块,用于利用训练好的肺功能进展预测模型对预处理后的肺部CT影像和肺部影响因子进行处理,以获取未来不同周数时的FVC预测值。

本发明的有益效果是:

(1)肺功能预测准确率高

本发明构建的肺功能进展预测模型通过CT特征提取网络提高对CT特征的提取能力,并通过多模态特征预测网络利用多模态数据对肺功能进展情况进行预测,有效提高模型预测的准确性。

(2)肺功能进展预测模型泛化能力增强

本发明构建的肺功能进展预测模型进行了一系列避免过拟合的措施:采用Adam优化器,在训练时自适应地调节学习率;对CT特征提取网络添加dropout层,缓解模型训练过程中过拟合现象的出现;在预测FVC线性变化率时使用平均绝对误差MAE作为损失函数,具有更好的鲁棒性;在多模态特征预测网络中对FVC值进行预测,使用K折交叉验证法进行训练,一定程度上减小过拟合的风险;在全连接层后使用ELU激活函数和GELU激活函数,提高对噪声的鲁棒性和模型泛化能力;设置训练提前终止值,loss值超过15次没有下降则停止训练。这些措施最终增强了模型的泛化能力。

(3)肺功能进展预测模型能够对肺功能进展进行有效预后

肺功能进展预测模型可以对未来不同周数的FVC值进行预测,以此更好地了解被预测者肺功能衰退的严重程度,对被预测者肺功能的预后具有指导意义。

附图说明

图1肺功能进展预测模型的结构示意图;

图2CT特征提取网络的结构示意图;

图3多尺度CT特征融合模块A的结构示意图;

图4多尺度CT特征融合模块B的结构示意图;

图5多尺度CT特征融合模块C的结构示意图;

图6改进CBAM的通道注意力模块的结构示意图;

图7多模态特征预测网络的结构示意图。

具体实施方式

下面结合具体实施例对本发明做进一步的分析。

一种非诊断目的的基于多模态数据的肺功能下降预测方法,包括以下步骤:

步骤(1)、获取历史肺部CT影像和对应临床文本数据;其中临床文本数据包括肺部影响因子、测量用力肺活量的周数、用力肺活量(Forced Vital Capacity,FVC)、用力肺活量与正常人标准值的百分比;所述肺部影响因子包括年龄、性别、吸烟情况;

表1临床文本数据

步骤(2)、对历史肺部CT影像和对应临床文本数据进行预处理,构建数据集;具体是:

2-1对肺部CT影像进行预处理;去除无法打开的DICOM医学图像文件和不包含肺部信息的无价值CT图像数据;调整图像大小为统一的512*512。

2-2对临床文本数据进行预处理;去除临床文本数据中不完整、错误的数据;对临床文本数据进行特征工程,生成更有效的数据特征用于模型训练;进行Min-Max标准化处理;最后得到预处理后的临床特征。

2-3对数据集中不同周数及其对应的FVC值按照最小二乘法进行计算,获取FVC线性变化率,作为训练集的其中一个标签。

步骤(3)、构建如图1肺功能进展预测模型,利用步骤(2)构建的数据集进行训练

所述肺功能进展预测模型包括CT特征提取网络、多模态特征预测网络;所述CT特征提取网络用于对预处理后的肺部CT影像li进行CT特征提取,0≤i<N,N表示预测的用力肺活量周数,所述多模态特征预测网络用于预测肺功能进展情况,输入为CT特征提取网络提取的CT特征与临床特征融合后形成的多模态特征,输出为未来不同周数时的FVC预测值FVCN

3-1构建如图2所示的CT特征提取网络

CT特征提取网络以InceptionV1为骨干网络,包括前端下采样模块和多尺度CT特征融合模块。

与InceptionV1网络相比,本网络添加残差加改进的CBAM通道注意力模块来扩大网络的感受野,关注肺部区域的有效特征,添加与卷积层并行的空洞卷积模块,补充丢失的细节信息,最终形成三个不同的多尺度CT特征融合模块并进行两次串联堆叠。以此对肺部CT影像进行多尺度特征提取、融合,增强网络对CT特征的提取能力,获取更加准确有效的CT特征。

所述前端下采样模块包括1*1、3*3卷积层和最大池化层,通过下采样获取高维特征,降低网络参数量,加快计算速度的同时防止过拟合。

所述前端下采样模块包括依次级联的三个串联的3*3卷积层、最大池化层、1*1卷积层、两个串联的3*3卷积层、最大池化层;

所述多尺度CT特征融合模块包括依次级联的2个串联的多尺度CT特征融合模块A、第一最大池化层、2个串联的多尺度CT特征融合模块B、第二最大池化层、2个串联的多尺度CT特征融合模块C、平均池化层、全局平均池化层、第一全连接层、第二全连接层。

如图3,所述多尺度CT特征融合模块A包括5条并联支路、特征融合层、残差连接层、改进CBAM的通道注意力模块、1*1卷积升维层;所述5条并联支路中第1支路包括一个1*1卷积层;第2支路包括1*1卷积层、3*3卷积层、空洞率为2的空洞卷积层、特征图相加层,1*1卷积层的输出端接3*3卷积层的输入端、空洞率为2的空洞卷积层的输入端,3*3卷积层的输出端、空洞率为2的空洞卷积层的输出端接特征图相加层;第3支路包括1*1卷积层、5*5卷积层、空洞率为2的空洞卷积层、特征图相加层,1*1卷积层的输出端接5*5卷积层的输入端、空洞率为2的空洞卷积层的输入端,5*5卷积层的输出端、空洞率为2的空洞卷积层的输出端接特征图相加层;第4支路包括依次级联的平均池化层、1*1卷积层;第1至第4支路的输出特征图将在特征融合层进行Concatenate连接,形成多尺度CT特征;残差连接层对多尺度CT特征与第5支路的原输入特征图进行相加,提升网络的表征能力;改进CBAM通道注意力模块用于接收残差连接层处理后的特征,为多尺度CT特征添加注意力权重,抑制无用信息;1*1卷积升维层对改进CBAM通道注意力模块输出的多尺度CT特征进行跨通道的特征融合,同时用最少的参数拓宽网络通道数。

如图4,所述多尺度CT特征融合模块B包括5条并联支路、特征融合层、残差连接层、1*1卷积升维层;所述5条并联支路中第1、2、4支路的结构与多尺度CT特征融合模块A相同;第3支路包括1*1卷积层、第一3*3卷积层、第二3*3卷积层、空洞率为2的空洞卷积层、特征图相加层,1*1卷积层的输出端接第一3*3卷积层的输入端、第二3*3卷积层的第一输入端、空洞率为2的空洞卷积层的输入端,第一3*3卷积层的输出端接第二3*3卷积层的第二输入端,第二3*3卷积层的输出端、空洞率为2的空洞卷积层的输出端接特征图相加层。第1至第4支路的输出特征图在特征融合层进行Concatenate连接,形成多尺度CT特征;残差连接层对多尺度CT特征与第5支路的原输入特征图进行相加,提升网络的表征能力;1*1卷积升维层对残差连接层处理后的特征进行跨通道的特征融合,同时用最少的参数拓宽网络通道数。

如图5,所述多尺度CT特征融合模块C包括5条并联支路、特征融合层、残差连接层、改进CBAM的通道注意力模块、1*1卷积升维层;所述5条并联支路中第1支路的结构与多尺度CT特征融合模块A相同;第2支路包括1*1卷积层、1*3卷积层、3*1卷积层、空洞率为2的空洞卷积层、特征图相加层,1*1卷积层的输出端接1*3卷积层的输入端、空洞率为2的空洞卷积层的输入端,1*3卷积层的输出端接3*1卷积层的输入端,3*1卷积层的输出端、空洞率为2的空洞卷积层的输出端接特征图相加层;第3支路包括1*1卷积层、1*3卷积层A、3*1卷积层A、空洞率为2的空洞卷积层A、特征图相加层A、1*3卷积层B、3*1卷积层B、空洞率为2的空洞卷积层B、特征图相加层B,1*1卷积层的输出端接1*3卷积层A的输入端、空洞率为2的空洞卷积层A的输入端、1*3卷积层B的输入端,1*3卷积层A的输出端接3*1卷积层A的输入端,3*1卷积层A的输出端、空洞率为2的空洞卷积层A的输出端接特征图相加层A的输入端,特征图相加层A的输出端接1*3卷积层B的输入端、空洞率为2的空洞卷积层B的输入端,1*3卷积层B的输出端接3*1卷积层B的输入端,3*1卷积层B的的输出端、空洞率为2的空洞卷积层B的输出端接特征图相加层B的输入端;第4支路包括平均池化层、最大池化层、特征图相加层、1*1卷积层,平均池化层的输出端、最大池化层的输出端接特征图相加层的输入端,特征图相加层的输出端接1*1卷积层的输入端。

如图6,所述改进CBAM的通道注意力模块具体是:

将CBAM原计算注意力权重的两个全连接层改为1*1卷积层,并将原来先相加再使用激活函数计算权重的计算过程,转变为对两个卷积输出特征直接使用Sigmoid激活函数分别计算注意力权重值,然后将两个注意力权重进行相乘后,输出通道注意力权重值,并去除了CBAM注意力机制中的空间注意力部分。即,在改进的CBAM通道注意力模块中,首先将输入特征图分别进行全局平均池化和全局最大池化,然后经过两个1*1卷积层并通过Sigmoid激活函数生成两个不同的通道注意力特征图,最后将两个通道注意力特征图相乘形成最终的注意力权重,并将其与输入特征图F逐像素相乘得到最终输出特征F'。具体过程可用式(1)表示:

F'=(ε(C(Pag(F)))×ε(C(Pmx(F))))⊙F (1)

式中,F表示输入特征图,Pag表示全局平均池化,Pmx表示全局最大池化,C代表两个1*1卷积层,ε表示Sigmoid激活函数,F'表示通过改进CBAM通道注意力模块后的输出特征。

3-2构建如图7多模态特征预测网络

所述多模态特征预测网络包括第一多模态特征模块和第二多模态特征模块;

所述第一多模态特征模块以CT特征提取网络提取的CT特征和预处理过后的肺部影响因子作为输入,li对应的FVC线性变化率作为输出,包括依次级联的Concatenate特征融合层A、全连接层;所述Concatenate特征融合层A用于将CT特征提取网络提取的CT特征和预处理过后的临床特征(年龄、性别、吸烟情况)进行Concatenate融合,得到第一多模态特征;所述全连接层用于预测FVC线性变化率。预测FVC线性变化率使用的损失函数为平均绝对误差MAE(Mean Absolute Error),它是目标值与预测值之差绝对值的和,表示了预测值的平均误差幅度,具有更好的鲁棒性。为了缓解网络训练过程中过拟合现象的出现,并使网络具有良好的泛化性,在全连接层之前添加了dropout层。

所述第二多模态特征模块以第一多模态特征模块输出的FVC线性变化率和所有临床特征(肺部影响因子、li对应的测量用力肺活量的周数、用力肺活量FVC、用力肺活量与正常人标准值的百分比)作为输入,周数N时的FVC值作为输出,包括依次级联的注意力模块、多层感知机(muti-layer perception,MLP)。

所述注意力模块计算过程用式(2)表示:

式中,Fx表示第一多模态特征模块输出的FVC线性变化率和所有临床特征,M表示两个全连接层,ε表示Sigmoid激活函数,Fwx表示通过注意力模块后的输出特征。多模态特征Fx经过两个全连接层,然后通过Sigmoid激活函数计算得到注意力权重,最后将注意力权重与输入特征Fx相乘再相加,得到最后的输出特征Fwx

所述多层感知机包括依次级联的第一全连接层、ELU激活函数层、第二全连接层、GELU激活函数层、第三全连接层,以注意力模块的输出特征图为输入,以FVC值为输出。最后,第三全连接层输出三个特征值,分别为Out1、Out2和Out3。其中,Out2为FVC的预测值,而Out3减去Out1为标准差的值,用于计算拉普拉斯对数似然得分,进而评估模型。预测未来不同周数时的FVC预测值时使用的损失函数为分位数损失函数(Quantile loss function),分位值为[0.2,0.5,0.8]。训练过程中使用K折交叉验证法进行训练,从全部的多模态特征中随机选择K-1/K的样例作为训练集,剩余的作为验证集,同时设置训练提前终止值patience为15,在损失函数下降到一定程度时停止训练,这样在一定程度上减小过拟合的风险。第二多模态特征模块选择K值为6,进行六次训练,最后的预测结果为六次预测结果总和的平均值。在全连接层后使用ELU激活函数和GELU激活函数,提高对噪声的鲁棒性和网络泛化能力。

步骤(4)、利用训练好的肺功能进展预测模型,实现肺功能进展预测。

将测试集中预处理后的肺部CT影像和临床特征输入训练好的肺功能进展预测模型,模型将会通过预处理后的肺部CT影像和临床特征对未来不同周数时的FVC预测值进行预测,最终输出对测试集中被预测者的肺功能进展预测结果。

为了在分位数损失函数计算过程中使用合适的分位值以得到较好的预测结果,对分位值的选取进行了实验,实验结果如表2所示。

表2不同分位值结果对比

由实验结果可得,[0.2,0.5,0.8]作为分位数损失函数的分位值可以获取更好的拉普拉斯对数似然得分,提高模型预测结果的准确性。

为了验证在多尺度CT特征融合模块中引入改进的CBAM通道注意力机制得到的性能提升效果更好,进行注意力机制引入位置对比实验,实验结果如表3所示。

表3注意力机制引入位置对比

实验结果表明,将注意力机制引入多尺度CT特征融合模块中的合适位置可以有效提高模型性能,提升预测的准确性,其中在多尺度CT特征融合模块A和多尺度CT特征融合模块C结构中同时添加注意力机制比效果最优。

为了对比不同注意力模块在CT特征提取网络中的作用,同时使用SE注意力模块、CBAM注意力模块、ECA注意力模块、scSE注意力模块、CBAM-ICA模块等注意力机制进行对比实验。注意力模块主要添加在CT特征提取网络中多尺度CT特征融合模块A、C的多尺度连接之后,实验结果如表4所示。

表4注意力模块对比实验

从表中可以看出,CBAM-ICA模块相比于其他注意力模块在提升模型性能上效果最好,而参数量提升不大、和CBAM注意力模块的参数量相同。

在将注意力机制和残差模块结合时有两种方式。一种结构a是在完成多尺度特征融合后,先对多尺度特征添加注意力权重,然后再与原始输入特征进行残差连接。另一种结构b是在完成多尺度特征融合后,先将多尺度特征与原始输入特征进行残差连接,再添加注意力机制。为验证两种模块结构的优劣,对模块结构a、b进行对比实验,实验结果如表5所示。

表5残差加CBAM-ICA模块结构对比

从表中可以看到,在相同参数量的条件下,结构b的拉普拉斯对数似然得分明显优于结构a。因此,CT特征提取网络中的多尺度CT特征融合模块采用结构b的方式进行残差加CBAM-ICA模块的构建。

为了验证肺功能进展预测模型中的CT特征提取网络提取CT特征的性能,本发明使用不同网络替换CT特征提取网络进行对比实验。实验选取的对比网络有InceptionV1、InceptionV3、Inception_Resnet_V2、ResNet50、DenseNet121和EfficientNetB0。实验结果如表6所示。

表6CT特征提取网络效果对比

从表中观察到,基于使用Inception多尺度模块的三个网络作为CT特征提取网络时在测试集上的拉普拉斯对数似然得分优于其他大部分网络,因此本发明以InceptionV1作为CT特征提取网络的骨干网络。与其他网络相比,本发明模型在测试集上得到的拉普拉斯对数似然得分为-6.8107,获得了最好的效果,而且所用网络参数量也较少。

对目前已有的几种肺功能进展预测方法进行对比实验,以验证肺功能进展预测模型的有效性。实验对比的模型方法有:FibrosisNet、Fibro-CoSANet、DNN+GBDT+NGBoost+ElasticNet集成模型。实验结果如表7所示。

表7肺功能进展预测方法对比

由实验结果可得,本发明模型相比于现有的肺功能进展预测方法获得了更好的拉普拉斯对数似然得分。因此,本发明模型可以更加准确的预测肺功能进展情况。

消融实验使用的基础模型为去除了CT特征提取网络中残差模块、空洞卷积模块、CBAM-ICA注意力模块这三个模块后的肺功能进展预测模型,并在实验过程中分别添加这三个模块进行消融实验,最终与基础模型的实验结果进行对比。实验结果如表8所示。

表8模型消融实验

由实验结果可得,肺功能进展预测模型中的CT特征提取网络在分别添加残差模块、空洞卷积模块、CBAM-ICA注意力模块后获得的拉普拉斯对数似然得分均有不同程度的提升。而当同时添加三个模块时,模型的预测得分最佳。这表明包含残差模块、空洞卷积模块、CBAM-ICA注意力模块的肺功能进展预测模型的预测结果更加准确。

为了验证多模态数据的有效性,对本发明基于多模态数据的肺功能进展预测模型与仅使用临床文本数据或肺部CT图像数据的模型方法进行对比,实验结果如表9所示。

表9不同模态数据预测效果

由实验结果可得,仅使用肺部CT图像数据的模型方法的拉普拉斯对数似然得分低于仅使用临床文本数据的模型方法。但这些只使用一种医学模态数据的方法的得分均远远小于基于多模态数据的肺功能进展预测模型的拉普拉斯对数似然得分。由此可见,多模态数据相比于单个医学模态数据,可以有效提高模型预测的准确性。

上述实施例并非是对于本发明的限制,本发明并非仅限于上述实施例,只要符合本发明要求,均属于本发明的保护范围。

18页详细技术资料下载
上一篇:一种医用注射器针头装配设备
下一篇:免疫检查点抑制剂相关MACE发生风险的预测方法

网友询问留言

已有0条留言

还没有人留言评论。精彩留言会获得点赞!

精彩留言,会给你点赞!