Adaptive Hinge Balance Loss for Document-Level Relation Extraction

Jize Wang, Xinyi Le, Xiaodi Peng, Cailian Chen


Abstract
Document-Level Relation Extraction aims at predicting relations between entities from multiple sentences. A common practice is to select multi-label classification thresholds to decide whether a relation exists between an entity pair. However, in the document-level task, most entity pairs do not express any relations, resulting in a highly imbalanced distribution between positive and negative classes. We argue that the imbalance problem affects threshold selection and may lead to incorrect “no-relation” predictions. In this paper, we propose to down-weight the easy negatives by utilizing a distance between the classification threshold and the predicted score of each relation. Our novel Adaptive Hinge Balance Loss measures the difficulty of each relation class with the distance, putting more focus on hard, misclassified relations, i.e. the minority positive relations. Experiment results on Re-DocRED demonstrate the superiority of our approach over other balancing methods. Source codes are available at https://github.com/Jize-W/HingeABL.
Anthology ID:
2023.findings-emnlp.253
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2023
Month:
December
Year:
2023
Address:
Singapore
Editors:
Houda Bouamor, Juan Pino, Kalika Bali
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
3872–3878
Language:
URL:
https://aclanthology.org/2023.findings-emnlp.253
DOI:
10.18653/v1/2023.findings-emnlp.253
Bibkey:
Cite (ACL):
Jize Wang, Xinyi Le, Xiaodi Peng, and Cailian Chen. 2023. Adaptive Hinge Balance Loss for Document-Level Relation Extraction. In Findings of the Association for Computational Linguistics: EMNLP 2023, pages 3872–3878, Singapore. Association for Computational Linguistics.
Cite (Informal):
Adaptive Hinge Balance Loss for Document-Level Relation Extraction (Wang et al., Findings 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.findings-emnlp.253.pdf