一种对话模型训练方法、装置、设备及介质

本申请公开了一种对话模型训练方法、装置、设备及介质，涉及人工智能技术领域，包括：通过记忆模块并基于动态权重策略调整预设记忆库中的各当前记忆单元的重要性权重，按照重要性权重的大小顺序过滤低权重的记忆单元，以更新预设记忆库，获取当前预设记忆库；当预训练模型接收到用户发起的训练请求，输出满足相似度条件的若干目标相似度计算结果；将与各目标相似度计算结果对应的记忆单元作为目标记忆单元，对各目标记忆单元以及查询向量进行加权聚合，以训练记忆模块基于各目标记忆单元与查询向量输出对应的聚合记忆信息，并将聚合记忆信息输入至预训练模型，以便预训练模型根据聚合记忆信息与查询向量生成与查询请求对应的响应信息。