AIRCoder: Adaptive Integration of Multi-dimensional Retrieval for Repository-level Code Completion

Chuanqi Shi, Miao Gao, Zhiqiang Gao


Abstract
Repository-level code completion relies on retrieval strategies to select relevant context from large codebases. Most existing methods employ single-dimensional retrieval based on textual similarity or dependency existence, leading to inconsistent performance across completion scenarios. Even task-adaptive and hybrid methods incur high computational costs while failing to explicitly consider structural hierarchy for retrieving functionally similar code. We introduce **AIRCoder**, a multi-dimensional retrieval framework that combines eight complementary metrics across three dimensions: textual similarity, dependency existence, and structural hierarchy. By proposing a structure-preserving chunking strategy and lightweight fusion module, AIRCoder learns context-dependent weights to adaptively integrate retrieval metrics for each query. Experiments on CrossCodeEval and RepoEval demonstrate that AIRCoder achieves an average improvement of **4.63%** in exact match over the best baseline, with **10.2×** higher efficiency and strong cross-language generalization across Python, Java, C#, and TypeScript.
Anthology ID:
2026.acl-long.1166
Volume:
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
25458–25470
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.acl-long.1166/
DOI:
Bibkey:
Cite (ACL):
Chuanqi Shi, Miao Gao, and Zhiqiang Gao. 2026. AIRCoder: Adaptive Integration of Multi-dimensional Retrieval for Repository-level Code Completion. In Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 25458–25470, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
AIRCoder: Adaptive Integration of Multi-dimensional Retrieval for Repository-level Code Completion (Shi et al., ACL 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.acl-long.1166.pdf
Checklist:
 2026.acl-long.1166.checklist.pdf