在审

轨道交通大模型检索增强生成方法、系统、应用及介质

徐绍龙、胡云卿、熊艳、黄赫、窦泽春、许清、钱卓、曾升、黄利辉、郭亮、李凯、左宇辰
株洲中车时代电气股份有限公司

摘要

本发明涉及一种轨道交通大模型检索增强生成方法、系统、应用及介质,该方法接收用户查询文本,利用大语言模型将其转换为三元组序列,并对三元组基于语义关系排序。通过嵌入模型获取序列中实体和关系的嵌入向量,据此从知识图谱中检索候选实体和关系集合。采用三元组序列检索算法,计算排序后序列与候选序列之间的三元组序列语义距离,选取前K个满足结构约束的候选序列。最后,将用户查询和前K个候选序列输入大语言模型生成回答,从而避免知识碎片化与结构失配,实现对城市轨道交通领域复杂多跳查询的可靠、精准响应。

暂无引用专利