3 papers
cs.LG2025
In-Context Curiosity: Distilling Exploration for Decision-Pretrained Transformers on Bandit Tasks
Huitao Yang, Guanting Chen
As large language models (LLMs) continue to grow in capability, there is increasing interest in incorporating them into decision-making tasks. A common pipeline for this is Decisio…
math.PR2025
Greedy Selection under Independent Increments: A Toy Model Analysis
Huitao Yang
We study an iterative selection problem over N i.i.d. discrete-time stochastic processes with independent increments. At each stage, a fixed number of processes are retained based…
cs.CL2024
Fast Best-of-N Decoding via Speculative Rejection
Hanshi Sun, Momin Haider, Ruiqi Zhang +6
The safe and effective deployment of Large Language Models (LLMs) involves a critical step called alignment, which ensures that the model's responses are in accordance with human p…