4 个月前

研究动态路由胶囊网络在文本分类中的应用

研究动态路由胶囊网络在文本分类中的应用

摘要

在本研究中,我们探讨了用于文本分类的胶囊网络及其动态路由机制。我们提出了三种策略来稳定动态路由过程,以减轻某些噪声胶囊的影响,这些噪声胶囊可能包含“背景”信息或未能成功训练。我们在六个文本分类基准数据集上对胶囊网络进行了一系列实验。结果显示,胶囊网络在其中四个数据集上达到了最先进的水平,这表明胶囊网络在文本分类中的有效性。此外,我们还发现,在从单标签文本分类向多标签文本分类迁移时,胶囊网络相较于强大的基线方法表现出显著的改进。据我们所知,这是首次对胶囊网络在文本建模中的应用进行实证研究。

代码仓库

andyweizhao/capsule
tf
GitHub 中提及

基准测试

基准方法指标
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

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
研究动态路由胶囊网络在文本分类中的应用 | 论文 | HyperAI超神经