← Search

shuai lu

25 accepted papers

2026

Careful Queries, Credible Results: Teaching RAG Models Advanced Web Search Tools with Reinforcement Learning

AAAI 2026technical

Retrieval-Augmented Generation (RAG) enhances large language models (LLMs) by integrating up-to-date external knowledge, yet real-world web environments present unique challenges. These limitations manifest as two key challenges: pervasive misinformation in the web environment, which introduces unre

Cited by 0SourcePDFScholar
2026

Pull Requests as a Training Signal for Repo-Level Code Editing

ICML 2026poster

Repository-level code editing requires models to understand complex dependencies and execute precise multi-file modifications across a large codebase. While recent gains on SWE-bench rely heavily on complex agent scaffolding, it remains unclear how much of this capability can be internalised via hig…

Cited by 0SourceScholar
2025

Alchemy: Amplifying Theorem-Proving Capability Through Symbolic Mutation

ICLR 2025poster

Formal proofs are challenging to write even for experienced experts. Recent progress in Neural Theorem Proving (NTP) shows promise in expediting this process. However, the formal corpora available on the Internet are limited compared to the general text, posing a significant data scarcity challenge…

2025

ArchiSet: Benchmarking Editable and Consistent Single-View 3D Reconstruction of Buildings with Specific Window-to-Wall Ratios

ICCV 2025poster

Image-based 3D Genetation has made significant progress in typical scenarios, achieving high fidelity in capturing intricate textures. However, in the Architecture, Engineering, and Construction (AEC) design stages, existing technologies still face considerable challenges, particularly in handling s…

Cited by 0SourcePDFScholar
2025

Automated Proof Generation for Rust Code via Self-Evolution

ICLR 2025poster

Ensuring correctness is crucial for code generation. Formal verification offers a definitive assurance of correctness, but demands substantial human effort in proof construction and hence raises a pressing need for automation. The primary obsta- cle lies in the severe lack of data—there is much fewe…

Cited by 4SourcePDFScholar
2025

CARD: Cross-modal Agent Framework for Generative and Editable Residential Design

EMNLP 2025

In recent years, architectural design automation has made significant progress, but the complexity of open-world environments continues to make residential design a challenging task, often requiring experienced architects to perform multiple iterations and human-computer interactions. Therefore, ass

Cited by 0SourcePDFScholar
2025

Dinomaly: The Less Is More Philosophy in Multi-Class Unsupervised Anomaly Detection

CVPR 2025poster

Recent studies highlighted a practical setting of unsupervised anomaly detection (UAD) that builds a unified model for multi-class images. Despite various advancements addressing this challenging task, the detection performance under the multi-class setting still lags far behind state-of-the-art cla…

2025

FloorPlan-LLaMa: Aligning Architects’ Feedback and Domain Knowledge in Architectural Floor Plan Generation

ACL 2025long

Floor plans serve as a graphical language through which architects sketch and communicate their design ideas. Actually, in the Architecture, Engineering, and Construction (AEC) design stages, generating floor plans is a complex task requiring domain expertise and alignment with user requirements. Ho…

Cited by 0SourcePDFScholar
2025

MindAligner: Explicit Brain Functional Alignment for Cross-Subject Visual Decoding from Limited fMRI Data

ICML 2025poster

Brain decoding aims to reconstruct visual perception of human subject from fMRI signals, which is crucial for understanding brain's perception mechanisms. Existing methods are confined to the single-subject paradigm due to substantial brain variability, which leads to weak generalization across ind…

Cited by 0SourcePDFScholar
2024

A Retinex Structure-based Low-light Enhancement Model Guided by Spatial Consistency

ICRA 2024poster

Images captured by robotics under low-light conditions are often plagued by several challenges, including diminished contrast, increased noise, loss of fine details, and unnatural color reproduction. These factors can significantly hinder the performance of computer vision tasks such as object detec…

Cited by 11SourceScholar
2024

AGIEval: A Human-Centric Benchmark for Evaluating Foundation Models

NAACL 2024findings

Assessing foundation models’ abilities for human-level tasks is crucial for Artificial General Intelligence (AGI) development.Traditional benchmarks, which rely on artificial datasets, may not accurately represent these capabilities. In this paper, we introduce AGIEval, a novel bilingual benchmark d…

2024

Competition-Level Problems are Effective LLM Evaluators

ACL 2024findings

