← Search

Jiajun Chai

17 accepted papers

2026

Contextual Rollout Bandits for Reinforcement Learning with Verifiable Rewards

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) is an effective paradigm for improving the reasoning capabilities of large language models. However, existing RLVR methods utilize rollouts in an indiscriminate and short-horizon manner: responses of heterogeneous quality within each prompt are t…

Cited by 0SourceScholar
2026

Enhancing Complex Symbolic Logical Rea­soning of Large Language Models via Sparse Multi-Agent Debate

ICLR 2026poster

Large language models (LLMs) struggle with complex logical reasoning. Previous work has primarily explored single-agent methods, with their performance remains fundamentally limited by the capabilities of a single model. To our knowledge, this paper first introduce a multi-agent approach specificall…

Cited by 0SourcecodeScholar
2026

GRASP: Graph Reasoning via Agentic Solving and Probing of LLMs

ICML 2026poster

Integrating graph knowledge into Large Language Models (LLMs) via passive representation faces critical bottlenecks: limited context windows, unreliable numerical computation, and structural hallucinations. To solve this, we propose **GRASP** (Graph Reasoning via Agentic Solving and Probing), shifti…

Cited by 0SourceScholar
2026

LogiConBench: Benchmarking Logical Consistencies of LLMs

ICLR 2026poster

Logical consistency, the requirement that statements remain non-contradictory under logical rules, is fundamental for trustworthy reasoning, yet current LLMs often fail to maintain it even on simple inference tasks. Existing benchmarks for LLM logical consistency are not scalable, not diverse, and n…

Cited by 0SourcecodeScholar
2026

Promoting Efficient Reasoning with Verifiable Stepwise Reward

AAAI 2026technical

Large reasoning models (LRMs) have recently achieved significant progress in complex reasoning tasks, aided by reinforcement learning with verifiable rewards. However, LRMs often suffer from overthinking, expending excessive computation on simple problems and reducing efficiency. Existing efficient

Cited by 0SourcePDFScholar
2026

ResRL: Boosting LLM Reasoning via Negative Sample Projection Residual Reinforcement Learning

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) enhances reasoning of Large Language Models (LLMs) but usually exhibits limited generation diversity due to the over-incentivization of positive rewards. Although methods like Negative Sample Reinforcement (NSR) mitigate this issue by upweighting…

Cited by 0SourceScholar
2026

ResT: Reshaping Token-Level Policy Gradients for Tool-Use Large Language Models

ICLR 2026poster

Large language models (LLMs) transcend passive generation and act as goal-directed agents by invoking external tools. Reinforcement learning (RL) offers a principled framework for optimizing these emergent tool-use policies, yet the prevailing paradigm relies exclusively on sparse outcome rewards an…

Cited by 0SourcecodeScholar
2026

Rethinking Personalization in Large Language Models at the Token Level

ICML 2026poster

With large language models (LLMs) now performing strongly across diverse tasks, there is growing demand for them to personalize outputs for individual users. Personalization is typically framed as an additional layer on top of a base NLP task, requiring model responses to meet user-specific needs wh…

Cited by 0SourceScholar
2026

SAE as a Crystal Ball: Interpretable Features Predict Cross-domain Transferability of LLMs without Training

ICLR 2026poster

In recent years, pre-trained large language models have achieved remarkable success across diverse tasks. Besides the pivotal role of self-supervised pre-training, their effectiveness in downstream applications also depends critically on the post-training process, which adapts models to task-specifi…

Cited by 0SourcecodeScholar
2026

SRFT: A Single-Stage Method with Supervised and Reinforcement Fine-Tuning for Reasoning

ICLR 2026poster

Large language models (LLMs) have achieved remarkable progress in reasoning tasks, yet optimally integrating Supervised Fine-Tuning (SFT) and Reinforcement Learning (RL) remains a fundamental challenge. Through a comprehensive analysis of token distributions, learning dynamics, and integration mecha…

Cited by 0SourcecodeScholar
2026

SSL4RL: Revisiting Self-supervised Learning as Intrinsic Reward for Visual-Language Reasoning

ICML 2026poster

Vision-language models (VLMs) have shown remarkable abilities by integrating large language models with visual inputs. However, they often fail to utilize visual evidence adequately, either depending on linguistic priors in vision-centric tasks or resorting to textual shortcuts during reasoning. Alt…

Cited by 0SourceScholar
2025

DipLLM: Fine-Tuning LLM for Strategic Decision-making in Diplomacy

ICML 2025poster

Diplomacy is a complex multiplayer game that re- quires both cooperation and competition, posing significant challenges for AI systems. Traditional methods rely on equilibrium search to generate extensive game data for training, which demands substantial computational resources. Large Lan- guage Mod…

Cited by 0SourcePDFScholar
2025

Empowering LLM Agents with Zero-Shot Optimal Decision-Making through Q-learning

ICLR 2025poster

Large language models (LLMs) are trained on extensive text data to gain general comprehension capability. Current LLM agents leverage this ability to make zero- or few-shot decisions without reinforcement learning (RL) but fail in making optimal decisions, as LLMs inherently perform next-token predi…

Cited by 3SourcePDFScholar
2025

INS: Interaction-aware Synthesis to Enhance Offline Multi-agent Reinforcement Learning

ICLR 2025poster

Data scarcity in offline multi-agent reinforcement learning (MARL) is a key challenge for real-world applications. Recent advances in offline single-agent reinforcement learning (RL) demonstrate the potential of data synthesis to mitigate this issue. However, in multi-agent systems, interactions bet…

Cited by 0SourcePDFScholar
2025

Learning and Planning Multi-Agent Tasks via an MoE-based World Model

NeurIPS 2025poster

Multi-task multi-agent reinforcement learning (MT-MARL) aims to develop a single model capable of solving a diverse set of tasks. However, existing methods often fall short due to the substantial variation in optimal policies across tasks, making it challenging for a single policy model to generaliz…

Cited by 0SourcecodeScholar
2025

RLAE: Reinforcement Learning-Assisted Ensemble for LLMs

EMNLP 2025

Ensembling large language models (LLMs) can effectively combine diverse strengths of different models, offering a promising approach to enhance performance across various tasks. However, existing methods typically rely on fixed weighting strategies that fail to adapt to the dynamic, context-dependen

Cited by 0SourcePDFScholar
2025

UIOrchestra: Generating High-Fidelity Code from UI Designs with a Multi-agent System

EMNLP 2025

Recent advances in large language models (LLMs) have significantly improved automated code generation, enabling tools such as GitHub Copilot and CodeWhisperer to assist developers in a wide range of programming tasks. However, the translation of complex mobile UI designs into high-fidelity front-end

Cited by 0SourcePDFScholar