(19)中华 人民共和国 国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111659465.4 (22)申请日 2021.12.31 (71)申请人 杭州趣链科技有限公司 地址 310000 浙江省杭州市西湖区文三路 199号13幢南楼5 01室 (72)发明人 黄方蕾　张珂杰　胡麦芳　尚璇　 张延楠　 (74)专利代理 机构 北京超凡宏宇专利代理事务 所(特殊普通 合伙) 11463 代理人 张萍 (51)Int.Cl. G06F 21/57(2013.01) G06N 3/04(2006.01) G06N 3/08(2006.01) (54)发明名称 基于深度学习的智能合约漏洞检测方法和 装置 (57)摘要 本发明提供了一种基于深度学习的智能合 约漏洞检测方法和装置， 涉及漏洞检测技术领 域， 该方法包括： 首先对待检测的智 能合约代码 进行处理， 生成待测代码； 然后根据待测代码生 成待测异构网络图； 再利用训练好的神经网络模 型对待测异构网络图进行漏洞检测， 生成检测结 果， 通过构建异构图利用模型进行漏洞检测， 缓 解了现有技术中存在的准确性低、 灵活性低的技 术问题， 达到了提高检测结果准确性、 扩大适用 范围的技 术效果。 权利要求书2页 说明书9页 附图2页 CN 114297665 A 2022.04.08 CN 114297665 A 1.一种基于深度学习的智能合约漏洞检测方法， 其特 征在于， 包括： 对待检测的智能合约代码进行处 理， 生成待测代码； 根据所述待测代码生成待测 异构网络图； 所述待测 异构网络图包括多个第 一节点以及 连接所述第一节点的第一 边； 所述第一节点的信息包括所述待测代码的变量重要性指标； 利用训练好的神经网络模型对所述待测异构网络图进行漏洞检测， 生成检测结果。 2.根据权利要求1所述的方法， 其特征在于， 所述变量重要性指标包括： 系统内在定义 函数、 代码自定义 函数和代码自变量。 3.根据权利要求2所述的方法， 其特征在于， 所述第 一边的信 息包括： 边类型和边特征； 所述边类型包括控制边和数据边； 所述 边特征用于表示所述 边是否存在漏洞合约。 4.根据权利要求1所述的方法， 其特征在于， 对待检测的智能合约代码进行处理， 生成 初始待测代码的步骤之前， 还 包括： 对预先获取的智能合约代码进行处 理， 生成训练代码； 根据所述训练代码生成训练异构网络图； 所述训练异构网络图包括多个第 二节点以及 连接所述第二节点的第二 边； 所述第二节点的信息包括所述训练代码的变量重要性指标； 利用所述训练异构网络图对待训练 的神经网络模型进行训练， 得到训练后的神经网络 模型。 5.根据权利要求4所述的方法， 其特征在于， 根据 所述训练代码生成训练异构网络图的 步骤， 包括： 根据所述训练代码生成合约代码逻辑图G和权 重更新图G ’； 以三元组子图的形式并在满足节点类型和边特征一致的条件下， 对所述权重更新图G ’ 的三元组子图与所述 合约代码逻辑图G进行匹配合并， 生成训练异构网络图。 6.根据权利要求5所述的方法， 其特征在于， 利用所述训练异构网络图对待训练 的神经 网络模型进行训练的步骤， 包括： 对所述权 重更新图G ′进行边特 征的缩放处 理； 在所述训练异构网络图中提取含有相同边特征类型的智能合约的漏洞 子图， 并计算每 条边存在漏洞的概 率。 7.根据权利要求1所述的方法， 其特征在于， 利用训练好的神经网络模型对所述待测 异 构网络图进行漏洞检测， 生成检测结果， 包括： 确定所述待测 异构网络图中每条边的漏洞特征信 息； 所述漏洞特征信 息包括存在漏洞 的概率； 基于所述漏洞特征信 息， 确定目标边序列； 所述目标边序列包括多条目标边， 所述目标 边满足预 先确定的存在漏洞的概 率； 将所述目标边序列输入训练好的神经网络模型中， 生成检测结果。 8.一种基于深度学习的智能合约漏洞检测装置， 其特 征在于， 包括： 第一生成模块， 用于对待检测的智能合约代码进行处 理， 生成待测代码； 第二生成模块， 用于根据所述待测代码生成待测异构网络 图； 所述待测异构网络 图包 括多个第一节点以及连接所述第一节点的第一边； 所述第一节点的信息包括所述待测代码 的变量重要性指标； 检测模块， 用于利用训练好的神经网络模型对所述待测异构网络 图进行漏洞检测， 生权　利　要　求　书 1/2 页 2 CN 114297665 A 2成检测结果。 9.一种电子设备， 包括存储器、 处理器， 所述存储器中存储有可在所述处理器上运行的 计算机程序， 其特征在于， 所述处理器执行所述计算机程序时实现上述权利要求1至7任一 项所述的方法的步骤。 10.一种计算机可读存储介质， 其特征在于， 所述计算机可读存储介质存储有机器可运 行指令， 所述计算机可运行指令在被处理器调用和运行时， 所述计算机可运行指令促使所 述处理器运行 所述权利要求1至7任一项所述的方法。权　利　要　求　书 2/2 页 3 CN 114297665 A 3