Large language models (LLMs) have demonstrated impressive reasoning capabilities, yet there is ongoing debate about these abilities and the potential data contamination problem recently. This paper aims to evaluate the reasoning capacities of LLMs, specifically in solving recent competition-level pr…

Cited by 23SourcePDFScholar
2024

Enhancing Large Language Models in Coding Through Multi-Perspective Self-Consistency

ACL 2024long

Large language models (LLMs) have exhibited remarkable ability in code generation. However, generating the correct solution in a single attempt still remains a challenge. Prior works utilize verification properties in software engineering to verify and re-rank solutions in a majority voting manner.…

2023

Code Execution with Pre-trained Language Models

ACL 2023findings

Code execution is a fundamental aspect of programming language semantics that reflects the exact behavior of the code. However, most pre-trained models for code intelligence ignore the execution trace and only rely on source code and syntactic structures. In this paper, we investigate how well pre-t…

2023

Intervention-Based Alignment of Code Search with Execution Feedback

EMNLP 2023long findings

One of the fundamental goals in code search is to retrieve a functionally correct code for a given natural language query. As annotating for correctness requires executing test cases (i.e. obtaining execution feedback), existing code search training datasets approximate text-code co-occurrences as…

Cited by 0SourceScholar
2023

ReContrast: Domain-Specific Anomaly Detection via Contrastive Reconstruction

NeurIPS 2023poster

Most advanced unsupervised anomaly detection (UAD) methods rely on modeling feature representations of frozen encoder networks pre-trained on large-scale datasets, e.g. ImageNet. However, the features extracted from the encoders that are borrowed from natural image domains coincide little with the f…

2022

ReACC: A Retrieval-Augmented Code Completion Framework

ACL 2022long

Code completion, which aims to predict the following code token(s) according to the code context, can improve the productivity of software development. Recent work has proved that statistical language modeling with transformers can greatly improve the performance in the code completion task via lear…

2022

Towards Compositional Generalization in Code Search

EMNLP 2022main

We study compositional generalization, which aims to generalize on unseen combinations of seen structural elements, for code search. Unlike existing approaches of partially pursuing this goal, we study how to extract structural elements, which we name a template that directly targets compositional g…

2022

UniXcoder: Unified Cross-Modal Pre-training for Code Representation

ACL 2022long

Pre-trained models for programming languages have recently demonstrated great success on code intelligence. To support both code-related understanding and generation tasks, recent works attempt to pre-train unified encoder-decoder models. However, such encoder-decoder framework is sub-optimal for au…

2021

CodeXGLUE: A Machine Learning Benchmark Dataset for Code Understanding and Generation

NeurIPS 2021poster

Benchmark datasets have a significant impact on accelerating research in programming language tasks. In this paper, we introduce CodeXGLUE, a benchmark dataset to foster machine learning research for program understanding and generation. CodeXGLUE includes a collection of 10 tasks across 14 datasets…

Cited by 981SourcecodeScholar
2021

GraphCodeBERT: Pre-training Code Representations with Data Flow

ICLR 2021poster

Pre-trained models for programming language have achieved dramatic empirical improvements on a variety of code-related tasks such as code search, code completion, code summarization, etc. However, existing pre-trained models regard a code snippet as a sequence of tokens, while ignoring the inherent…

2021

Long-Range Modeling of Source Code Files with eWASH: Extended Window Access by Syntax Hierarchy

EMNLP 2021main

Statistical language modeling and translation with transformers have found many successful applications in program understanding and generation tasks, setting high benchmarks for tools in modern software development environments. The finite context window of these neural models means, however, that…

2021

WhiteningBERT: An Easy Unsupervised Sentence Embedding Approach

EMNLP 2021finding

Producing the embedding of a sentence in anunsupervised way is valuable to natural language matching and retrieval problems in practice. In this work, we conduct a thorough examination of pretrained model based unsupervised sentence embeddings. We study on fourpretrained models and conduct massive e…

2019

Why Do Neural Dialog Systems Generate Short and Meaningless Replies? a Comparison between Dialog and Translation

ICASSP 2019accepted

This paper addresses the question: In neural dialog systems, why do sequence-to-sequence (Seq2Seq) neural networks generate short and meaningless replies for open-domain response generation? We conjecture that in a dialog system, due to the randomness of spoken language, there may be multiple equall…

Cited by 0SourceScholar