← Search

Kexin Pei

10 accepted papers

2026

EXVERUS: Verus Proof Repair via Counterexample Reasoning

ICML 2026poster

Large Language Models (LLMs) have shown promising results in automating formal verification. However, existing approaches treat proof generation as a static, end-to-end prediction over source code, relying on limited verifier feedback and lacking access to concrete program behaviors. We present EXVE…

Cited by 0SourceScholar
2026

SemRep: Code Transformation with Semantics-Preserving Representations

ICML 2026poster

Code transformation is a foundational capability in the software development process, where its effectiveness relies on constructing a high-quality code representation to characterize the input code semantics and guide the transformation. Existing approaches treat code transformation as an end-to-en…

Cited by 0SourceScholar
2025

EditLord: Learning Code Transformation Rules for Code Editing

ICML 2025poster

Code editing is a foundational task in software development, where its effectiveness depends on whether it introduces desired code property changes without changing the original code's intended functionality. Existing approaches often formulate code editing as an implicit end-to-end task, omitting…

Cited by 0SourcePDFScholar
2025

Synthesizing Performance Constraints for Evaluating and Improving Code Efficiency

NeurIPS 2025poster

Large Language Models (LLMs) have been increasingly used to optimize code efficiency. Evaluating their effectiveness and further suggesting optimization opportunities often rely on high-quality tests to demonstrate the performance bottlenecks presented in the program. However, existing approaches re…

Cited by 0SourcecodeScholar
2024

Exploiting Code Symmetries for Learning Program Semantics

ICML 2024spotlight

This paper tackles the challenge of teaching code semantics to Large Language Models (LLMs) for program analysis by incorporating code symmetries into the model architecture. We introduce a group-theoretic framework that defines code symmetries as semantics-preserving transformations, where forming…

Cited by 7SourcePDFScholar
2024

SWE-bench: Can Language Models Resolve Real-world Github Issues?

ICLR 2024oral

Language models have outpaced our ability to evaluate them effectively, but for their future development it is essential to study the frontier of their capabilities. We find real-world software engineering to be a rich, sustainable, and challenging testbed for evaluating the next generation of langu…

Cited by 500SourcePDFScholar
2023

Can Large Language Models Reason about Program Invariants?

ICML 2023poster

Identifying invariants is an important program analysis task with applications towards program understanding, bug finding, vulnerability analysis, and formal verification. Existing tools for identifying program invariants rely on dynamic analysis, requiring traces collected from multiple executions…

Cited by 87SourcePDFScholar
2018

Efficient Formal Safety Analysis of Neural Networks

NeurIPS 2018poster

Neural networks are increasingly deployed in real-world safety-critical domains such as autonomous driving, aircraft collision avoidance, and malware detection. However, these networks have been shown to often mispredict on inputs with minor adversarial or even accidental perturbations. Consequences…