Skip to main navigation Skip to search Skip to main content

Improving text classification with weighted word embeddings via a multi-channel TextCNN model

  • Xi'an Jiaotong University
  • University of Colorado Boulder

Research output: Contribution to journalArticlepeer-review

269 Scopus citations

Abstract

In recent years, convolutional neural networks (CNNs) have gained considerable attention in text classification because of the remarkable good performance they achieved in various situations. The usual practice is to first perform word embedding (i.e., mapping each word into a word vector), and then employ a CNN to perform classification. To improve classification accuracy, term weighting approaches have been proven to be quite effective. But to the best of our knowledge, almost all these methods assign only one weight to each term (word). Considering the fact that one term generally has different importance in documents with different class labels, we propose in this paper a novel term weighting scheme to be combined with word embeddings to enhance the classification performance of CNNs. In the novel method, multiple weights are assigned to each term and these weights are applied to the word embeddings of the words separately. Subsequently, the transformed features are fed into a multi-channel CNN model to predict the label of the sentence. By comparing the novel method with several other baseline methods with five benchmark data sets, the results manifest that the classification accuracy of the proposed method exceeds that of other methods by an amazing margin. Moreover, the weights assigned by different weighting schemes are also analyzed to get more insights of their working mechanism.

Original languageEnglish
Pages (from-to)366-374
Number of pages9
JournalNeurocomputing
Volume363
DOIs
StatePublished - 21 Oct 2019

Keywords

  • Convolutional neural network
  • Term frequency-inverse document frequency (TF-IDF)
  • Term weighting
  • Text classification
  • Word embedding

Fingerprint

Dive into the research topics of 'Improving text classification with weighted word embeddings via a multi-channel TextCNN model'. Together they form a unique fingerprint.

Cite this