基于Bert-BiLSTM-CRF模型的电子病历隐私信息识别方法*

2022-05-13 13:35陈逸墨叶辉易珺周华文方丹丹曹东
自动化与信息工程 2022年2期
关键词:病历正确率实体

陈逸墨 叶辉 易珺 周华文 方丹丹 曹东

基于Bert-BiLSTM-CRF模型的电子病历隐私信息识别方法*

陈逸墨1叶辉1易珺2周华文1方丹丹1曹东1

(1.广州中医药大学医学信息工程学院,广东 广州 510006 2.广东药科大学医药信息工程学院,广东 广州 510006)

随着电子病历数据开放共享的需求越来越大,电子病历去隐私性问题亟需解决。利用自然语言处理技术,提出一种基于Bert-BiLSTM-CRF模型的电子病历隐私信息识别方法。采用某三甲中医院的电子病历作为数据来源,结合当前公开的数据集进行训练,得到正确率为94.02%、召回率为94.25%、1为93.98%的中医电子病历隐私信息识别模型。与其他传统模型进行对比实验表明,Bert-BiLSTM-CRF模型能有效识别并保护电子病历中的隐私数据,有助于医疗数据的开放共享。

隐私信息;Bert;双向长短时记忆网络;条件随机场;电子病历

0 引言

随着信息时代的到来,各行各业开始与互联网、信息技术交融并飞速发展。为加快我国医学领域的发展进程,卫生部发布了《电子病历基本规范(试行)》。电子病历也称计算机化的病案系统,是用电子设备保存、管理、传输和重现的数字化医疗记录[1]。电子病历在提高信息交流效率的同时也面临诸多挑战,其中如何有效识别并隐藏患者的隐私信息成为关键问题。

目前,中文命名实体识别方法主要基于规则、统计机器学习和深度学习等方法[2]。其中,基于规则的方法依赖手工规则,结合命名实体库,通过实体与规则的相符情况进行类型判断。该方法能够得到较好的识别效果,但不同领域的规则各有不同且这些规则不能互用,因此机器学习的方法逐渐兴起。目前,用于中文命名实体识别的机器学习模型主要有隐马尔科夫模型(Hidden Markov model, HMM)、条件随机场(condition random field, CRF)[3]等。随着硬件计算能力的提升,基于深度学习的方法越来越普遍,且效果较基于统计机器学习的方法更胜一筹。目前,基于深

度学习的方法主要通过神经网络来训练模型,主流神经网络模型有卷积神经网络(convolutional neural networks, CNN)[4]、循环神经网络(recurrent neural network, RNN)[5]、长短时记忆神经网络(long short–term memory, LSTM)[6]等。中医电子病历具有复杂性高、词语多义性强、专业性强等特点,传统模型虽然可以实现实体识别功能,但效果不尽如人意。

近年来提出的Bert预训练语言模型,凭借优秀的表意能力,使与之结合的神经网络模型效果更佳。本文提出由Bert、BiLSTM和CRF 三个模块构成的模型对中医电子病历中的隐私信息进行识别。

1 模型原理

Bert-BiLSTM-CRF模型框架如图1所示。

首先,待处理的数据输入Bert进行预训练;然后,BiLSTM层进行语义编码处理;最后,将得到的数据输入CRF层计算最终结果。与传统的基于深度学习方法相比,本文方法引入了Bert预训练语言模型。Bert-BiLSTM-CRF模型是经过大量语料及长时间训练得到的,能根据上下文信息计算出字的向量表示,可有效表现字的多义性,增强句子的语义表示[7]。

图1 Bert-BiLSTM-CRF模型示意图

1.1 Bert预训练语言模型

2015年,DAI和LE首次提出预训练语言模型[8]。2018年DEVLIN等经过改进,提出Bert预训练语言模型[9]。该模型的构成元素为表义能力较强的Trans- former[10]。Transformer是一种基于Attention机制的深度网络,具有良好的并行计算能力且善于捕捉长距离特征,结构如图2所示。

图2 Transformer编码单元

1.2  BiLSTM

1997年,HOCHREITER提出基于RNN改进的LSTM[11]。LSTM模型较于RNN模型具有可利用长距离信息的特点,并解决了RNN模型存在的梯度消失问题。2005年,GRAVES根据LSTM和双向RNN提出双向长短时记忆网络(BiLSTM)[12]。LSTM单元主要由输入门、遗忘门、输出门3部分组成。其中,输入门确定保留信息;遗忘门确定丢弃信息;输出门确定可输出信息,结构图如图3所示。

图3 LSTM单元结构图

1.3 CRF

CRF是LAFFERTY等在2001年提出的一种判别式模型,属于随机场的一种[13]。由于BiLSTM模块的输出结果会出现一些无意义的字符和标签,也没有考虑其间的依赖关系,故通过加入CRF模块从训练数据中获得约束性规则,保证标签是合法的[14]。BiLSTM模块的输出序列作为CRF模块的输入,如句子(x1, x2, ..., xi)通过Bert预训练语言模型和BiLSTM层后,得到每个字的预测序列(1,2, ... ,y),进入CRF后被转换为BIO标注法所定义的标记(1,2, ... ,tag),其中表示标记维度。通过CRF层为标记打分,采用Softmax函数进行归一化,以BIO标注法对标记序列进行规整,完成隐私信息的识别[15]。

