融合多头注意力的中文命名实体识别方法OA
A Chinese Named Entity Recognition Method Integrating Multi-Head Attention
针对现有中文命名实体识别模型特征抽取能力不足,难以捕捉长距离依赖等问题,提出一种融合多头注意力(multi-head attention,MA)和双向门控循环单元(bidirectional gate recurrent unit,BiGRU)的中文命名实体识别方法.首先,使用 ALBERT(a lite BERT)预训练模型生成动态表示向量,并将向量序列输入到 BiGRU 来提取全局语义特征,再利用多头注意力机制捕捉长距离依赖信息来增强语义特征,最后,通过条件随机场(conditional ran-dom field,CRF)解码获得最优序列.结果表明,该方法在《人民日报》和 MSRA 中文数据集上 F1 值均超过 95%,优于其他模型;同时,该方法相比 BERT-BiLSTM-CRF 模型,训练时间减少约 14.5%,证明了模型的有效性和通用性.
To address the limited feature extraction capacity and the difficulty in modeling long-range dependencies in existing Chinese named entity recognition(NER)models,this study proposed a method that integrates multi-head attention(MA)with a bidirectional gated recurrent unit(BiGRU).Specifically,it first used the ALBERT(a lite BERT)pre-trained language model to generate contextualized representations,which are then fed into a BiGRU to extract global semantic features.A multi-head attention mechanism is subsequently applied to capture long-range dependencies and further enhance the semantic representation then the multi-head attention mechanism is used to capture long-distance dependent information to enhance semantic representations.Finally,a conditional random field(CRF)layer decodes the optimal label sequence.Experimental results show that the proposed method achieves F1 scored above 95%on both the People's Daily and MSRA datasets,outperforming competing models.In addition,compared with the BERT-BiLSTM-CRF model,this approach reduces training time by approximately 14.5%,demonstrating its effectiveness and generalizability.
胡德洲;李贯峰;李瑞;王云丽;高文馨
宁夏大学 信息工程学院,宁夏 银川 750021宁夏大学 信息工程学院,宁夏 银川 750021宁夏大学 信息工程学院,宁夏 银川 750021宁夏大学 信息工程学院,宁夏 银川 750021宁夏大学 信息工程学院,宁夏 银川 750021
信息技术与安全科学
命名实体识别多头注意力双向门控循环单元条件随机场
named entity recognitionmulti-head attentionbidirectional gated recurrent unitconditional random field
《西北工程技术学报》 2026 (1)
27-32,6
国家自然科学基金项目(62066038)宁夏自然科学基金项目(2022AAC03026)宁夏大学研究生创新项目(CXXM202356)
评论