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

Hierarchical Attentional Hybrid Neural Networks for Document Classification

About

Document classification is a challenging task with important applications. The deep learning approaches to the problem have gained much attention recently. Despite the progress, the proposed models do not incorporate the knowledge of the document structure in the architecture efficiently and not take into account the contexting importance of words and sentences. In this paper, we propose a new approach based on a combination of convolutional neural networks, gated recurrent units, and attention mechanisms for document classification tasks. The main contribution of this work is the use of convolution layers to extract more meaningful, generalizable and abstract features by the hierarchical representation. The proposed method in this paper improves the results of the current attention-based approaches for document classification.

Jader Abreu, Luis Fred, David Mac\^edo, Cleber Zanchettin• 2019

Related benchmarks

TaskDatasetResultRank
Document ClassificationIMDB (test)
Accuracy95.17
10
Document ClassificationYelp 2018 (test)
Accuracy73.28
5
Showing 2 of 2 rows

Other info

Code

Follow for update