← Search

Kechi Zhang

13 accepted papers

2026

Detecting Data Contamination from Reinforcement Learning Post-training for Large Language Models

ICLR 2026poster

Data contamination poses a significant threat to the reliable evaluation of Large Language Models (LLMs). This issue arises when benchmark samples may inadvertently appear in training sets, compromising the validity of reported performance. While detection methods have been developed for the pre-tra…

Cited by 0SourcecodeScholar
2026

Weights to Code: Extracting Interpretable Algorithms from the Discrete Transformer

ICML 2026poster

Algorithm extraction aims to synthesize executable programs directly from models trained on algorithmic tasks, enabling *de novo* algorithm discovery without relying on human-written code. However, applying this paradigm to Transformer is hindered by representation entanglement (e.g., superposition)…

Cited by 0SourceScholar
2025

Benchmarking Long-Context Language Models on Long Code Understanding

ACL 2025long

Current advanced long-context language models offer great potential for real-world software engineering applications. However, progress in this critical domain remains hampered by a fundamental limitation: the absence of a rigorous evaluation framework for long code understanding. To gap this obstac…

Cited by 0SourcePDFScholar
2025

CodeDPO: Aligning Code Models with Self Generated and Verified Source Code

ACL 2025long

Code generation models have shown significant potential for programming tasks. However, existing training methods like supervised fine-tuning face key limitations: they do not effectively teach models to prioritize correct over incorrect solutions in ambiguous situations, nor do they effectively opt…

Cited by 0SourcePDFScholar
2025

Focused-DPO: Enhancing Code Generation Through Focused Preference Optimization on Error-Prone Points

ACL 2025finding

Code generation models have shown significant potential for automating programming tasks. However, the challenge of generating accurate and reliable code persists due to the highly complex and long-reasoning nature of the task. Even state-of-the-art models often fail in code generation due to small…

Cited by 0SourcePDFScholar
2025

Improving Formal Reasoning of Transformer with State Stack

NeurIPS 2025poster

The Transformer architecture has emerged as a landmark advancement within the broad field of artificial intelligence, effectively catalyzing the advent of large language models (LLMs). However, despite its remarkable capabilities and the substantial progress it has facilitated, the Transformer archi…

Cited by 0SourceScholar
2025

Reasoning is Periodicity? Improving Large Language Models Through Effective Periodicity Modeling

NeurIPS 2025poster

Periodicity, as one of the most important basic characteristics, lays the foundation for facilitating structured knowledge acquisition and systematic cognitive processes within human learning paradigms. However, the potential flaws of periodicity modeling in Transformer affect the learning efficienc…

Cited by 0SourceScholar
2025

Revisit Self-Debugging with Self-Generated Tests for Code Generation

ACL 2025long

Large language models (LLMs) have demonstrated significant advancements in code generation, yet they still face challenges when tackling tasks that extend beyond their basic capabilities. Recently, the concept of self-debugging has been proposed as a way to enhance code generation performance by lev…

Cited by 0SourcePDFScholar
2025

SATURN: SAT-based Reinforcement Learning to Unleash LLMs Reasoning

NeurIPS 2025spotlight

How to design reinforcement learning (RL) tasks that effectively unleash the reasoning capability of large language models (LLMs) remains an open question. Existing RL tasks (e.g., math, programming, and constructing reasoning tasks) suffer from three key limitations: (1) Scalability. They rely heav…

Cited by 0SourceScholar
2024

CodeAgent: Enhancing Code Generation with Tool-Integrated Agent Systems for Real-World Repo-level Coding Challenges

ACL 2024long

Large Language Models (LLMs) have shown promise in automated code generation but typically excel only in simpler tasks such as generating standalone code units. However, real-world software development often involves complex code repositories with complex dependencies and extensive documentation. To…

2024

HiRoPE: Length Extrapolation for Code Models Using Hierarchical Position

ACL 2024long

Addressing the limitation of context length in large language models for code-related tasks is the primary focus of this paper. Existing LLMs are constrained by their pre-trained context lengths, leading to performance issues in handling long complex code sequences. Inspired by how human programmers…