← Search

Zhengyan Shi

8 accepted papers

2026

Gistify: Codebase-Level Understanding via Runtime Execution

ICLR 2026poster

As coding agents are increasingly deployed in large codebases, the need to automatically design challenging, codebase-level evaluation is central. We propose Gistify, a task where a coding LLM must create a single, minimal, self-contained file that can reproduce a specific functionality of a codebas…

Cited by 0SourceScholar
2025

EventRAG: Enhancing LLM Generation with Event Knowledge Graphs

ACL 2025long

Retrieval-augmented generation (RAG) systems often struggle with narrative-rich documents and event-centric reasoning, particularly when synthesizing information across multiple sources. We present EventRAG, a novel framework that enhances text generation through structured event representations. We…

Cited by 0SourcePDFScholar
2025

Learning to Solve Complex Problems via Dataset Decomposition

NeurIPS 2025poster

Curriculum learning is a class of training strategies that organizes the data being exposed to a model by difficulty, gradually from simpler to more complex examples. This research explores a reverse curriculum generation approach that recursively decomposes complex datasets into simpler, more lear…

Cited by 0SourceScholar
2025

Mixture of Noise for Pre-Trained Model-Based Class-Incremental Learning

NeurIPS 2025poster

Class Incremental Learning (CIL) aims to continuously learn new categories while retaining the knowledge of old ones. Pre-trained models (PTMs) show promising capabilities in CIL. However, existing approaches that apply lightweight fine-tuning to backbones still induce parameter drift, thereby compr…

Cited by 0SourcecodeScholar
2025

RiOT: Efficient Prompt Refinement with Residual Optimization Tree

ACL 2025long

Recent advancements in large language models (LLMs) have highlighted their potential across a variety of tasks, but their performance still heavily relies on the design of effective prompts. Existing methods for automatic prompt optimization face two challenges: lack of diversity, limiting the explo…

2025

SelKD: Selective Knowledge Distillation via Optimal Transport Perspective

ICLR 2025poster

Knowledge Distillation (KD) has been a popular paradigm for training a (smaller) student model from its teacher model. However, little research has been done on the practical scenario where only a subset of the teacher's knowledge needs to be distilled, which we term selective KD (SelKD). This deman…

2025

When Can Proxies Improve the Sample Complexity of Preference Learning?

ICML 2025poster

We address the problem of reward hacking, where maximising a proxy reward does not necessarily increase the true reward. This is a key concern for Large Language Models (LLMs), as they are often fine-tuned on human preferences that may not accurately reflect a true objective. Existing work uses vari…

Cited by 0SourcePDFScholar
2024

Instruction Tuning With Loss Over Instructions

NeurIPS 2024poster

Instruction tuning plays a crucial role in shaping the outputs of language models (LMs) to desired styles. In this work, we propose a simple yet effective method, Instruction Modelling (IM), which trains LMs by applying a loss function to the instruction and prompt part rather than solely to the out…