Code Completion by Modeling Flattened Abstract Syntax Trees as Graphs

Authors

  • Yanlin Wang Microsoft Research Asia
  • Hui Li Xiamen University

Keywords:

Generation, Software Engineering

Abstract

Code completion has become an essential component of integrated development environments. Contemporary code completion methods rely on the abstract syntax tree (AST) to generate syntactically correct code. However, they cannot fully capture the sequential and repetitive patterns of writing code and the structural information of the AST. To alleviate these problems, we propose a new code completion approach named CCAG, which models the flattened sequence of a partial AST as an AST graph. CCAG uses our proposed AST Graph Attention Block to capture different dependencies in the AST graph for representation learning in code completion. The sub-tasks of code completion are optimized via multi-task learning in CCAG, and the task balance is automatically achieved using uncertainty without the need to tune task weights. The experimental results show that CCAG has superior performance than state-of-the-art approaches and it is able to provide intelligent code completion.

Downloads

Published

2021-05-18

How to Cite

Wang, Y., & Li, H. (2021). Code Completion by Modeling Flattened Abstract Syntax Trees as Graphs. Proceedings of the AAAI Conference on Artificial Intelligence, 35(16), 14015-14023. Retrieved from https://ojs.aaai.org/index.php/AAAI/article/view/17650

Issue

Section

AAAI Technical Track on Speech and Natural Language Processing III