GMVD: Smart Contract Vulnerability Detection Based on GAT-Mamba Framework

Smart contracts hold billions of dollars worth of digital currencies, and hacking attacks can not only cause users to lose their assets but also destabilize the blockchain ecosystem.Vulnerability detection in smart contracts remains a major challenge in blockchain security. Existing methods typically rely on a fixed expert mode, which leads to low accuracy. Moreover, GNN-based models fail to effectively differentiate the significance of various interaction information, while transformer models suffer from high computational complexity. To solve this problem, we propose the GAT-Mamba framework, named GMVD, to perform the smart contract vulnerability detection task. The approach first extracts expert-defined vulnerability patterns from smart contract functions. Then, the graph features are extracted by GAT. Finally, Mamba is used to model the high-dimensional vector expression of expert mode features to improve the calculation efficiency of the model. Subsequently, graph features are extracted through GAT, and Mamba is then employed to model the high-dimensional vector representation of expert pattern features, thereby enhancing computational efficiency. Experimental results on three common vulnerabilities, reentrancy, timestamp dependency, and infinite loop, demonstrate that our framework significantly outperforms existing cutting-edge technologies. Specifically, our method achieves 94.29% accuracy in detecting reentrancy, 93.71% in timestamp dependency, and 82.49% in infinite loop detection.