HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Investigating Capsule Networks with Dynamic Routing for Text Classification

Wei Zhao; Jianbo Ye; Min Yang; Zeyang Lei; Suofei Zhang; Zhou Zhao

Investigating Capsule Networks with Dynamic Routing for Text Classification

Abstract

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.

Code Repositories

abhishek-924/Capsule-Network
Mentioned in GitHub
andyweizhao/capsule_text_classification
Official
tf
Mentioned in GitHub
andyweizhao/capsule
tf
Mentioned in GitHub
kevindeangeli/capsuleNetwork
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
sentiment-analysis-on-crCapsule-B
Accuracy: 85.1
sentiment-analysis-on-mrCapsule-B
Accuracy: 82.3
sentiment-analysis-on-sst-2-binaryCapsule-B
Accuracy: 86.8
subjectivity-analysis-on-subjCapsule-B
Accuracy: 93.8
text-classification-on-ag-newsCapsule-B
Error: 7.4
text-classification-on-trec-6Capsule-B
Error: 7.2

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp