Reward-based Input Construction for Cross-document Relation Extraction

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Na, Byeonghu, Jo, Suhyeon, Kim, Yeongmin, Moon, Il-Chul
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914817609039872
author Na, Byeonghu
Jo, Suhyeon
Kim, Yeongmin
Moon, Il-Chul
author_facet Na, Byeonghu
Jo, Suhyeon
Kim, Yeongmin
Moon, Il-Chul
contents Relation extraction (RE) is a fundamental task in natural language processing, aiming to identify relations between target entities in text. While many RE methods are designed for a single sentence or document, cross-document RE has emerged to address relations across multiple long documents. Given the nature of long documents in cross-document RE, extracting document embeddings is challenging due to the length constraints of pre-trained language models. Therefore, we propose REward-based Input Construction (REIC), the first learning-based sentence selector for cross-document RE. REIC extracts sentences based on relational evidence, enabling the RE module to effectively infer relations. Since supervision of evidence sentences is generally unavailable, we train REIC using reinforcement learning with RE prediction scores as rewards. Experimental results demonstrate the superiority of our method over heuristic methods for different RE structures and backbones in cross-document RE. Our code is publicly available at https://github.com/aailabkaist/REIC.
format Preprint
id arxiv_https___arxiv_org_abs_2405_20649
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Reward-based Input Construction for Cross-document Relation Extraction
Na, Byeonghu
Jo, Suhyeon
Kim, Yeongmin
Moon, Il-Chul
Computation and Language
Machine Learning
Relation extraction (RE) is a fundamental task in natural language processing, aiming to identify relations between target entities in text. While many RE methods are designed for a single sentence or document, cross-document RE has emerged to address relations across multiple long documents. Given the nature of long documents in cross-document RE, extracting document embeddings is challenging due to the length constraints of pre-trained language models. Therefore, we propose REward-based Input Construction (REIC), the first learning-based sentence selector for cross-document RE. REIC extracts sentences based on relational evidence, enabling the RE module to effectively infer relations. Since supervision of evidence sentences is generally unavailable, we train REIC using reinforcement learning with RE prediction scores as rewards. Experimental results demonstrate the superiority of our method over heuristic methods for different RE structures and backbones in cross-document RE. Our code is publicly available at https://github.com/aailabkaist/REIC.
title Reward-based Input Construction for Cross-document Relation Extraction
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2405.20649