多头注意力机制的图同构网络智能合约源码漏洞检测

来源：计算机工程与应用杂志2024年第7期北京时间：

作者：师自通, 师智斌, 刘冬明, 雷海卫, 龚晓元

单位：中北大学 计算机科学与技术学院，太原 030051

摘要：针对智能合约源码转化为字节码后部分语法、语义丢失，且现有漏洞检测方法精度低、误报率高，特别是对重入漏洞和时间戳漏洞的检测能力有限等问题，提出一种多头注意力机制的图同构网络智能合约源码漏洞检测方法。使用智能合约源码，结合重入漏洞和时间戳漏洞特点构建图结构并将其规范化；将规范化后的图结构数据投入图同构网络进行迭代训练，利用该网络强大的节点表示和图表示能力进行漏洞检测；在图同构网络的基础上增加多头注意力机制，进一步增强图同构网络的节点表示能力。实验结果显示该方法对重入漏洞和时间戳漏洞检测准确率达到93.08%和92.30%，相较于普通图同构网络方法分别提升1.44和2.00个百分点。证明该方法对相关漏洞的检测能力要优于其他检测工具。

关键词：智能合约,漏洞检测,重入漏洞,时间戳漏洞,图同构网络,多头注意力机制