← Search

Kaiyan Zhang

27 accepted papers

2026

Attention as a Compass: Efficient Exploration for Process-Supervised RL in Reasoning Models

ICLR 2026poster

Reinforcement Learning (RL) has shown remarkable success in enhancing the reasoning capabilities of Large Language Models (LLMs). Process-Supervised RL (PSRL) has emerged as a more effective paradigm compared to outcome-based RL. However, existing PSRL approaches suffer from limited exploration effi…

Cited by 0SourcecodeScholar
2026

FlowRL: Matching Reward Distributions for LLM Reasoning

ICLR 2026poster

We propose FlowRL: matching the full reward distribution via flow balancing instead of solely maximizing rewards in large language model (LLM) reinforcement learning (RL). Recent advanced reasoning models adopt reward-maximizing methods (e.g., PPO and GRPO), which tend to over-optimize dominant rewa…

Cited by 0SourcecodeScholar
2026

GenPRM: Scaling Test-Time Compute of Process Reward Models via Generative Reasoning

AAAI 2026technical

Recent advancements in Large Language Models (LLMs) have shown that it is promising to utilize Process Reward Models (PRMs) as verifiers to enhance the performance of LLMs. However, current PRMs face three key challenges: (1) limited process supervision and generalization capabilities, (2) dependenc

Cited by 0SourcePDFScholar
2026

How Far Can Unsupervised RLVR Scale LLM Training?

ICLR 2026poster

Unsupervised Reinforcement Learning with Verifiable Rewards (URLVR) offers a pathway for Large Language Models (LLMs) to improve without human supervision. Particularly, many works use model intrinsic information as rewards for URLVR, showing promising improvements, yet their potential and limitatio…

Cited by 0SourceScholar
2026

MARTI: A Framework for Multi-Agent LLM Systems Reinforced Training and Inference

ICLR 2026poster

We present MARTI (Multi-Agent Reinforced Training and Inference), an open-source framework designed to facilitate scalable and efficient learning of multi-agent LLM systems. MARTI supports centralized multi-agent interactions and distributed policy training, with the added capability of multi-turn a…

Cited by 0SourcecodeScholar
2026

SimpleVLA-RL: Scaling VLA Training via Reinforcement Learning

ICLR 2026poster

Vision-Language-Action (VLA) models have emerged as a powerful paradigm for robotic manipulation. Despite substantial progress enabled by large-scale pretraining and supervised fine-tuning (SFT), these models face two fundamental challenges: (i) the scarcity and high cost of large-scale robotic traj…

Cited by 0SourcecodeScholar
2025

AdsQA: Towards Advertisement Video Understanding

ICCV 2025poster

Large language models (LLMs) have taken a great step towards AGI. Meanwhile, an increasing number of domain-specific problems such as math and programming boost these general-purpose models to continuously evolve via learning deeper expertise. Now is thus the time further to extend the diversity of…

2025

Fourier Position Embedding: Enhancing Attention’s Periodic Extension for Length Generalization

ICML 2025poster

Extending the context length of Language Models (LMs) by improving Rotary Position Embedding (RoPE) has become a trend. While prior works mainly address RoPE's limitations within attention, this paper uncovers the adverse effects on length generalization from nearly all parts of LMs. Using *Discrete…

2025

Free Process Rewards without Process Labels

ICML 2025poster

Different from its counterpart outcome reward models (ORMs), which evaluate the entire responses, a process reward model (PRM) scores a reasoning trajectory step by step, providing denser and more fine-grained rewards. However, training a PRM requires labels annotated at every intermediate step, pre…

2025

Fusing Highly Specialized Language Models for Comprehensive Expertise

ACL 2025long

Underlying data distributions of natural language, programming code, and mathematical symbols vary vastly, presenting a complex challenge for large language models (LLMs) that strive to achieve high performance across all three domains simultaneously. Achieving a very high level of proficiency for a…

Cited by 0SourcePDFScholar
2025

How to Synthesize Text Data without Model Collapse?

ICML 2025poster

Model collapse in synthetic data indicates that iterative training on self-generated data leads to a gradual decline in performance. With the proliferation of AI models, synthetic data will fundamentally reshape the web data ecosystem. Future GPT-$\{n\}$ models will inevitably be trained on a blend…

Cited by 4SourcePDFScholar
2025

Intuitive Fine-Tuning: Towards Simplifying Alignment into a Single Process

ACL 2025long

Supervised Fine-Tuning (SFT) and Preference Optimization (PO) are key processes for aligning Language Models (LMs) with human preferences post pre-training. While SFT excels in efficiency and PO in effectiveness, they are often combined sequentially without integrating their optimization objectives.…

2025

MedXpertQA: Benchmarking Expert-Level Medical Reasoning and Understanding

ICML 2025poster

