TextCNN：文本分类卷积神经网络

news2025/1/12 1:49:19

模型原理

1、前言
2、模型结构
3、示例
- - 3.1、词向量层
  - 3.2、卷积层
  - 3.3、最大池化层
  - 3.4、Fully Connected层
- 4、总结

1、前言

TextCNN 来源于《Convolutional Neural Networks for Sentence Classification》发表于2014年，是一个经典的模型，Yoon Kim将卷积神经网络CNN应用到文本分类任务，利用多个不同size的kernel来提取句子中的关键信息（类似统计语言模型的N-Gram），从而能够更好地捕捉局部相关性。TextCNN的核心思想是将卷积神经网络 (CNN) 应用到文本分类中，从而提取文本特征。本文将逐步对TextCNN做详细介绍。

2、模型结构

在这里插入图片描述

Embedding层：这一层将输入的自然语言文本编码成分布式表示，可以使用预训练好的词向量，如word2vec，或者直接在训练过程中训练出一套词向量。对于没有出现在训练好的词向量表中的词，可以采取使用随机初始化为0或者偏小的正数表示。
Convolution层：这一层通过卷积操作提取不同的n-gram特征。输入的文本通过embedding层后，会转变成一个二维矩阵，然后通过卷积核进行卷积操作，得到一个向量。在TextCNN网络中，需要同时使用多个不同类型的kernel，同时每个size的kernel又可以有多个。
Pooling层：这一层通过池化操作，如最大池化，得到更高级别的特征表示。
Fully Connected层：最后，将卷积池化得到的特征向量通过全连接层映射到标签域，并通过Softmax层得到文本属于每一类的概率，取概率最大的类作为文本的标签。

3、示例

在这里插入图片描述

3.1、词向量层

首先通过分词工具将"这是个经典的模型。"这句话分成“这\是\个\经典\的\模型\。”再将其转换为词向量。得到一个输入层为n $\times$ d的矩阵，其中n为句子的词数，d为词的维度。在以上例子为了方便演示词的维度设定为d=5。

词向量的获取一般是通过预训练的词嵌入模型（如Word2Vec）来实现的，它可以将每个单词映射到一个高维空间中的向量，这些向量能够捕捉单词之间的语义关系。

3.2、卷积层

TextCNN与CNN在卷积操作上存在一些差异，主要体现在输入数据的维度和卷积核形状上。具体分析如下：

输入数据维度：CNN通常处理的是二维数据，例如图像，其卷积核是在二维空间内从左到右、从上到下滑动以进行特征提取。而TextCNN处理的是一维的文本数据，其卷积核只在一维空间内滑动（即纵向滑动），这是因为文本数据通常是序列化的单词或字符，不具有二维结构。

卷积核形状：在TextCNN中，卷积核不是正方形的，而是与词向量的维度相等，这意味着卷积核的宽度与词嵌入的维度相同，而长度则代表了n-gram的窗口大小。这与CNN中用于图像处理的正方形卷积核不同，后者的宽度和高度通常是相等的，以便捕捉图像中的局部特征。

在以上例子中使用了3种卷积核分别为2，3，4，每种卷积核为2个，每种卷积核提取2个特征矩阵。

3.3、最大池化层

通过最大池化层分别提取2个更高级别的特征，共计6个特征向量，并将其串联起来。

3.4、Fully Connected层

最后，将卷积池化得到的特征向量通过全连接层映射到标签域，并通过Softmax函数得到文本属于每一类的概率。

4、总结

TextCNN作为一种基于卷积神经网络的文本分类模型，具有以下优缺点：

优点：

网络结构简洁：TextCNN的网络结构相对简单，这使得模型容易理解和实现。
训练速度快：由于网络结构的简单性，TextCNN的训练速度较快，这对于需要快速迭代的场景非常有利。
特征提取能力强：TextCNN能够有效地捕捉文本中的上下文信息，这得益于其卷积层的设计，可以处理不同长度的文本，避免了传统文本分类模型需要对文本进行固定长度截断的问题。
适应性强：通过引入预训练好的词向量，TextCNN即使在网络结构简洁的情况下也能取得不错的效果，在多项数据集上超越了基准模型。

缺点：