arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2106.01367cs.CRcs.AIcs.LGcs.PLcs.SE
收藏

On using distributed representations of source code for the detection of C security vulnerabilities

INESC-ID(里斯本系统与计算机工程研究所);IST/U.Lisbon(里斯本大学高等技术学院);FEUP(波尔图大学工程学院);Carnegie Mellon University(卡内基梅隆大学)

David Coimbra, Sofia Reis, Rui Abreu, Corina Păsăreanu, Hakan Erdogmus

英文摘要:

This paper presents an evaluation of the code representation model Code2vec when trained on the task of detecting security vulnerabilities in C source code. We leverage the open-source library astminer to extract path-contexts from the abstract syntax trees of a corpus of labeled C functions. Code2vec is trained on the resulting path-contexts with the task of classifying a function as vulnerable or non-vulnerable. Using the CodeXGLUE benchmark, we show that the accuracy of Code2vec for this task is comparable to simple transformer-based methods such as pre-trained RoBERTa, and outperforms more naive NLP-based methods. We achieved an accuracy of 61.43% while maintaining low computational requirements relative to larger models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