Skip to main navigation Skip to search Skip to main content

HKT-SmartAudit: Distilling Lightweight Models for Smart Contract Auditing

  • Zhiyuan Wei
  • , Jing Sun
  • , Zijian Zhang*
  • , Zhe Hou
  • , Xianhao Zhang
  • , Meng Li*
  • , Yuqiang Sun
  • , Daoyuan Wu
  • , Yang Liu
  • , Chunmiao Li
  • , Mingchao Wan
  • , Jin Dong
  • *Corresponding author for this work
  • Beijing Institute of Technology
  • The University of Auckland
  • Griffith University Queensland
  • Hefei University of Technology
  • Nanyang Technological University
  • Lingnan University
  • Beijing Academy of Blockchain and Edge Computing

Research output: Contribution to journalArticlepeer-review

Abstract

The rapid growth of blockchain technology has driven the widespread adoption of smart contracts; however, their inherent vulnerabilities have led to significant financial losses. Traditional auditing methods, while essential, struggle to keep pace with the increasing complexity and scale of smart contracts. Large language models (LLMs) offer promising capabilities for automating vulnerability detection, but their adoption is often limited by high computational costs. Although prior work has explored leveraging large models through agents or workflows, relatively little attention has been given to improving the performance of smaller, fine-tuned models - a critical factor for achieving both efficiency and data privacy. In this paper, we introduce HKT-SmartAudit, a framework for developing lightweight models optimized for smart contract auditing. It features a multi-stage knowledge distillation pipeline that integrates classical distillation, external domain knowledge, and reward-guided learning to transfer high-quality insights from large teacher models. A single-task learning strategy is employed to train compact student models that maintain high accuracy and robustness while significantly reducing computational overhead. Experimental results show that our distilled models outperform both commercial tools and larger models in detecting complex vulnerabilities and logical flaws, offering a practical, secure, and scalable solution for smart contract auditing. The source code is available in the GitHub repository at https://github.com/LLMSmartAudit/FTSmartAudit

Original languageEnglish
Pages (from-to)4446-4459
Number of pages14
JournalIEEE Transactions on Information Forensics and Security
Volume21
DOIs
Publication statusPublished - 2026
Externally publishedYes

Keywords

  • Blockchain
  • LLMs
  • distillation
  • fine-tuning
  • smart contract auditing

Fingerprint

Dive into the research topics of 'HKT-SmartAudit: Distilling Lightweight Models for Smart Contract Auditing'. Together they form a unique fingerprint.

Cite this