Text Classification using Capsules
arXiv:1808.03976
Abstract
This paper presents an empirical exploration of the use of capsule networks for text classification. While it has been shown that capsule networks are effective for image classification, their validity in the domain of text has not been explored. In this paper, we show that capsule networks indeed have the potential for text classification and that they have several advantages over convolutional neural networks. We further suggest a simple routing method that effectively reduces the computational complexity of dynamic routing. We utilized seven benchmark datasets to demonstrate that capsule networks, along with the proposed routing method provide comparable results.
References in corpus (9)
- TensorFlow: A system for large-scale machine learning
- Character-level Convolutional Networks for Text Classification
- Convolutional Neural Networks for Sentence Classification
- Seeing stars: Exploiting class relationships for sentiment categorization with respect to rating scales
- A Convolutional Neural Network for Modelling Sentences
- "Why Should I Trust You?": Explaining the Predictions of Any Classifier
- Investigating Capsule Networks with Dynamic Routing for Text Classification
- Bi-Directional Block Self-Attention for Fast and Memory-Efficient Sequence Modeling
- Information Aggregation via Dynamic Routing for Sequence Encoding