Modeling Uncertainty in Neural Relation Extraction

Yu Hong,Yanghua Xiao,Wei Wang,Yunwen Chen
DOI: https://doi.org/10.1007/978-3-031-00129-1_29
2022-01-01
Abstract:Previous work on neural relation extraction mainly focus on point-based methods and ignores uncertainty within the bag, thus making poor predictions when there are insufficient instances of the bag. To solve this problem, in this paper, we propose two density-based methods. Specifically, we assume each bag is a Gaussian distribution and sentences in the bag are drawn from it. We use predicted variance, capturing bag's uncertainty, as well as predicted mean to draw more samples to enrich one-instance bags. We also use predicted variance to vote for good representation and temper the loss. To the best of our knowledge, this is the first paper to model uncertainty in neural relation extraction. Experiment results on NYT-10 show significant improvements over baselines.
What problem does this paper attempt to address?