计算机工程与设计2024,Vol.45Issue(9) :2599-2605.DOI:10.16208/j.issn1000-7024.2024.09.006

面向小样本的威胁情报命名实体识别方法

Named entity recognition method for thread intelligence for small samples

萨仁高娃 邬超慧 张振 张悦
计算机工程与设计2024,Vol.45Issue(9) :2599-2605.DOI:10.16208/j.issn1000-7024.2024.09.006

面向小样本的威胁情报命名实体识别方法

Named entity recognition method for thread intelligence for small samples

萨仁高娃 1邬超慧 1张振 1张悦1
扫码查看

作者信息

  • 1. 内蒙古电力集团有限责任公司 内蒙古电力经济技术研究院,内蒙古呼和浩特 010020
  • 折叠

摘要

为解决威胁情报领域的数据源不足、实体专业性强等问题,提出一种面向小样本的威胁情报命名实体识别模型AbNER.设计基于自注意力机制的隐式特征引导模块,引入prompt模板,融合专业领域的先验知识,结合两者共同完成识别实体.对模型输入层结构进行优化,有效提升编码性能.分析AbNER在通用和电网等两类威胁情报数据上的测试结果,模型在5个全量数据集和3个小样本数据集上均达到最优表现,验证了 AbNER的实体识别优势和小样本能力.

Abstract

To solve the problems of insufficient data sources and strong entity professionalism in the field of threat intelligence,AbNER,a named entity recognition model for threat intelligence for small samples was proposed.An implicit feature guidance module based on the self-attention mechanism was designed,a prompt template was introduced to integrate the prior knowledge of the professional field,and the two were combined to jointly complete the identification of entities.The model input layer struc-ture was optimized to effectively improve the coding performance.The test results of AbNER on two types of threat intelligence data,such as general and power grid,were analyzed.The model achieves the best performance on 5 full data sets and 3 small sample data sets,which verifies the entity recognition advantages and small sample capabilities of AbNER.

关键词

命名实体识别/威胁情报/小样本/自注意力机制/大规模语言模型/提示学习/网络安全

Key words

named entity recognition/threat intelligence/small samples/self-attention mechanisms/large language model/prompt learning/cybersecurity

引用本文复制引用

基金项目

国家自然科学基金项目(59637050)

出版年

2024
计算机工程与设计
中国航天科工集团二院706所

计算机工程与设计

CSTPCD北大核心
影响因子:0.617
ISSN:1000-7024
参考文献量1
段落导航相关论文