We introduce MedXpertQA, a highly challenging and comprehensive benchmark to evaluate expert-level medical knowledge and advanced reasoning. MedXpertQA includes 4,460 questions spanning 17 specialties and 11 body systems. It includes two subsets, Text for text evaluation and MM for multimodal evalua…

Cited by 16SourcePDFScholar
2025

OpenPRM: Building Open-domain Process-based Reward Models with Preference Trees

ICLR 2025poster

Scaling inference-time computation is increasingly seen as the next frontier in scaling laws for large language models. Previous work in mathematics and coding has demonstrated the remarkable potential for inference-time scaling. During such scaling, fine-grained supervision through process-based re…

Cited by 3SourcePDFScholar
2025

Retrieval-Augmented Visual Question Answering via Built-in Autoregressive Search Engines

AAAI 2025technical

Retrieval-augmented generation (RAG) has emerged to address the knowledge-intensive visual question answering (VQA) task. Current methods mainly employ separate retrieval and generation modules to acquire external knowledge and generate answers, respectively. We propose ReAuSE, an alternative to the…

2025

ReviewRL: Towards Automated Scientific Review with RL

EMNLP 2025

Peer review is essential for scientific progress but faces growing challenges due to increasing submission volumes and reviewer fatigue. Existing automated review approaches struggle with factual accuracy, rating consistency, and analytical depth, often generating superficial or generic feedback lac

2025

SciArena: An Open Evaluation Platform for Non-Verifiable Scientific Literature-Grounded Tasks

NeurIPS 2025spotlight

We present SciArena, an open and collaborative platform for evaluating foundation models on scientific literature-grounded tasks. Unlike traditional benchmarks for scientific literature understanding and synthesis, SciArena engages the research community directly, following the Chatbot Arena evalua…

Cited by 0SourceScholar
2025

Video-T1: Test-time Scaling for Video Generation

ICCV 2025poster

With the scale capability of increasing training data, model size, and computational cost, video generation has achieved impressive results in digital creation, enabling users to express creativity across various domains. Recently, researchers in Large Language Models (LLMs) have expanded the scalin…

Cited by 0SourcePDFScholar
2024

CoGenesis: A Framework Collaborating Large and Small Language Models for Secure Context-Aware Instruction Following

ACL 2024long

With the advancement of language models (LMs), their exposure to private data is increasingly inevitable, and their deployment (especially for smaller ones) on personal devices, such as PCs and smartphones, has become a prevailing trend. In contexts laden with user information, enabling models to bo…

2024

Generative Multi-Modal Knowledge Retrieval with Large Language Models

AAAI 2024technical

Knowledge retrieval with multi-modal queries plays a crucial role in supporting knowledge-intensive multi-modal applications. However, existing methods face challenges in terms of their effectiveness and training efficiency, especially when it comes to training and integrating multiple retrievers to…

2024

PaD: Program-aided Distillation Can Teach Small Models Reasoning Better than Chain-of-thought Fine-tuning

NAACL 2024long

While large language models (LLMs) excel in various natural language processing tasks, their huge size and the inaccessibility of parameters present challenges for practical deployment. Previous studies try to distill task-specific ability from LLMs to smaller models, using data synthesis and chain-…

2024

SMR: State Memory Replay for Long Sequence Modeling

ACL 2024findings

Despite the promising performance of state space models (SSMs) in long sequence modeling, limitations still exist. Advanced SSMs like S5 and S6 (Mamba) in addressing non-uniform sampling, their recursive structures impede efficient SSM computation via convolution. To overcome compatibility limitatio…

2024

Scalable Efficient Training of Large Language Models with Low-dimensional Projected Attention

EMNLP 2024main

Improving the effectiveness and efficiency of large language models (LLMs) simultaneously is a critical yet challenging research goal. In this paper, we find that low-rank pre-training, normally considered as efficient methods that will compromise performance, can be scalably effective when reduced…

2024

UltraMedical: Building Specialized Generalists in Biomedicine

NeurIPS 2024spotlight

Large Language Models (LLMs) have demonstrated remarkable capabilities across various domains and are moving towards more specialized areas. Recent advanced proprietary models such as GPT-4 and Gemini have achieved significant advancements in biomedicine, which have also raised privacy and security…

2023

CRaSh: Clustering, Removing, and Sharing Enhance Fine-tuning without Full Large Language Model

EMNLP 2023long main

Instruction tuning has recently been recognized as an effective way of aligning Large Language Models (LLMs) to enhance their generalization ability across various tasks. However, when tuning publicly accessible, centralized LLMs with private instruction data, privacy concerns are inevitable. While…

Cited by 0SourcecodeScholar
2021

BoB: BERT Over BERT for Training Persona-based Dialogue Models from Limited Personalized Data

ACL 2021long

Maintaining a consistent persona is essential for dialogue agents. Although tremendous advancements have been brought, the limited-scale of annotated personalized dialogue datasets is still a barrier towards training robust and consistent persona-based dialogue models. This work shows how this chall…