一种基于跨模态知识蒸馏的肺癌风险预测方法

    公开(公告)号:CN118366654A

    公开(公告)日:2024-07-19

    申请号:CN202410422193.3

    申请日:2024-04-09

    摘要: 本发明属于医疗健康大数据领域,具体涉及一种基于跨模态知识蒸馏的肺癌风险预测方法,包括选取训练完成的基于CT图像输入的I3D模型为教师模型;采用多层感知机模型构建学生模型;将CT图像数据集输入教师模型,获取第一中间特征向量集合;将结构化数据作为学生输入训练学生模型,学生模型训练过程中从结构化数据中抽取出的特征定义为第二中间特征;基于第一中间特征向量集合和第二中间特征向量集合,计算教师模型空间分布和学生模型空间分布;采用KL散度计算教师模型空间分布和学生模型空间分布间的散度损失;计算散度损失和交叉熵损失的加权和,并反向传播训练学生模型,直至学生模型参数收敛;将训练完成的学生模型作为肺癌风险预测模型,用于预测患者的肺癌风险数据辅助医生诊断。