A Toolkit for Generating Code Knowledge Graphs
Ibrahim Abdelaziz, Julian Dolby, et al.
K-CAP 2021
Understanding the functional (dis)-similarity of source code is significant for code modeling tasks such as software vulnerability and code clone detection. We present DISCO (DISsimilarity of COde), a novel self-supervised model focusing on identifying (dis)similar functionalities of source code. Different from existing works, our approach does not require a huge amount of randomly collected datasets. Rather, we design structure-guided code transformation algorithms to generate synthetic code clones and inject real-world security bugs, augmenting the collected datasets in a targeted way. We propose to pre-train the Transformer model with such automatically generated program contrasts to better identify similar code in the wild and differentiate vulnerable programs from benign ones. To better capture the structural features of source code, we propose a new cloze objective to encode the local tree-based context (e.g., parents or sibling nodes). We pre-train our model with a much smaller dataset, the size of which is only 5% of the state-of-the-art models' training datasets, to illustrate the effectiveness of our data augmentation and the pre-training approach. The evaluation shows that, even with much less data, DISCO can still outperform the state-of-the-art models in vulnerability and code clone detection tasks.
Ibrahim Abdelaziz, Julian Dolby, et al.
K-CAP 2021
Yutian Yan, Yunhui Zheng, et al.
ICSE 2023
Michael Muller, April Yi Wang, et al.
IUI 2021
Jasmine Shih, Vishal Mohanty, et al.
CHI 2024