2 实验

本实验用来识别中医电子病历中的隐私信息,包括人名、地名、机构名、年龄。实验数据主要来自人民日报语料库和某三甲中医院的电子病历,其中电子病历349份,共11 465 469个字。将人民日报语料库与电子病历中的数据以1∶9的比例分割后作为测试集和训练集。为保证数据整洁,人民日报语料库已标记的数据不做变动,在电子病历数据中新增“年龄”实体类型,用以识别年龄信息。实验主要分为数据预处理、数据导入模型、评判结果3个步骤。

2.1 数据预处理

本文所用数据均采用BIO标注法,标签有9种: B-PER、I-PER、B-LOC、I-LOC、B-ORG、I-ORG、B-AGE、I-AGE、O。其中,B表示实体开始部分;I表示实体非开始部分;O表示非实体;PER表示人名实体;LOC表示地名实体;ORG表示机构实体;AGE表示年龄实体。BIO标签集如表1所示。

表1 BIO标签集

未标记的数据使用自主开发的标注软件进行实体标注,操作界面如图4所示,标注结果如图5所示。其中,C为需要标注的实体;P为实体在文本中的位置;T为实体类型。

图4 标注软件操作界面

图5 标注结果

2.2 实验参数设置

3个模型的常规参数设置如表2所示。其中“Max sequence length”表示字数个数;“epoch”表示时期,一个epoch就是将所有训练样本训练一次的过程;“dropout”表示防止过拟合参数;“Learning rate”表示学习率,合适的学习率可以使目标函数在合适的时间内收敛到局部最小值;“Batch size”表示每批样本的大小;“Max checkpoints”表示训练过程中的最大模型快照。

表2 实验参数

2.3 实验结果

本文以正确率、召回率、和1值作为评判指标。其中,正确率为预测准确样本中真实准确样本的比例;召回率为真实准确样本中预测准确样本的比例;1为正确率和召回率的加权平均值。

实验环境如表3所示。

表3 实验环境

本文实验中所有模型均在上述配置下完成训练。各实体类型的识别结果如表4所示。

表4 3种模型对不同实体类型的识别结果

其中,模型耗时BiLSTM为158.771 min;BiLSTM-CRF为336.951 min;Bert-BiLSTM-CRF为1 718.366 min。

2.4 实验结果分析

训练后得到平均正确率为94.02%、平均召回率为94.25%、平均1为93.98%的中医电子病历隐私信息识别模型。

从模型方面来看:Bert-BiLSTM-CRF模型的平均正确率达到94.02%,在4个实体类型上的识别效果都优于BiLSTM模型和BiLSTM-CRF模型;由此可见,Bert-BiLSTM-CRF模型比传统的LSTM模型效果更好。

从实体类型来看:PER和LOC的识别效果较好,特别是Bert-BiLSTM-CRF模型对这2种实体类型识别的1值均超过了0.95,这是由于人民日报语料中人名和地名的标注质量较高且这些实体不会因其他客观因素而改变;ORG和AGE的识别效果较差,主要原因是ORG有时用缩略词或组合词对识别产生干扰,如“广州中医药大学”被缩略为“广中医”、“中山大学第三附属医院”被缩略为“中大三附院”;年龄实体由数字组成,而病例中存在其他与年龄无关的数字,导致AGE识别不准确,如“药品剂量50 mg/1日”中“50”被识别成年龄,“日期2011-11-23”中“11”和“23”被识别为年龄,产生信息混淆。

3 结语

本文提出的Bert-BiLSTM-CRF模型已达到可以使用的水平,相比传统的BiLSTM模型和BiLSTM-CRF模型,本文模型识别不同类别隐私信息的能力更强。陈衍旭[16]提出的Bert-BiLSTM-CRF模型的隐私信息识别1值为0.932 9,本文模型在此基础上有一定程度的提升。在之后的工作中,需要丰富数据集并且对模型进行适当改进,以提高模型的识别效率。如明确年龄实体与其他包含数字的实体的分类;通过增加原始数据数量来增加训练量。近年来有融入注意力机制[17]的新模型出现。因此,下一步可以考虑从数据处理和融入新机制入手来提升模型性能。

[1] 中华人民共和国卫生部.电子病历基本规范(试行)[J].中国药房,2010,21(12):1063-1064.

[2] 刘浏,王东波.命名实体识别研究综述[J].情报学报,2018, 37(3):329-340.

[3] 宫义山,段亚奇.基于不同模型的中文命名实体识别方法研究[J].长江信息通信,2021,34(1):84-86.

[4] LECUN Y, BOSER B, DENKER J S, et al. Backpropagation applied to handwritten zip code recognition[J]. Neural compu-tation, 1989,1(4):541-551.

