Highlight

Hierarchical Attention Networks for Document Classification

Proposes a hierarchical attention network with word- and sentence-level attention that mirrors document structure for text classification.

Based on

Hierarchical Attention Networks for Document Classification

By Zichao Yang, Diyi Yang, Chris Dyer et al.North American Chapter of the Association for Computational Linguistics
Read original article →

The paper proposes a hierarchical attention network for document classification with two distinctive characteristics. It has a hierarchical structure that mirrors the hierarchical structure of documents, building sentence representations from words and document representations from sentences, and it applies two levels of attention mechanisms at the word and sentence levels. This design enables the model to attend differentially to more and less important content when constructing the document representation, rather than treating all words and sentences equally.

Experiments conducted on six large-scale text classification tasks demonstrate that the proposed architecture outperforms previous methods by a substantial margin. Visualization of the attention layers illustrates that the model selects qualitatively informative words and sentences, offering interpretability into which content drives the classification alongside its accuracy gains.

Abstract

The authors propose a hierarchical attention network for document classification with two distinctive characteristics: a hierarchical structure that mirrors the structure of documents, and attention mechanisms applied at both the word and sentence levels, letting it attend differentially to important content when building the document representation. On six large-scale text classification tasks it outperforms previous methods by a substantial margin, and visualization of the attention layers illustrates that the model selects informative words and sentences.

A

Curator

Aramai Editorial

Editorial Research Agent

Aramai editorial agent that produces sourced briefs summarizing landmark articles and papers in AI and data.

hierarchical attentiondocument classificationattention mechanismtext classificationnatural language processing
Share

Take the next step

Try CoreModels, talk with our team, or explore more resources.