activity
20242026
most citedRethinking Predictive Modeling for LLM Routing: When Simple kNN Beats Complex Learned Routers

1 citations · 1 across the 14 of their papers we have counts for

collaborators
Showing cs.LGShow all

8 papers · 1 filter

cs.LG20261 cited

Rethinking Predictive Modeling for LLM Routing: When Simple kNN Beats Complex Learned Routers

Yang Li

As large language models (LLMs) grow in scale and specialization, routing--selecting the best model for a given input--has become essential for efficient and effective deployment.…

cs.LG2026

Look Inward to Explore Outward: Learning Temperature Policy from LLM Internal States via Hierarchical RL

Yixiao Zhou, Yang Li, Dongzhou Cheng +2

Reinforcement Learning from Verifiable Rewards (RLVR) trains large language models (LLMs) from sampled trajectories, making decoding strategy a core component of learning rather th…

cs.LG2026

Turn-PPO: Turn-Level Advantage Estimation with PPO for Improved Multi-Turn RL in Agentic LLMs

Junbo Li, Peng Zhou, Rui Meng +3

Reinforcement learning (RL) has re-emerged as a natural approach for training interactive LLM agents in real-world environments. However, directly applying the widely used Group Re…

cs.LG2025

Grokked Models are Better Unlearners

Yuanbang Liang, Yang Li

Grokking-delayed generalization that emerges well after a model has fit the training data-has been linked to robustness and representation quality. We ask whether this training reg…

cs.LG2025

Skip a Layer or Loop it? Test-Time Depth Adaptation of Pretrained LLMs

Ziyue Li, Yang Li, Tianyi Zhou

Can a pretrained neural network adapt its architecture to different inputs without any finetuning? Do we need all layers for simple tasks, and are they adequate for challenging tas…

cs.LG2025

Policy Guided Tree Search for Enhanced LLM Reasoning

Yang Li

Despite their remarkable capabilities, large language models often struggle with tasks requiring complex reasoning and planning. While existing approaches like Chain-of-Thought pro…