Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Convolutional Neural Networks for Sentence Classification

About

We report on a series of experiments with convolutional neural networks (CNN) trained on top of pre-trained word vectors for sentence-level classification tasks. We show that a simple CNN with little hyperparameter tuning and static vectors achieves excellent results on multiple benchmarks. Learning task-specific vectors through fine-tuning offers further gains in performance. We additionally propose a simple modification to the architecture to allow for the use of both task-specific and static vectors. The CNN models discussed herein improve upon the state of the art on 4 out of 7 tasks, which include sentiment analysis and question classification.

Yoon Kim• 2014

Related benchmarks

TaskDatasetResultRank
Natural Language InferenceSNLI (test)
Accuracy83.2
681
Document ClassificationRVL-CDIP (test)
Accuracy80.5
306
Subjectivity ClassificationSubj
Accuracy93.4
266
Text ClassificationAG News (test)--
210
Question ClassificationTREC
Accuracy97.32
205
Text ClassificationSST-2 (test)
Accuracy87.2
185
Text ClassificationTREC
Accuracy93.6
179
Sentiment ClassificationSST-2
Accuracy88.1
174
Natural Language InferenceSNLI
Accuracy82.1
174
Sentiment AnalysisSST-5 (test)
Accuracy48
173
Showing 10 of 150 rows
...

Other info

Code

Follow for update