On the Robustness of Document-Level Relation Extraction Models to Entity Name Variations

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Meng, Shiao, Hu, Xuming, Liu, Aiwei, Ma, Fukun, Yang, Yawen, Li, Shuang, Wen, Lijie
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913386773610496
author Meng, Shiao
Hu, Xuming
Liu, Aiwei
Ma, Fukun
Yang, Yawen
Li, Shuang
Wen, Lijie
author_facet Meng, Shiao
Hu, Xuming
Liu, Aiwei
Ma, Fukun
Yang, Yawen
Li, Shuang
Wen, Lijie
contents Driven by the demand for cross-sentence and large-scale relation extraction, document-level relation extraction (DocRE) has attracted increasing research interest. Despite the continuous improvement in performance, we find that existing DocRE models which initially perform well may make more mistakes when merely changing the entity names in the document, hindering the generalization to novel entity names. To this end, we systematically investigate the robustness of DocRE models to entity name variations in this work. We first propose a principled pipeline to generate entity-renamed documents by replacing the original entity names with names from Wikidata. By applying the pipeline to DocRED and Re-DocRED datasets, we construct two novel benchmarks named Env-DocRED and Env-Re-DocRED for robustness evaluation. Experimental results show that both three representative DocRE models and two in-context learned large language models consistently lack sufficient robustness to entity name variations, particularly on cross-sentence relation instances and documents with more entities. Finally, we propose an entity variation robust training method which not only improves the robustness of DocRE models but also enhances their understanding and reasoning capabilities. We further verify that the basic idea of this method can be effectively transferred to in-context learning for DocRE as well.
format Preprint
id arxiv_https___arxiv_org_abs_2406_07444
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On the Robustness of Document-Level Relation Extraction Models to Entity Name Variations
Meng, Shiao
Hu, Xuming
Liu, Aiwei
Ma, Fukun
Yang, Yawen
Li, Shuang
Wen, Lijie
Computation and Language
68T50
I.2.7
Driven by the demand for cross-sentence and large-scale relation extraction, document-level relation extraction (DocRE) has attracted increasing research interest. Despite the continuous improvement in performance, we find that existing DocRE models which initially perform well may make more mistakes when merely changing the entity names in the document, hindering the generalization to novel entity names. To this end, we systematically investigate the robustness of DocRE models to entity name variations in this work. We first propose a principled pipeline to generate entity-renamed documents by replacing the original entity names with names from Wikidata. By applying the pipeline to DocRED and Re-DocRED datasets, we construct two novel benchmarks named Env-DocRED and Env-Re-DocRED for robustness evaluation. Experimental results show that both three representative DocRE models and two in-context learned large language models consistently lack sufficient robustness to entity name variations, particularly on cross-sentence relation instances and documents with more entities. Finally, we propose an entity variation robust training method which not only improves the robustness of DocRE models but also enhances their understanding and reasoning capabilities. We further verify that the basic idea of this method can be effectively transferred to in-context learning for DocRE as well.
title On the Robustness of Document-Level Relation Extraction Models to Entity Name Variations
topic Computation and Language
68T50
I.2.7
url https://arxiv.org/abs/2406.07444