[5] 夏瑜潞.循环神经网络的发展综述[J].电脑知识与技术,2019, 15(21):182-184.

[6] HOCHREITER S, SCHMIDHUBER J. LSTM can solve hard long time lag problems[J]. Advances in neural information processing systems, 1997: 473-479.

[7] 王远志,曹子莹.Bert-BLSTM-CRF模型的中文命名实体识别[J].安庆师范大学学报(自然科学版),2021,27(1):59-65.

[8] DAI A M, LE Q V. Semi-supervised sequence learning[J]. Advances in neural information processing systems, 2015,28: 3079-3087.

[9] DEVLIN J, CHANG M W, LEE K, et al. Bert: Pre-training of deep bidirectional transformers for language understanding[J]. arXiv preprint arXiv:1810.04805, 2018.

[10] VASWANI A, SHAZEER N, PARMAR N. et al. Attention is all you need[C]. Advances in Neural Information Processing Systems, 2017:5998-6008.

[11] HOCHREITER S, SCHMIDHUBER J. Long short-term memory[J]. Neural Computation, 1997, 9(8): 1735-1780.

[12] GRAVES A, SCHMIDHUBER J. Framewise phoneme classification with bidirectional LSTM and other neural network architectures[J]. Neural Networks, 2005, 18(5-6): 602-610.

[13] LAFFERTY J, MCCALLUM A, PEREIRA F. Condition random fields: probabilistic models for segmenting and labeling sequence data[C].Proceedings of the 18th Interna-tional Conference on Machine Learning, 2001, 951:282-289.

[14] 罗熹,夏先运,安莹,等.结合多头自注意力机制与BiLSTM- CRF的中文临床实体识别[J].湖南大学学报(自然科学版),2021,48(4):45-55.

[15] 刘一斌,叶辉,易珺,等.基于朴素贝叶斯和word2vec的中医电子病历文本信息抽取[J].世界科学技术-中医药现代化,2020,22(10):3563-3568.

[16] 陈衍旭.面向临床文本的知识获取与应用[D].哈尔滨工业大学,2019.

[17] 张华丽,康晓东,李博,等.结合注意力机制的Bi-LSTM-CRF中文电子病历命名实体识别[J].计算机应用,2020,40(S1): 98-102.

Private Information Recognition Method of Electronic Medical Records Based on Bert-BiLSTM-CRF Model

CHEN Yimo1YE Hui1YI jun2ZHOU Huawen1FANG Dandan1CAO Dong1

(1.School of Medical Information Engineering, Guangzhou University of Chinese Medicine,Guangzhou 510006, China 2. College of Medical Information Engineering, Guangdong Pharmaceutical University, Guangzhou 510006, China)

With the increasing demand for open sharing of electronic medical record data, the privacy of electronic medical record needs to be solved urgently. Using natural language processing technology, a privacy information recognition method of electronic medical record based on Bert-BiLSTM-CRF model is proposed. Using the electronic medical record of a three-tier traditional Chinese medicine hospital as the data source, combined with the current public data set for training, we get the privacy information recognition model of traditional Chinese medicine electronic medical record with the accuracy rate of 94.02%, the recall rate of 94.25% and1 of 93.98%. Compared with other traditional models, the experiment shows that Bert-BiLSTM-CRF model can effectively identify and protect the private data in EMR, which is conducive to the open sharing of medical data.

privacy information; Bert; BiLSTM; CRF; electronic medical record

TP391.1

A

1674-2605(2022)02-0006-06

10.3969/j.issn.1674-2605.2022.02.006

陈逸墨,男,1997年生,在读研究生,主要研究方向:医学自然语言处理。

叶辉,男,1978年生,硕士,讲师,主要研究方向:医学自然语言处理。

易珺,女,1976年生,硕士,副教授,主要研究方向:医学自然语言处理。

周华文,男,1997年生,在读研究生,主要研究方向:医学自然语言处理。

方丹丹,女,1998年生,在读研究生,主要研究方向:医学自然语言处理。

曹东(通信作者)男,1975年生,博士研究生,教授,主要研究方向:医学自然语言处理、医学信号传感与检测。 E-mail: caodong@ gzucm.edu.cn

基金项目:国家重点研发计划资助(2019YFC1710400):广东省普通高校重点领域专项(2020ZDZX3080)。

陈逸墨,叶辉,易珺,等.基于BERT-BiLSTM-CRF模型的电子病历隐私信息识别方法[J].自动化与信息工程, 2022,43(2):35-40.

CHEN Yimo, YE Hui, YI jun, et al. Private information recognition method of electronic medical records based on Bert-BiLSTM-CRF model[J]. Automation & Information Engineering, 2022,43(2):35-40.

猜你喜欢
病历正确率实体
个性化护理干预对提高住院患者留取痰标本正确率的影响
强迫症病历簿
医院信息化建设下电子病历系统应用探究
“病例”和“病历”
前海自贸区:金融服务实体
实体书店步入复兴期?
基于两种LSTM结构的文本情感分析
两会进行时:紧扣实体经济“钉钉子”
振兴实体经济地方如何“钉钉子”
生意