基于提示增强和联合学习的网络威胁情报三元组抽取模型的训练方法及三元组抽取方法

    公开(公告)号:CN119520029A

    公开(公告)日:2025-02-25

    申请号:CN202411481491.6

    申请日:2024-10-22

    Applicant: 广州大学

    Abstract: 本发明公开了基于提示增强和联合学习的网络威胁情报三元组抽取模型的训练方法及三元组抽取方法,包括:S1:构建问题模板;S2:构建解码器;S3:建立由标签答案词向关系标签的第一映射、由关系标签向答案语义词的第二映射;预测实体对的关系为某答案词的概率;S4:输入隐藏层表示和可学习答案语义词的嵌入向量至关系预测层,得到预测头;S5:选择概率最高的答案词根据第一映射映射至关系标签得到预测关系;S6:计算总损失;S7:进行预训练;S8:使用训练数据集训练模型得到三元组联合提取模型。通过输入网络威胁情报数据至所述模型,即可得到关系三元组。本发明采用联合提取框架,有效解决了实体和关系任务间缺乏交互的问题。

    一种网络安全关系抽取方法
    4.
    发明公开

    公开(公告)号:CN119539047A

    公开(公告)日:2025-02-28

    申请号:CN202411482849.7

    申请日:2024-10-22

    Applicant: 广州大学

    Abstract: 本发明公开了一种网络安全关系抽取方法,其中方法步骤如下:S1:获得安全事件数据集,对安全事件数据集中的数据进行清洗、预处理以及切分标记,最后将完成标记的数据集分为训练数据集和测试数据集S2:构建关系抽取预测模型S3:将训练数据集输入关系抽取预测模型,获得安全事件预测结果,根据预测结果对所述关系预抽取模型的参数进行调整S4:将测试数据集数据输入所述关系预抽取模型,得到模型的性能参数,则重复步骤S3至性能参数高于预设值,后将该模型保存为安全关系抽取模型。

Patent Agency Ranking