collaborators

5 papers

cs.LG2026

Does RL Expand the Capability Boundary of LLM Agents? A PASS@(k,T) Analysis

Zhiyuan Zhai, Wenjing Yan, Xiaodan Shao +1

Does reinforcement learning genuinely expand what LLM agents can do, or merely make them more reliable? For static reasoning, recent work answers the second: base and RL pass@k cur…

cs.LG2026

Adaptive Test-Time Compute Allocation for Reasoning LLMs via Constrained Policy Optimization

Zhiyuan Zhai, Bingcong Li, Bingnan Xiao +2

Test-time compute scaling, the practice of spending extra computation during inference via repeated sampling, search, or extended reasoning, has become a powerful lever for improvi…

cs.AI2026

When More Thinking Hurts: Overthinking in LLM Test-Time Compute Scaling

Shu Zhou, Rui Ling, Junan Chen +3

Scaling test-time compute through extended chains of thought has become a dominant paradigm for improving large language model reasoning. However, existing research implicitly assu…

cs.LG2026

HiFloat4 Format for Language Model Pre-training on Ascend NPUs

Mehran Taghian, Yunke Peng, Xing Huang +22

Large foundation models have become central to modern machine learning, with performance scaling predictably with model size and data. However, training and deploying such models i…

cs.LG2025

StagFormer: Time Staggering Transformer Decoding for RunningLayers In Parallel

Dylan Cutler, Arun Kandoor, Nishanth Dikkala +3

Decoding in a Transformer based language model is inherently sequential as a token's embedding needs to pass through all the layers in the network before the generation of the next…