Fold2Vec: Towards a Statement-Based Representation of Code for Code Comprehension
Francesco Bertolotti, Walter Cazzola · ACM Transactions on Software Engineering and Methodology · 2022
We introduce a novel approach to source code representation to be used in combination with neural networks. Such a representation is designed to permit the production of a continuous vector for each code statement. In particular, we present how the representation is produced in the case of Java source code. We test our representation for three tasks:code summarization,statement separation, andcode search. We compare with the state-of-the-artnon-autoregressiveandend-to-endmodels for these tasks. We conclude that all tasks benefit from the proposed representation to boost their performance in terms of F1-score, accuracy, and mean reciprocal rank, respectively. Moreover, we show how models trained on code summarization and models trained on statement separation can be combined to address methods with tangled responsibilities, meaning that these models can be used to detect code misconduct.