Tong Yu

Papers on Lattice

Total citations

Topics

Publication activitypapers/week, last 8 weeks

Research focus

Reasoning & Chain-of-Thought (2)Training Efficiency & Optimization (2)RLHF & Preference Learning (1)Architecture Design (Transformers, SSMs, MoE) (1)

Frequent co-authors

Gagan Mundada (1)Zihan Huang (1)Rohan Surana (1)Sheldon Yu (1)

Papers (2)

Feb 19, 2026

Gagan Mundada +73w ago

WS-GRPO: Weakly-Supervised Group-Relative Policy Optimization for Rollout-Efficient Reasoning

Stop wasting compute: WS-GRPO learns when an LLM should stop reasoning, slashing rollout lengths without sacrificing accuracy by turning final answer correctness into prefix-level guidance.

Gagan Mundada, Zihan Huang, Rohan Surana +5

Reasoning & Chain-of-Thought RLHF & Preference Learning Training Efficiency & Optimization

Feb 12, 2026

Amazon ScienceFeb 12, 2026·also NJU

ThinkRouter: Efficient Reasoning via Routing Thinking between Latent and Discrete Spaces

Achieve near 20-point accuracy gains in reasoning tasks by dynamically routing between latent and discrete reasoning spaces based on model confidence.

Tong Yu, Xiang Chen, Haoliang Wang +2

Architecture Design (Transformers, SSMs, MoE)Reasoning & Chain-of-Thought Training Efficiency & Optimization

Search

Tong Yu

Publication activitypapers/week, last 8 weeks

Research focus

Frequent co-authors

Papers (2)