BiSPN: Generating Entity Set and Relation Set Coherently in One Pass

Yuxin He, Buzhou Tang


Abstract
By modeling the interaction among instances and avoiding error propagation, Set Prediction Networks (SPNs) achieve state-of-the-art performance on the tasks of named entity recognition and relation triple extraction respectively. However, how to jointly extract entities and relation triples via SPNs remains an unexplored problem, where the main challenge is the maintenance of coherence between the predicted entity/relation sets during one-pass generation. In this work, we present Bipartite Set Prediction Network (BiSPN), a novel joint entity-relation extraction model that can efficiently generate entity set and relation set in parallel. To overcome the challenge of coherence, BiSPN is equipped with a novel bipartite consistency loss as well as an entity-relation linking loss during training. Experiments on three biomedical/clinical datasets and a general-domain dataset show that BiSPN achieves new state of the art in knowledge-intensive scene and performs competitively in general-domain, while being more efficient than two-stage joint extraction methods.
Anthology ID:
2023.findings-emnlp.136
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:
2066–2077
Language:
URL:
https://aclanthology.org/2023.findings-emnlp.136
DOI:
10.18653/v1/2023.findings-emnlp.136
Bibkey:
Cite (ACL):
Yuxin He and Buzhou Tang. 2023. BiSPN: Generating Entity Set and Relation Set Coherently in One Pass. In Findings of the Association for Computational Linguistics: EMNLP 2023, pages 2066–2077, Singapore. Association for Computational Linguistics.
Cite (Informal):
BiSPN: Generating Entity Set and Relation Set Coherently in One Pass (He & Tang, Findings 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.findings-emnlp.136.pdf