Hierarchical Attentional Hybrid Neural Networks for Document Classification

Code

• github.com/luisfredgs/cnn-hierarchical-network-for-document-classification
  OfficialIn papertf★ 0
• github.com/abhilash1910/MiniAttention
  tf★ 0

Abstract

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.

Tasks

Benchmark Results

Reproductions