NL2SQL融合知识图谱在设备运维数据检索中的应用
Application of NL2SQL with knowledge graph fusion in equipment maintenance data retrieval作者机构:东华大学计算机科学与技术学院上海201620
出 版 物:《智能计算机与应用》 (Intelligent Computer and Applications)
年 卷 期:2024年第14卷第9期
页 面:118-124页
学科分类:081203[工学-计算机应用技术] 08[工学] 0835[工学-软件工程] 0812[工学-计算机科学与技术(可授工学、理学学位)]
摘 要:设备运维现场需要进行设备信息和运行状况的信息检索,但功能固定的查询程序无法满足多变的信息获取需求,而使用自然语言的交互检索可以提供更友好的支持。本文以行车设备运维的数据检索为目标,基于M-SQL模型设计并实现了自然语言到结构化查询语句(NL2SQL)的生成和自动检索,并融合知识图谱提高SQL语句中字段表达的准确性。首先基于开源通用领域数据集训练得到基于子任务的NL2SQL多任务学习模型;然后在标注的行车领域查询数据集上进行微调,使模型理解行车领域词汇;最后利用行车数据库知识图谱实体链接修正生成的SQL查询语句中数据不规范的问题。实验结果表明,该系统方案可以显著提高模型在行车数据的查询效率和准确性,满足使用自然语言的人机交互查询需求。