Search papers, labs, and topics across Lattice.
This paper investigates the use of Large Language Models (LLMs) for cross-document relation extraction (RE) and finds that they don't consistently outperform smaller models due to the difficulty of classifying among numerous predefined relations. To address this, they propose HCRE, an LLM-based hierarchical classification model that uses a hierarchical relation tree to reduce the number of relation options at each level. HCRE also employs a prediction-then-verification strategy to mitigate error propagation, achieving state-of-the-art results.
LLMs struggle with cross-document relation extraction because of the sheer number of possible relations, but a hierarchical classification approach can unlock their potential.
Cross-document relation extraction (RE) aims to identify relations between the head and tail entities located in different documents. Existing approaches typically adopt the paradigm of ``\textit{Small Language Model (SLM) + Classifier}''. However, the limited language understanding ability of SLMs hinders further improvement of their performance. In this paper, we conduct a preliminary study to explore the performance of Large Language Models (LLMs) in cross-document RE. Despite their extensive parameters, our findings indicate that LLMs do not consistently surpass existing SLMs. Further analysis suggests that the underperformance is largely attributed to the challenges posed by the numerous predefined relations. To overcome this issue, we propose an LLM-based \underline{H}ierarchical \underline{C}lassification model for cross-document \underline{RE} (HCRE), which consists of two core components: 1) an LLM for relation prediction and 2) a \textit{hierarchical relation tree} derived from the predefined relation set. This tree enables the LLM to perform hierarchical classification, where the target relation is inferred level by level. Since the number of child nodes is much smaller than the size of the entire predefined relation set, the hierarchical relation tree significantly reduces the number of relation options that LLM needs to consider during inference. However, hierarchical classification introduces the risk of error propagation across levels. To mitigate this, we propose a \textit{prediction-then-verification} inference strategy that improves prediction reliability through multi-view verification at each level. Extensive experiments show that HCRE outperforms existing baselines, validating its effectiveness.
