English

XSearch: Explainable Code Search via Concept-to-Code Alignment

Software Engineering 2026-05-18 v1 Artificial Intelligence

Abstract

Semantic code search has been widely adopted in both academia and industry. These approaches embed natural-language queries and code snippets into a shared embedding space and retrieve results based on vector similarity. Despit strong performance on benchmark datasets, they often suffer from poor explainability and generalization. Retrieved code may appear semantically similar yet miss critical functional requirements of the query, while providing no explanation of why the result was retrieved. Moreover, such failures become more severe under distribution shift, where models struggle to generalize to unseen benchmarks. In this work, we propose XSearch, an intrinsically explainable code search framework. Our key insight is that by relying on global embedding similarity, existing retrievers inherently take an inductive view. They learn statistical patterns rather than truly understanding the query's functional requirements. We address this problem by reformulating code search as a deductive concept alignment problem. XSearch (i) identifies functional concepts in the query and (ii) explicitly aligns them with corresponding code statements. This explain-then-predict design produces inherent concept-level explanations and mitigates shortcut learning that harms out-of-distribution generalization. We train an encoder with explicit concept-alignment objectives and perform retrieval through explicit matching between query concepts and code statements. Experiments show that, trained on CodeSearchNet using GraphCodeBERT (125M parameters), XSearch improves performance on out-of-distribution benchmarks from 0.02 to 0.33 (15x) over eight state-of-the-art retrievers, and consistently outperforms both encoder- and decoder-based baselines with up to 7B parameters. A user study demonstrates that concept-alignment explanations enable users to evaluate retrieved results faster and more accurately.

Keywords

Cite

@article{arxiv.2605.16046,
  title  = {XSearch: Explainable Code Search via Concept-to-Code Alignment},
  author = {Yiming Liu and Ruofan Liu and Yun Lin and Zicong Zhang and Weiyu Kong and Pengnian Qi and Xiao Cheng and Weinan Zhang and Qianxiang Wang and Linpeng Huang},
  journal= {arXiv preprint arXiv:2605.16046},
  year   = {2026}
}

Comments

Accepted to ISSTA 2026