xlnet+bilstm实现菜品正负评价分类

Posted 2021-12-05 AI浩

摘要

CMU和google brain联手推出了bert的改进版xlnet。在这之前也有很多公司对bert进行了优化，包括百度、清华的知识图谱融合，微软在预训练阶段的多任务学习等等，但是这些优化并没有把bert致命缺点进行改进。xlnet作为bert的升级模型，主要在以下三个方面进行了优化

• 采用AR模型替代AE模型，解决mask带来的负面影响
• 双流注意力机制
• 引入transformer-xl

今天我们使用xlnet+BiLSTM实现一个二分类模型。

数据集

数据集如下图：

是顾客对餐厅的正负评价。正面的评论是1，负面的是0。这类的数据集很多，比如电影的正负评论，商品的正负评论。

模型

模型结构如下：

思路：将xlnet做为嵌入层提取特征，然后传入BiLSTM，最后使用全连接层输出分类。创建xlnet_lstm模型，代码如下：

class xlnet_lstm(nn.Module):
    def __init__(self, xlnetpath, hidden_dim, output_size, n_layers, bidirectional=True, drop_prob=0.5):
        super(xlnet_lstm, self).__init__()

        self.output_size = output_size
        self.n_layers = n_layers
        self.hidden_dim = hidden_dim
        self.bidirectional = bidirectional

        # xlnet ----------------重点，xlnet模型需要嵌入到自定义模型里面
        self.xlnet = XLNetModel.from_pretrained(xlnetpath)
        for param in self.xlnet.parameters():
            param.requires_grad = True

        # LSTM layers
        self.lstm = nn.LSTM(768, hidden_dim, n_layers, batch_first=True, bidirectional=bidirectional)

        # dropout layer
        self.dropout = nn.Dropout(drop_prob)

        # linear and sigmoid layers
        if bidirectional:
            self.fc = nn.Linear(hidden_dim * 2, output_size)
        else:
            self.fc = nn.Linear(hidden_dim, output_size)

        # self.sig = nn.Sigmoid()

    def forward(self, x, hidden):
        # 生成xlnet字向量
        x = self.xlnet(x)[0]  # xlnet 字向量

        # lstm_out
        # x = x.float()
        lstm_out, (hidden_last, cn_last) = self.lstm(x, hidden)
        # print(lstm_out.shape)   #[batchsize,64,768]
        # print(hidden_last.shape)   #[4, batchsize, 384]
        # print(cn_last.shape)    #[4,batchsize, 384]

        # 修改 双向的需要单独处理
        if self.bidirectional:
            # 正向最后一层，最后一个时刻
            hidden_last_L = hidden_last[-2]#[batchsize, 384]
            # 反向最后一层，最后一个时刻
            hidden_last_R = hidden_last[-1]#[batchsize, 384]
            # 进行拼接
            hidden_last_out = torch.cat([hidden_last_L, hidden_last_R], dim=-1) #[batchsize, 768]
        else:
            hidden_last_out = hidden_last[-1]  # [batchsize, 384]
        # dropout and fully-connected layer
        out = self.dropout(hidden_last_out) #out的shape[batchsize,768]
        out = self.fc(out)

        return out

    def init_hidden(self, batch_size):
        weight = next(self.parameters()).data

        number = 1
        if self.bidirectional:
            number = 2

        if (USE_CUDA):
            hidden = (weight.new(self.n_layers * number, batch_size, self.hidden_dim).zero_().float().cuda(),
                      weight.new(self.n_layers * number, batch_size, self.hidden_dim).zero_().float().cuda()
                      )
        else:
            hidden = (weight.new(self.n_layers * number, batch_size, self.hidden_dim).zero_().float(),
                      weight.new(self.n_layers * number, batch_size, self.hidden_dim).zero_().float()
                      )

        return hidden

xlnet_lstm需要的参数功6个，参数说明如下：

​ --xlnetpath：xlnet预训练模型的路径

​ --hidden_dim：隐藏层的数量。

​ --output_size：分类的个数。

​ --n_layers：lstm的层数

​ --bidirectional：是否是双向lstm

​ --drop_prob：dropout的参数

定义xlnet的参数，如下：

class ModelConfig:
    batch_size = 2
    output_size = 2
    hidden_dim = 384  # 768/2
    n_layers = 2
    lr = 2e-5
    bidirectional = True  # 这里为True，为双向LSTM
    # training params
    epochs = 10
    # batch_size=50
    print_every = 10
    clip = 5  # gradient clipping
    use_cuda = USE_CUDA
    xlnet_path = 'xlnet-base-chinese'  # 预训练bert路径
    save_path = 'xlnet_bilstm.pth'  # 模型保存路径

batch_size：batchsize的大小，根据显存设置。

output_size：输出的类别个数，本例是2.

hidden_dim：隐藏层的数量。

n_layers：lstm的层数。

bidirectional：是否双向

print_every：输出的间隔。

use_cuda：是否使用cuda，默认使用，不用cuda太慢了。

xlnet_path：预训练模型存放的文件夹。

save_path：模型保存的路径。

下载预训练模型

本例使用的预训练模型是xlnet-base-cased，下载地址：xlnet-base-cased at main (huggingface.co)

将上图画框的文件下载下来，如果下载后的名字和上面显示的名字不一样，则要修改回来。

将下载好的文件放入xlnet-base-chinese文件夹中。

配置环境

需要下载transformers和sentencepiece，执行命令：

conda install sentencepiece
conda install transformers

训练、验证和预测

训练详见train_model函数，验证详见test_model，单次预测详见predict函数。

代码和模型链接：
https://download.csdn.net/download/hhhhhhhhhhwwwwwwwwww/36194843