Learning program semantics for vulnerability detection via vulnerability-specific inter-procedural slicing

• Main Authors: WU, Bozhi, LIU, Shangqing, YANG, Xiao, LI, Zhiming, SUN, Jun, LIN, Shang-Wei
• Format: text
• Language: English
• Published: Institutional Knowledge at Singapore Management University 2023
• Subjects: code representations; program semantics; Vulnerability detection; Artificial Intelligence and Robotics; Information Security; Theory and Algorithms
• Online Access: https://ink.library.smu.edu.sg/sis_research/8578; https://ink.library.smu.edu.sg/context/sis_research/article/9581/viewcontent/LearningProgamSemantics_pvoa_cc_by.pdf
• Institution: Singapore Management University

Learning-based approaches that learn code representations for software vulnerability detection have been proven to produce inspiring results. However, they still fail to capture complete and precise vulnerability semantics for code representations. To address the limitations, in this work, we propose a learning-based approach namely SnapVuln, which first utilizes multiple vulnerability-specific inter-procedural slicing algorithms to capture vulnerability semantics of various types and then employs a Gated Graph Neural Network (GGNN) with an attention mechanism to learn vulnerability semantics. We compare SnapVuln with state-of-the-art learning-based approaches on two public datasets, and confirm that SnapVuln outperforms them. We further perform an ablation study and demonstrate that the completeness and precision of vulnerability semantics captured by SnapVuln contribute to the performance improvement.