公开(公告)号：CN118335329B

公开(公告)日：2024-12-20

申请号：CN202410440608.X

申请日：2024-04-12

Applicant: 重庆邮电大学

Inventor： 代劲 ,  汪志勋 ,  胡峰 ,  于洪

IPC: G16H50/30 ,  G06N3/045 ,  G06N3/08

Abstract: 本发明属于医疗领域，具体涉及一种基于多头注意力机制的胃癌肝转移风险预测方法，包括：获取胃癌数据，并对胃癌数据进行预处理，得到预处理结果；利用多头注意力机制对预处理结果进行特征交互，得到特征交互信息；利用改进的多头注意力机制对预处理结果进行样本交互，得到样本交互信息；根据特征交互信息和样本交互信息利用神经网络得到胃癌肝转移风险结果；根据胃癌肝转移风险结果计算损失函数值，根据损失函数值更新模型参数；当达到预设的迭代次数时，得到训练好的胃癌肝转移风险预测模型；本发明通过两种多头注意力机制同时捕获特征间和样本间的信息，并利用神经网络将特征间和样本间的信息融合，增强了特征的表示，提高了风险预测精度。

公开(公告)号：CN118335329A

公开(公告)日：2024-07-12

申请号：CN202410440608.X

申请日：2024-04-12

Applicant: 重庆邮电大学

Inventor： 代劲 ,  汪志勋 ,  胡峰 ,  于洪

IPC: G16H50/30 ,  G06N3/045 ,  G06N3/08

Abstract: 本发明属于医疗领域，具体涉及一种基于多头注意力机制的胃癌肝转移风险预测方法，包括：获取胃癌数据，并对胃癌数据进行预处理，得到预处理结果；利用多头注意力机制对预处理结果进行特征交互，得到特征交互信息；利用改进的多头注意力机制对预处理结果进行样本交互，得到样本交互信息；根据特征交互信息和样本交互信息利用神经网络得到胃癌肝转移风险结果；根据胃癌肝转移风险结果计算损失函数值，根据损失函数值更新模型参数；当达到预设的迭代次数时，得到训练好的胃癌肝转移风险预测模型；本发明通过两种多头注意力机制同时捕获特征间和样本间的信息，并利用神经网络将特征间和样本间的信息融合，增强了特征的表示，提高了风险预测精度。