PyTorch 实现循环神经网络判断人名属于哪个国家的常用名

PyTorch入门实战教程

在本文中,我们将构建并训练一个字符级别的循环神经网络来对单词进行分类。在基于字符级别的循环神经网络里,单词将是一个个字符(字母)的序列,利用循环神经网络具备一定的时序记忆功能来实现判断一个单词的类别。具体在本文的示例中,我们将对来自18种语言的上千个人名进行训练,并让网络根据给定名字的字母拼写来预测这个名字最可能是哪个国家常用的名字。

推荐先期阅读

在阅读本文前,读者最好具备Python,Pytorch的基本知识,并能理解“张量”这个概念。同时有一定的循环神经网络和LSTM知识对于理解本文也是很有帮助的。

本文从“准备数据”、“创建网络模型”、“训练网络”、“评估网络”、“自我练习”等五各方面来展开,其中最后一部分“自我练习”是留给读者自己实践的内容。

准备数据

此处下载数据并把它解压到当前目录。下载的压缩文件中data/names目录下包含18个以语言名命名的txt文件,每一个文件里包含大量的人名字符串,每一行表示一个人名。大多数是常用字符,不过我们仍然需要把这些人名字符串从unicode字符转化为ASCII字符。

我们要把所有这些数据加载到一个字典中,字典的键是语言类别字符串,对应的值是一个包含了该语言人名的列表,类似于{language:[names …]}。同时我们也维护一个类别列表,保存了所有语言类别,类似于[language …]。下面的代码做了一些准备工作:

执行上面的代码会输出:

category_lines字典和all_categories列表保存了我们要训练的数据,后文要用到。用下面的代码简要了解下字典保存的数据样式:

输出:

将人名转化为张量

为了能对人名进行分析,我们需要把以字符串表示的人名转化为以张量形式表示的人名。由于本文介绍的是基于字符级别的循环网络,我们将使用独热向量(one-hot vector)技术来表示人人名中出现的单个字符,而用一个2D的矩阵来表示一个人名字符串。单个字符的独热向量的尺寸是<1 × n_letters>。其中该向量里大部分维度的值均为0,只有对应该字符在所有字符列表(all_letters)中的索引位置的值为1。例如字符”b”=<0 1 0 0 0 …>。

把组成人名的字符串的每一个字符的向量连起来,就是表示人名的2维矩阵。中间多出来的长度为1的维度是因为PyTorch默认所有的张量都是基于块(batch)的,这里我们不使用块,因此将其设为1。

我们使用下面的代码来完成上述工作,并用字符”J”和人名字符串”Jones”来看看效果:

输出:

创建网络模型

基于PyTorch可以自动计算梯度,构建反向传播算法这一优势,我们只需要按照传统的前向运算的步骤来创建网络。这里我们使用的RNN网络仅包括2个分别针对输入数据和隐藏状态的线性变换层、以及一个对输出层的LogSoftmax变换。网络结构如下:

具体的代码如下:

实际运行一个循环网络,我们需要送给网络输入数据,同时网络自身隐藏层的输出也作为下一时刻的输入的一个组成部分。同时在每一个时间步,我们针对网络隐藏层的输出做一个Logsoftmax变换,得到整个网络的输出。可以认为,每一个时刻网络的输入由人名字符串中的某一个字符所对应的张量与网络隐藏层的输出联合组成。整个网络除了确定数据之间的变换类型外,最重要的参数就是输入数据、隐藏层输出以及网络整体输出的尺寸了。本例中,输入数据的尺寸对应于单个字符独热向量的大小,也就是所有字符数量;隐藏层尺寸可以根据需要适当调整,这里我们使用128;输出层尺寸即是所有语言的数量。

由于PyTorch网络模型接受的数据都必须是Variable对象,因此在把数据送入网络前都要把张量变为对应的Variable,下面的代码展示了把字符\\’A\\’送入网络运行得到的输出。

为了提高效率,我们以人名字符串而不是单个字符为单位来创建张量和对应的变量,在实际送入网络运行时,我们会从变量中按照索引来送入对应的字符变量。同样模型在运行前需要对隐藏层进行初始化。下面的代码展示了我们如何创建一个基于人名字符串的变量、隐藏层的初始化以及将一个字符串送入模型计算得到的输出结果:

输出:

你可以看到,输出是一个<1 × n_categories>的张量,其中的每一个数据代表着该字符串对应语言的几率,数值越大,人名字符串属于该语言的可能性越高。

训练网络

准备训练

在训练网络前,我们还需要做一些准备工作,我们需要写一个方法来直观地查看网络的输出:

代码输出:

我们也需要一个办法来快速的获取训练数据(输入数据和标签数据):

上段代码输出:

训练过程

有了之前的准备工作,训练一个网络就变得很简单了,我们只需要声明一下我们使用的损失计算方法和参数更新方法,就可以把数据送入网络训练了:

在每一个训练循环里:

1) 生成一个输入数据张量和输出标签张量

2) 生成一个初始化为0的隐藏层状态

3) 把每一个字符的张量数据、隐藏层状态数据送入网络输入层

4) 计算网络的输出,比较其与标签张量的差别,得到损失。

5) 反向传播更新参数

6) 给出网络的输出和损失

我们可以使用上面的方法训练大量的样本数据,同时由于训练方法同时给出了损失值,我们还可以将损失值每隔一定迭代次数就读取出来,最后绘制成损失随迭代次数曲线图,观察网络训练进度:

上面的代码给出的输出类似如下:

绘制结果

下面的代码将训练进度绘制来:

评估网络

我们可以设计一个矩阵来反映模型分辨不同语言的能力,暂称这个矩阵为“混淆矩阵”,该矩阵的数据提示了模型对于每一个实际的语言类型(行)的预测语言类型(列)。我们编写一个evaluate方法来完成这个工作,该方法类似于上面提到的train方法,只不过不需要反向传播更新参数:

代码输出混淆矩阵图:

上图中,用不同颜色的小方格表示了模型把实际语言类型(行)的人名预测为另一种语言(列)的相对次数,对角线的次数较多,说明正确预测的次数也较多,对于非对角线颜色较亮的区域则是误判较多的例子。比如经常把“汉语”人名解读为“越南语”或“韩语”等。

我们也可以编写代码在终端接受输入一个人名,给出该人名字符串最可能的三种语言类型:

输出:

以上就是示例的所有内容。上述代码被组织成多个文件,读者可以在这里下载:

你可以运行train.py文件来训练和保存网络模型;联合一个人名参数运行predict.py观察网络预测结果:

运行server.py后访问:localhost:5533/Yourname 来获取json格式表示的预测结果

自我练习

  • 尝试不同的数据集映射:一行 –> 类别。例如:输入一个单词训练并判断其属于哪一种语言;根据姓名中的姓训练并判断其性别;根据作品中的角色名称训练并预测该作品的作者;根据网页的标题训练并判断是属于博客还是Reddit网站的某一子版块。
  • 你也可以针对RNN网络做一些完善工作,比如在网络里增加更多的线性变换层;尝试使用基于LSTM或GRU的RNN;或者把多个RNNs组合起来形成一个更高级的网络等等。

文章来源:知乎专栏

PyTorch入门实战教程
除特别注明外,本站所有文章均为 PyTorch 中文网原创,转载请注明出处:https://www.pytorchtutorial.com/pytorch-rnn-for-name-classification/

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部