[发明专利]基于记忆强化Transformer的胸腔X光诊断报告生成方法

摘要

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