Graph-tree Fusion Model with Bidirectional Information Propagation for Long Document Classification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Roy, Sudipta Singha, Wang, Xindi, Mercer, Robert E., Rudzicz, Frank
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910632084766720
author Roy, Sudipta Singha
Wang, Xindi
Mercer, Robert E.
Rudzicz, Frank
author_facet Roy, Sudipta Singha
Wang, Xindi
Mercer, Robert E.
Rudzicz, Frank
contents Long document classification presents challenges in capturing both local and global dependencies due to their extensive content and complex structure. Existing methods often struggle with token limits and fail to adequately model hierarchical relationships within documents. To address these constraints, we propose a novel model leveraging a graph-tree structure. Our approach integrates syntax trees for sentence encodings and document graphs for document encodings, which capture fine-grained syntactic relationships and broader document contexts, respectively. We use Tree Transformers to generate sentence encodings, while a graph attention network models inter- and intra-sentence dependencies. During training, we implement bidirectional information propagation from word-to-sentence-to-document and vice versa, which enriches the contextual representation. Our proposed method enables a comprehensive understanding of content at all hierarchical levels and effectively handles arbitrarily long contexts without token limit constraints. Experimental results demonstrate the effectiveness of our approach in all types of long document classification tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2410_02930
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Graph-tree Fusion Model with Bidirectional Information Propagation for Long Document Classification
Roy, Sudipta Singha
Wang, Xindi
Mercer, Robert E.
Rudzicz, Frank
Computation and Language
Long document classification presents challenges in capturing both local and global dependencies due to their extensive content and complex structure. Existing methods often struggle with token limits and fail to adequately model hierarchical relationships within documents. To address these constraints, we propose a novel model leveraging a graph-tree structure. Our approach integrates syntax trees for sentence encodings and document graphs for document encodings, which capture fine-grained syntactic relationships and broader document contexts, respectively. We use Tree Transformers to generate sentence encodings, while a graph attention network models inter- and intra-sentence dependencies. During training, we implement bidirectional information propagation from word-to-sentence-to-document and vice versa, which enriches the contextual representation. Our proposed method enables a comprehensive understanding of content at all hierarchical levels and effectively handles arbitrarily long contexts without token limit constraints. Experimental results demonstrate the effectiveness of our approach in all types of long document classification tasks.
title Graph-tree Fusion Model with Bidirectional Information Propagation for Long Document Classification
topic Computation and Language
url https://arxiv.org/abs/2410.02930