Papers by Yangruibo Ding
Towards Learning (Dis)-Similarity of Source Code from Program Contrasts (2022.acl-long)
Copied to clipboard
| Challenge: | Existing models that focus on identifying functional (dis)similarity of source code get confused when trying to identify functional (Dis)-similarities. |
| Approach: | They propose to pre-train a Transformer model with such automatically generated program contrasts to better identify similar code in the wild and differentiate vulnerable programs from benign ones. |
| Outcome: | The proposed model outperforms existing models in vulnerability and code clone detection tasks even with much less data. |
CoCoMIC: Code Completion by Jointly Modeling In-file and Cross-file Context (2024.lrec-main)
Copied to clipboard
Yangruibo Ding, Zijian Wang, Wasi U. Ahmad, Murali Krishna Ramanathan, Ramesh Nallapati, Parminder Bhatia, Dan Roth, Bing Xiang
| Challenge: | Pre-trained language models (LMs) for code have shown promising performance in code completion tasks but ignore the rich semantics in other files within the same project. |
| Approach: | They propose a framework that jointly learns the in-file and cross-file context on top of code LMs and a static-analysis-based tool that locates and retrieves the most relevant project-level cross- file context for code completion. |
| Outcome: | The proposed framework improves existing code LMs with a 33.94% relative increase in exact match and 28.69% in identifier matching when the cross-file context is provided. |