A Partition Filter Network for Joint Entity and Relation Extraction

In joint entity and relation extraction, existing work either sequentially encode task-specific features, leading to an imbalance in inter-task feature interaction where features extracted later have no direct contact with those that come first. Or they encode entity features and relation features in a parallel manner, meaning that feature representation learning for each task is largely independent of each other except for input sharing. We propose a partition filter network to model two-way interaction between tasks properly, where feature encoding is decomposed into two steps: partition and filter. In our encoder, we leverage two gates: entity and relation gate, to segment neurons into two task partitions and one shared partition. The shared partition represents inter-task information valuable to both tasks and is evenly shared across two tasks to ensure proper two-way interaction. The task partitions represent intra-task information and are formed through concerted efforts of both gates, making sure that encoding of task-specific features is dependent upon each other. Experiment results on six public datasets show that our model performs significantly better than previous approaches. In addition, contrary to what previous work has claimed, our auxiliary experiments suggest that relation prediction is contributory to named entity prediction in a non-negligible way. The source code can be found at https://github.com/Coopercoppers/PFN.

PDF Abstract EMNLP 2021 PDF EMNLP 2021 Abstract

Results from the Paper

Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Relation Extraction ACE 2004 PFN NER Micro F1 89.3 # 3
RE+ Micro F1 62.5 # 2
Cross Sentence No # 1
Relation Extraction ACE 2005 PFN NER Micro F1 89.0 # 6
RE+ Micro F1 66.8 # 4
Sentence Encoder ALBERT # 1
Cross Sentence No # 1
Relation Extraction ADE Corpus PFN NER Macro F1 91.3 # 1
RE+ Macro F1 83.2 # 1
Relation Extraction Adverse Drug Events (ADE) Corpus PFN (ALBERT XXL, no aggregation) RE+ Macro F1 83.2 # 4
NER Macro F1 91.3 # 3
Relation Extraction NYT PFN F1 92.4 # 7
NER Micro F1 95.8 # 1
Joint Entity and Relation Extraction SciERC PFN Entity F1 66.8 # 9
RE+ Micro F1 38.4 # 2
Cross Sentence No # 1
Relation Extraction SciERC PFN RE+ Micro F1 38.4 # 1
NER Micro F1 66.8 # 1
Relation Extraction WebNLG PFN F1 93.6 # 2
NER Micro F1 98.0 # 1