基于One-shot 学习的网络安全知识抽取方法和装置

摘要

本申请涉及一种基于One‑shot学习的网络安全知识抽取方法和装置。所述方法包括：从互联网中获取目标文本，以及得到分段集合，对分段集合中的段落进行位置编码，采用注意力机制，计算分段集合中每一段落的注意力权重；将每一段落输入预先训练的领域分类模型，得到段落的领域标签；采用攻击指示器提取的方式，提取威胁类型集合和威胁类型的触发词集合；根据领域标签、威胁类型集合以及触发词集合，确定段落的域标记；根据域标记，在知识库中提取候选安全知识集合；根据候选安全知识集合和注意力权重，采用One‑shot学习的方式，得到目标文本的网络安全知识。本方法能够提取安全知识的可靠性。