arXiv · 2204.09191
Unleashing the Power of Compiler Intermediate Representation to Enhance Neural Program Embeddings
Abstract
Neural program embeddings have demonstrated considerable promise in a range of program analysis tasks, including clone identification, program repair, code completion, and program synthesis. However, most existing methods generate neural program embeddings directly from the program source codes, by learning from features such as tokens, abstract syntax trees, and control flow graphs. This paper takes a fresh look at how to improve program embeddings by leveraging compiler intermediate representation (IR). We first demonstrate simple yet highly effective methods for enhancing embedding quality by training embedding models alongside source code and LLVM IR generated by default optimization levels (e.g., -O2). We then introduce IRGen, a framework based on genetic algorithms (GA), to identify (near-)optimal sequences of optimization flags that can significantly improve embedding quality.
Explore related subjects
Keep this discovery
Zongjie Li, Pingchuan Ma, Huaijin Wang, Shuai Wang, Qiyi Tang, Sen Nie, Shi Wu. 2022-04-20. Unleashing the Power of Compiler Intermediate Representation to Enhance Neural Program Embeddings. https://arxiv.org/abs/2204.09191
Cite the original work for its findings. Save a collection to share your selection of sources.