← Search

Xuekai Zhu

18 accepted papers

2026

DriveMoE: Mixture-of-Experts for Vision-Language-Action Model in End-to-End Autonomous Driving

CVPR 2026

End-to-end autonomous driving (E2E-AD) demands effective processing of multi-view sensor data and robust handling of diverse and complex driving scenarios, particularly rare maneuvers such as aggressive turns. The recent success of the Mixture-of-Experts (MoE) architecture in Large Language Models (

Cited by 0SourcecodeScholar
2026

Flow of Spans: Generalizing Language Models to Dynamic Span-Vocabulary via GFlowNets

ICLR 2026poster

Standard autoregressive language models generate text token-by-token from a fixed vocabulary, inducing a *tree-structured state space* when viewing token sampling as an action, which limits flexibility and expressiveness. Recent work introduces dynamic vocabulary by sampling retrieved text spans but…

Cited by 0SourceScholar
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

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

MultiBreak: A Scalable and Diverse Multi-turn Jailbreak Benchmark for Evaluating LLM Safety

ICML 2026poster

We present MultiBreak, a scalable and diverse multi-turn jailbreak benchmark to evaluate large language model (LLM) safety. Multi-turn jailbreaks mimic natural conversational settings, making them easier to bypass safety-aligned LLM than single-turn jailbreaks. Existing multi-turn benchmarks are lim…

Cited by 0SourceScholar
2026

Reasoning with Exploration: An Entropy Perspective

AAAI 2026technical

Balancing exploration and exploitation is a central goal in reinforcement learning (RL). Despite recent advances in enhancing language model (LM) reasoning, most methods lean toward exploitation, and increasingly encounter performance plateaus. In this work, we revisit entropy -- a signal of explora

Cited by 0SourcePDFScholar
2026

SEMA: Simple yet Effective Learning for Multi-Turn Jailbreak Attacks

ICLR 2026poster

Multi-turn jailbreaks capture the real threat model for safety-aligned chatbots, where single-turn attacks are merely a special case. Yet existing approaches break under exploration complexity and intent drift. We propose SEMA, a simple yet effective framework that trains a multi-turn attacker witho…

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

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

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

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
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

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
2023

StoryTrans: Non-Parallel Story Author-Style Transfer with Discourse Representations and Content Enhancing

ACL 2023long

Non-parallel text style transfer is an important task in natural language generation. However, previous studies concentrate on the token or sentence level, such as sentence sentiment and formality transfer, but neglect long style transfer at the discourse level. Long texts usually involve more compl…