Beyond Function-Level Search: Repository-Aware Dual-Encoder Code Retrieval with Adversarial Verification
This addresses the need for repository-level code retrieval in software engineering, offering a novel benchmark and method for change request scenarios.
The paper tackles the problem of retrieving contextually relevant code for change requests in complex codebases, introducing a benchmark and a method that achieves a 12.2% improvement in Top-5 Accuracy and 7.1% in Recall over state-of-the-art baselines.
The escalating complexity of modern codebases has intensified the need for retrieval systems capable of interpreting cross-component change intents, a capability fundamentally absent in conventional function-level search paradigms. While recent studies have improved the alignment between natural language queries and code snippets, retrieving contextually relevant code for specific change requests remains largely underexplored. To address this gap, we introduce RepoAlign-Bench, the first benchmark specifically designed to evaluate repository-level code retrieval under change request driven scenarios, encompassing 52k annotated instances. This benchmark shifts the retrieval paradigm from function-centric matching to holistic repository-level reasoning. Furthermore, we propose ReflectCode, an adversarial reflection augmented dual-tower architecture featuring disentangled code_encoder and doc_encoder components. ReflectCode dynamically integrates syntactic patterns, function dependencies, and semantic expansion intents through large language model guided reflection. Comprehensive experiments demonstrate that ReflectCode achieves 12.2% improvement in Top-5 Accuracy and 7.1% in Recall over state-of-the-art baselines, establishing a new direction for context-aware code retrieval.