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

Investigating Capsule Networks with Dynamic Routing for Text Classification

About

In this study, we explore capsule networks with dynamic routing for text classification. We propose three strategies to stabilize the dynamic routing process to alleviate the disturbance of some noise capsules which may contain "background" information or have not been successfully trained. A series of experiments are conducted with capsule networks on six text classification benchmarks. Capsule networks achieve state of the art on 4 out of 6 datasets, which shows the effectiveness of capsule networks for text classification. We additionally show that capsule networks exhibit significant improvement when transfer single-label to multi-label text classification over strong baseline methods. To the best of our knowledge, this is the first work that capsule networks have been empirically investigated for text modeling.

Wei Zhao, Jianbo Ye, Min Yang, Zeyang Lei, Suofei Zhang, Zhou Zhao• 2018

Related benchmarks

TaskDatasetResultRank
Subjectivity ClassificationSubj
Accuracy93.8
266
Text ClassificationSST-2 (test)
Accuracy86.8
185
Text ClassificationTREC
Accuracy92.8
179
Sentiment ClassificationCR
Accuracy85.1
142
Text ClassificationSST-2
Accuracy86.8
121
Text ClassificationAGNews
Accuracy92.6
119
Text ClassificationMR
Accuracy82.3
93
Question AnsweringTrecQA clean (test)
MRR70.12
24
Text ClassificationRCV1
Precision@196.63
8
Multi-label Text ClassificationReuters-Multi-label (test)
Error Rate57.2
7
Showing 10 of 11 rows

Other info

Code

Follow for update