Lu Zhang
Author directoryOther people with similar names: Lu Zhang, Lu Zhang, Lu Zhang
Unverified author pages with similar names: Lu Zhang
2026
MD-RE: A Multi-Discrimination Framework for Document-Level Relation Extraction with Adaptive Threshold Shifted Loss
Huangming Xu | Fu Zhang | Lu Zhang | Zhixuan Yang | Jingwei Cheng
Findings of the Association for Computational Linguistics: EMNLP 2026
Huangming Xu | Fu Zhang | Lu Zhang | Zhixuan Yang | Jingwei Cheng
Findings of the Association for Computational Linguistics: EMNLP 2026
Document-level relation extraction (DocRE) aims to identify relations for an entity pair within a document. Existing methods can be broadly classified into two categories: direct encoding of the entire document or enhancement using extracted evidence sentences. However, the former often introduces noise unrelated to relations, while the latter is heavily dependent on the quality of evidence extraction. Moreover, these DocRE models typically use an adaptive threshold to predict all potential relations for an entity pair. As a result, class imbalance in DocRE often leads the model to learn a high threshold for an entity pair, which in turn causes the model to frequently predict that the entity pair has no relation.
To address these issues, we propose a Multi-Discrimination framework (MD-RE) that does not rely on evidence sentences. MD-RE employs three discriminators with dynamically adjusted thresholds to independently predict relations, and aggregates their outputs via a weighted fusion strategy. Furthermore, we propose an Adaptive Threshold Shifted Loss (ATSL), which encourages a lower threshold to alleviate the high false negative rate resulting from class imbalance. Experiments on three datasets demonstrate that MD-RE achieves new state-of-the-art results. In addition, ATSL significantly improves the performance of various existing DocRE models. Moreover, combining other losses with MD-RE also yields competitive results. Code is available at https://github.com/xhm-code/MD-RE.
To address these issues, we propose a Multi-Discrimination framework (MD-RE) that does not rely on evidence sentences. MD-RE employs three discriminators with dynamically adjusted thresholds to independently predict relations, and aggregates their outputs via a weighted fusion strategy. Furthermore, we propose an Adaptive Threshold Shifted Loss (ATSL), which encourages a lower threshold to alleviate the high false negative rate resulting from class imbalance. Experiments on three datasets demonstrate that MD-RE achieves new state-of-the-art results. In addition, ATSL significantly improves the performance of various existing DocRE models. Moreover, combining other losses with MD-RE also yields competitive results. Code is available at https://github.com/xhm-code/MD-RE.
ATGL: An Adaptive-Threshold Global Loss for Document-level Relation Extraction
Huangming Xu | Fu Zhang | Zhixuan Yang | Lu Zhang | Jingwei Cheng
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Huangming Xu | Fu Zhang | Zhixuan Yang | Lu Zhang | Jingwei Cheng
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Document-level relation extraction (DocRE) aims to determine which relations hold between a given entity pair within a document. As a multi-label classification task, the most commonly adopted paradigm introduces a learnable threshold to distinguish positive and negative classes for an entity pair. Under this paradigm, existing losses decouple the optimization into independent positive and negative losses, which interact solely with a shared threshold. This leads to two inherent limitations: (i) threshold instability caused by conflicting gradient updates from the decoupled losses; and (ii) optimization bias exacerbated by the severe imbalance between limited positive samples and abundant negative samples inherent in DocRE, which makes the model more likely to predict that no relation exists.To address these issues, we propose the Adaptive-Threshold Global Loss (ATGL). Unlike prior work, ATGL integrates positive, negative, and threshold optimization into a unified logit space and explicitly enforces ranking constraints on their contributions to the objective. Furthermore, ATGL incorporates an imbalance-aware optimization mechanism, thereby effectively addressing the severe class imbalance in DocRE. Our ATGL serves as a general optimization objective that can be readily applied to different DocRE models. Experiments on four datasets show that ATGL outperforms other DocRE losses and achieves state-of-the-art results, while consistently improving the performance of existing DocRE models. Code is available at https://github.com/xhm-code/ATGL.