深度学习项目——基于循环神经网络（RNN）的智能聊天机器人系统 – AI大道理

深度学习项目——基于循环神经网络（RNN）的智能聊天机器人系统

基于循环神经网络（RNN）的智能聊天机器人系统

　　本设计研究智能聊天机器人技术，基于循环神经网络构建了一套智能聊天机器人系统，系统将由以下几个部分构成：制作问答聊天数据集、RNN神经网络搭建、seq2seq模型训练、智能聊天。经过实验，确定该系统可对本人的聊天话语进行快速并准确的回应，且回复可以模仿朋友的语气风格。

关键词: RNN神经网络； seq2seq模型； 聊天机器人；TensorFlow；

一、设计目标  

1.掌握聊天机器人系统原理； 

2.掌握循环神经网络（RNN）原理； 

3.掌握循环神经网络模型搭建与训练过程； 

4.掌握seq2seq、skip-gram、GRU等相关原理； 

设计内容与要求 

1.完成基于循环神经网络的聊天机器人系统神经网络设计； 

2.完成基于循环神经网络的聊天机器人系统seq2seq模型训练；

 3.搭建出聊天机器人系统；

二、聊天机器人系统组成

2.1 系统框架

 

图2.1　聊天机器人系统框架

 

 2.2 系统流程

 

 

图2.2　聊天机器人系统流程

三、聊天机器人开发

3.1 问答聊天数据采集和制作

本系统的聊天语聊来自与朋友的微信聊天记录，采集了一周的聊天并处理成一问一答的形式保存成one.txt与two.txt两个文件。

 

 

图3-1 聊天语料文件1

 

 

图3-2 聊天语料文件2

 

 3.2 数据预处理

（1）获取文件列表，找到one.txt与two.txt文件，从文件中读取中文词。

（2）去掉所有数字与标点符号，保留纯文字，用jieba进行分词，得到切割好的额分词与词的大小。One.txt中有3059个词，two.txt中有3160个词。

 

 

图3-3 分词结果

（3）统计文件中字符出现的次数，从小到大顺序进行排列，每个字符对应的排序就是它在字典中的编号。创建了1509个词的词典。

 

 

图3-4 词典

（4）将句子转成id数据经过分词后获取词典的索引值就是原文件里文字的id。把文件中问和答的id数据放到不同的文件里。将文件批量转成id文件。问文件的id为data_source_test.txt，答文件的id为data_target_test.txt。

 

 

图3-5 id形式

 

 

图3-6 问句id

 

 

 

图3-7 答句id

 

3.3 基于注意力机制的seq2seq模型搭建与训练

3.3.1 网络结构

网络结构为两层，每层100个GRUcell组成的网络，在seq2seq模型中编码器encoder与解码器decoder同为相同的结构。

 

 

 

 

 

图3-8 RNN神经网络内部分析

 

 

图3-9 GRU内部分析

 

 

 

3.3.2 输入层

Encoder的输入层输入为问句x的id，输入的长度不同问句长度不一样，最大可以接受100个词的问句；decoder的输入层由encoder的生成结果节点C与答句y共同输入，其中节点C参与到decoder的每一个序列都会经过权重w，权重w就可以以loss的方式通过优化器来调节就逐渐逼近与它紧密的那个词。标签y既参与计算loss，又参与节点运算。

3.3.3 隐藏层

在seq2seq模型中编码器encoder与解码器decoder同为相同的结构，都是100个GRUcell组成。在每个时刻， 隐层的输出ht依赖于当前词输入xt和前一时刻的隐层状态ht-1。

 

 

3.3.4 输出层

Encoder将输入编码映射到语义空间得到固定维数的向量，每个时刻Encoder都会生成c ，都将参与Decoder中解码的每个时刻，而不只是参与初始时刻。Decoder输出预测值。

 

 

3.3.5 训练结果

 

 

图3-10 模型训练结果

 3.4 利用模型进行智能聊天

 

 

 

  

 

四、 总结

本次设计主要研究了基于循环神经网络的智能聊天机器人系统。系统将由以下几个部分构成：制作问答聊天数据集、数据预处理、GRU网络搭建、seq2seq模型训练、智能聊天。经过实验，确定该系统可对本人的聊天话语进行快速并准确的回应，且回复可以模仿朋友的语气风格。 

五、 代码

见个人GitHub：https://github.com/hhhvvvddd/RNN_ChattingRobot_Week

 

——————

浅谈则止，深入理解AI大道理

扫描下方“AI大道理”，选择“关注”公众号

 

              欢迎加入！