Named Entity Recognition for Medical Dialogue Based on BERT and Adversarial Training

Mingzhe Cheng,Hongjun Li,Zelin Yang, Wenjie Fan, Yujin Gan

2022 5th International Conference on Pattern Recognition and Artificial Intelligence (PRAI)(2022)

引用 1|浏览7
暂无评分
摘要
In the medical dialogue named entity recognition, in order to solve the problem that the traditional named entity recognition method is difficult to recognize medical corpus, this paper proposes a BERT-based medical dialogue named entity recognition method BERT-BiLSTM-CRF-ADV. First, we use the BERT pre-training model to obtain word vectors with rich semantic information, then send the word vectors to BiLSTM to extract features, and finally input them into CRF for restriction correction and output. During the training process, we use adversarial training to improve the model performance. This paper applies the proposed method to the medical field to perform named entity recognition on medical dialogue materials. Experimental results show that this method achieves 92.82% of the F1 value in medical dialogue materials, which is greater than the baseline model BERT-CRF.
更多
查看译文
关键词
named entity recognition,knowledge graph,BERT,medical dialogue
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要