Document-Level Named Entity Recognition with Q-Network

2019 
Named entity recognition (NER) is typically viewed as a sequence labeling problem, where the solution is optimized in sentence level. In this paper, we explore utilizing document-level label consistency to improve NER performance by reinforcement learning. The process entails searching similar mentions and reconciling the labels, which are repeated until sufficient evidence is collected. We employ a Q-network, trained to maximize the total reward that reflects labeling accuracy while penalizing extra effort. On three publicly available datasets, our approach achieves an F\(_1\) of 90.75%, outperforming the ensemble method (89.80%) by 0.95% and the best base tagger (88.16%) by 2.59% (The source code is available at https://github.com/KrisWentaoWong/dqnner/).
    • Correction
    • Source
    • Cite
    • Save
    • Machine Reading By IdeaReader
    27
    References
    1
    Citations
    NaN
    KQI
    []