一句话:提出CIVP框架,利用LLM语义嵌入、指令执行图与增强GraphSAGE,实现无需重训即可预测未见过程序的指令级脆弱性。
硬件或环境扰动导致的位翻转正被视为一种威胁程序可靠性的指令级脆弱性,但现有方法通常局限于单个程序,难以泛化到未见代码且需重新训练,实用性和响应速度受限。为此,论文提出CIVP框架,用于上下文感知的指令脆弱性预测,并能无需重训练地泛化到未见程序。CIVP首先利用大语言模型提取指令的语义嵌入,再构建包含复杂程序执行关系的指令执行图以刻画错误传播路径,并通过多跳扩散增强GraphSAGE以捕获跨程序结构模式与上下文依赖,同时采用伪标签提升对脆弱指令的泛化能力。在26个真实程序数据集上的实验表明,CIVP显著优于现有方法,召回率最高提升20.5%,F1分数最高提升18.5%,且能有效泛化到未见程序,为部署前的指令级主动加固提供了高效可扩展的方案。
Discovering potential vulnerabilities has long been a fundamental goal in software security. Among them, bit flips, caused by hardware or environmental disturbances, are increasingly recognized as a new type of vulnerabilities that threaten program reliability at the instruction level. However, existing work is often restricted to individual programs and requires retraining when applied to unseen code, severely limiting their practicality and responsiveness. In this paper, we propose CIVP, a novel framework for context-aware instruction vulnerability prediction, generalizing to unseen programs without retraining. Specifically, to capture the rich contextual semantics of instructions, CIVP first leverages Large Language Models (LLMs) to accurately extract semantic embeddings of instructions. Then, CIVP further constructs an instruction execution graph containing complex relations of program execution, which implicates the potential path of error propagation. To improve instruction representation for vulnerability prediction, CIVP enhances GraphSAGE with multi-hop diffusion to capture inter-program structural patterns and contextual dependencies, and adopts pseudo-labeling to improve the model’s generalization for vulnerable instructions. Extensive experiments on a dataset of 26 real-world programs demonstrate that CIVP significantly outperforms the state-of-the-art approaches, achieving up to 20.5%↑ Recall and 18.5%↑ F1-score improvements. Notably, CIVP generalizes well to unseen programs, offering an efficient and scalable solution for proactive instruction-level hardening before software deployment.
DOI 原文 ·
@inproceedings{paperbot9214,
title = {Towards Generalizable Instruction Vulnerability Prediction via LLM-Enhanced Code Representation},
author = {Bao Wen and Jingjing Gu and Jingxuan Zhang and Yang Liu and Pengfei Yu and Yanchao Zhao},
booktitle = {ASE},
year = {2025},
doi = {10.1109/ase63991.2025.00177}
}