计算机与现代化 ›› 2025, Vol. 0 ›› Issue (08): 16-23.doi: 10.3969/j.issn.1006-2475.2025.08.003
摘要:
摘要:面向推荐的对话生成任务旨在通过人机对话交互获取用户偏好,以实现精准推荐。针对现有研究工作存在对话推荐类型单一和生成回复质量低的问题,本文提出一种基于统一预训练语言模型(Unified Language Model pre-training, UniLM)的目标驱动的推荐对话生成模型(Goal Driven Recommendation-oriented Dialog Generation model, GDRDG)。该模型包括文本表示模块、多头编码模块、解码模块以及一种特殊的注意力掩码机制。其中,文本表示模块通过UniLM对输入文本进行向量化表示,确保模型能捕获文本的深层次语义特征;多头编码模块利用多头自注意力机制捕捉全局上下文信息,提高生成回复的连贯性和相关性;解码模块生成当前轮对话目标及基于该目标的回复,确保回复符合上下文并将对话向预期目标引导;特殊的注意力掩码机制则通过控制解码过程中的信息流,确保模型仅关注当前轮次相关信息,以提高回复质量。实验结果表明,GDRDG模型在BLEU、Distinct、F1和Hit@1等指标上均优于现有方法,验证了模型的有效性和先进性。
中图分类号: