Smart Contract Vulnerability Detection Based on Inverted Residual Network and Transfer Learning

Abstract

As the core application of blockchain technology, smart contracts have been widely used in many fields such as finance, supply chain, and copyright management. Smart contracts are prone to various vulnerabilities that attackers can exploit to steal or freeze funds. Traditional vulnerability detection methods rely heavily on complex rules defined by experts, which are difficult to adapt to the explosion of smart contracts. Some recent studies of neural network-based vulnerability detection methods rely on contract source code, and the accuracy of bytecode-level vulnerability detection methods is low. To overcome the limitations of existing methods, we propose CV-IRTL, a new method for smart contract vulnerability detection. Specifically, CV-IRTL designs a vulnerability detection framework for smart contracts based on inverted residual network architecture and transfer learning. In particular, CV-IRTL enables vulnerability detection at the bytecode level, simplifies data preprocessing, utilizes transfer learning to better capture vulnerability characteristics and effectively address dataset imbalances. We have extensively tested CV-IRTL on a dataset containing six vulnerabilities. The experimental results show that the macro average F1-score is 90.75%, and the overall false positive rate is 9.6%, which is better than representative methods in performance.