← Search

Rubing Yang

5 accepted papers

2026

SlimInfer: Accelerating Long-Context LLM Inference via Dynamic Token Pruning

AAAI 2026technical

Long-context inference for Large Language Models (LLMs) is heavily limited by high computational demands. While several existing methods optimize attention computation, they still process the full set of hidden states at each layer, limiting overall efficiency. In this work, we propose SlimInfer, an

Cited by 0SourcePDFScholar
2026

SpecExit: Accelerating Large Reasoning Model via Speculative Exit

ICML 2026poster

Despite their strong performance on reasoning tasks, large reasoning models (LRMs) often suffer from overthinking, producing unnecessarily long outputs and incurring high end-to-end latency, a significant limitation to their real-world deployment. To address overthinking, early-exit mechanisms have …

Cited by 0SourceScholar
2024

Prospective Learning: Learning for a Dynamic Future

NeurIPS 2024poster

In real-world applications, the distribution of the data, and our goals, evolve over time. The prevailing theoretical framework for studying machine learning, namely probably approximately correct (PAC) learning, largely ignores time. As a consequence, existing strategies to address the dynamic natu…

2023

A Picture of the Space of Typical Learnable Tasks

ICML 2023poster

We develop information geometric techniques to understand the representations learned by deep networks when they are trained on different tasks using supervised, meta-, semi-supervised and contrastive learning. We shed light on the following phenomena that relate to the structure of the space of tas…