基于记忆强化Transformer的胸腔X光诊断报告生成方法

作者：李放; 栾金锋; 王晓玲

来源：2023-03-03, 中国, CN202310207651.7.

本发明公开了一种基于记忆强化Transformer的胸腔X光诊断报告生成方法,根据实际需要收集若干胸腔X光图像及对应的诊断报告,基于诊断报告构建词表,得到每个诊断报告中词表包含单词的单词概率矩阵和词向量矩阵,连同胸腔X光图像构成一个训练样本,从而得到训练样本集；构建基于Transformer和记忆模块的诊断报告单词预测模型；采用训练样本对诊断报告单词预测模型进行训练,将待生成诊断报告胸腔X光图像输入训练好的诊断报告单词预测模型,得到单词概率矩阵,最后采用文本序列生成算法生成最终的诊断报告文本。本发明在Transformer模型中引入记忆模块,实现在仅有胸腔X光图像的情况下自动生成对应的诊断报告,同时提高诊断报告文本生成的质量。