← Search

Aurick Qiao

6 accepted papers

2025

Efficiently Scaling LLM Reasoning Programs with Certaindex

NeurIPS 2025poster

Test-time reasoning algorithms such as chain-of-thought, self-consistency, and MCTS enhance LLM problem-solving but can wastefully generate many tokens without improving accuracy. At the same time, we observe that these algorithms exhibit answer stabilization: their intermediate solutions often ceas…

Cited by 36SourcecodeScholar
2025

STUN: Structured-Then-Unstructured Pruning for Scalable MoE Pruning

ACL 2025long

Mixture-of-experts (MoEs) have been adopted for reducing inference costs by sparsely activating experts in large language models (LLMs). Despite these reductions, the massive number of parameters in MoEs still makes them expensive to serve. Conventionally, unstructured or structured pruning has been…

2025

SuffixDecoding: Extreme Speculative Decoding for Emerging AI Applications

NeurIPS 2025spotlight

Speculative decoding is widely adopted to reduce latency in large language model (LLM) inference by leveraging smaller draft models capable of handling diverse user tasks. However, emerging AI applications, such as LLM-based agents, present unique workload characteristics: instead of diverse indepen…

Cited by 0SourcecodeScholar
2025

SwiftKV: Fast Prefill-Optimized Inference with Knowledge-Preserving Model Transformation

EMNLP 2025

LLM inference for enterprise applications, such as summarization, RAG, and code-generation, typically observe much longer prompt than generations, leading to high prefill cost and response latency. We present SwiftKV, a novel model transformation and distillation procedure targeted at reducing the p

2024

Efficient LLM Scheduling by Learning to Rank

NeurIPS 2024poster

In Large Language Model (LLM) inference, the output length of an LLM request is typically regarded as not known a priori. Consequently, most LLM serving systems employ a simple First-come-first-serve (FCFS) scheduling strategy, leading to Head-Of-Line (HOL) blocking and reduced throughput and servic…