← Search

Wanru Zhao

10 accepted papers

2026

Cascadia: An Efficient Cascade Serving System for Large Language Models

ICLR 2026poster

Recent advances in large language models (LLMs) have intensified the need to deliver both rapid responses and high-quality outputs. More powerful models yield better results but incur higher inference latency, whereas smaller models are faster yet less capable. Recent work proposes balancing this la…

Cited by 6SourceScholar
2026

Dirichlet-Prior Shaping: Guiding Expert Specialization in Upcycled MoEs

ICML 2026poster

Upcycling pre-trained dense models into sparse Mixture-of-Experts (MoEs) efficiently increases model capacity but often suffers from poor expert specialization due to naive weight replication. We introduce Dirichlet-Prior Shaping Loss (DPSL), a novel router regularization technique that directly sha…

Cited by 0SourceScholar
2026

Prototype Entropy Alignment: Reinforcing Structured Uncertainty in LLM Reasoning

AAAI 2026technical

Recent research reveals that a minority of high-entropy tokens significantly influence the reasoning quality of large language models (LLMs). Inspired by this, we propose Prototype Entropy Alignment (PEA), a reinforcement learning framework that models effective reasoning not as a single path but as

Cited by 0SourcePDFScholar
2026

Rethinking Data Curation in LLM Training: Online Reweighting Offers Better Generalization than Offline Methods

ICLR 2026poster

Data curation is a critical yet underexplored component in large language model (LLM) training. Existing approaches (such as data selection and data mixing) operate in an offline paradigm, decoupled from the training process. This separation introduces extra engineering overhead and makes curated su…

Cited by 0SourcecodeScholar
2026

TokSuite: Measuring the Impact of Tokenizer Choice on Language Model Behavior

ICML 2026oral

Tokenizers provide the fundamental basis through which text is represented and processed by language models (LMs). Despite the importance of tokenization, its role in LM performance and behavior is poorly understood due to the challenge of measuring the impact of tokenization in isolation. To addres…

Cited by 0SourceScholar
2025

FedDQC: Data Quality Control in Federated Instruction-tuning of Large Language Models

ACL 2025finding

Federated Learning (FL) enables privacy-preserving collaborative instruction tuning of large language models (LLMs) by leveraging massively distributed data. However, the decentralized nature of FL exacerbates data quality challenges, as local clients lack global visibility to filter noisy or low-qu…

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
2024

Breaking Physical and Linguistic Borders: Multilingual Federated Prompt Tuning for Low-Resource Languages

ICLR 2024poster

Pretrained large language models (LLMs) have emerged as a cornerstone in modern natural language processing, with their utility expanding to various applications and languages. However, the fine-tuning of multilingual LLMs, particularly for low-resource languages, is fraught with challenges steming…

Cited by 17SourcePDFScholar
2024

CLUES: Collaborative Private-domain High-quality Data Selection for LLMs via Training Dynamics

NeurIPS 2024poster

Recent research has highlighted the importance of data quality in scaling large language models (LLMs). However, automated data quality control faces unique challenges in collaborative settings where sharing is not allowed directly between data silos. To tackle this issue, this paper proposes a nove…

Cited by 2SourcePDFScholar
2024

MR-Ben: A Meta-Reasoning Benchmark for Evaluating System-2 Thinking in LLMs

NeurIPS 2024poster

Large language models (LLMs) have shown increasing capability in problem-solving and decision-making, largely based on the step-by-step chain-of-thought reasoning processes. However, evaluating these reasoning abilities has become increasingly challenging. Existing outcome-based benchmarks are begin…

Cited by 14SourcePDFScholar