基于BiLSTM和注意力机制的智能合约漏洞检测方案 |
| |
作者姓名: | 张光华 刘永升 王鹤 于乃文 |
| |
作者单位: | 1. 西安电子科技大学网络与信息安全学院;2. 河北科技大学信息科学与工程学院 |
| |
基金项目: | 国家自然科学基金[U1836210]; |
| |
摘 要: | 针对传统智能合约漏洞检测方案的检测准确率低以及采用深度学习的方案检测漏洞类型单一等问题,文章提出基于双向长短期记忆(Bi-Directional Long Short-Term Memory,BiLSTM)网络和注意力机制的智能合约漏洞检测方案。首先,利用Word2vec词嵌入技术对数据进行训练,通过训练获得操作码词向量表示;然后,通过将词向量传入BiLSTM来提取序列特征,并利用注意力机制为不同的特征赋予不同的权重以突出关键特征;最后,通过激活函数进行归一化处理,实现智能合约漏洞的检测与识别。文章在以太坊上收集了3000个智能合约,并利用这些合约对模型进行实验和评估。实验结果表明,与深度学习模型和传统工具相比,文章所提方案的精确率、召回率和F1分数均有一定提升,能够准确识别出4种类型的智能合约漏洞,准确率达86.34%。
|
关 键 词: | BiLSTM 区块链 智能合约 注意力机制 漏洞检测 |
|