知识图谱嵌入的机械故障问题生成

摘要

问题生成对于提高问答系统、信息搜索等下游任务的智能水平至关重要.然而,目前问题生成方法主要面向通用领域,在特定垂直领域,由于因缺乏专业的训练数据,这些方法往往会产生误解或幻觉.针对机械故障领域,文中提出一种知识图谱嵌入预训练模型的问题生成方法.首先,通过改进预训练模型T5的输入输出和训练策略,使其适应问题生成任务;其次,在T5的编码端,通过嵌入领域知识图谱,以增强模型对机械故障领域知识的理解能力.实验表明,相对于基线模型,本文所提方法在BLEU-4和ROUGE-L评价指标上都取得了显著提升,生成的问题更加多样,而且具有更好的可解释性.验证了引入知识图谱嵌入技术对提升问题生成质量方面的有效性.

关键词

机械故障 / 问题生成 / 知识图谱 / 预训练模型