← Search

Shenggui Li

6 accepted papers

2026

DSA: Efficient Inference For Video Generation Models via Distributed Sparse Attention

ICLR 2026poster

Diffusion Transformer models have driven the rapid advances in video generation, achieving state-of-the-art quality and flexibility. However, their attention mechanism remains a major performance bottleneck, as its dense computation scales quadratically with the sequence length. To overcome this lim…

Cited by 0SourceScholar
2026

PACE: Proactive Agent-Level Admission Control for Efficient Agentic Batch Inference

ICML 2026poster

Batch inference for agentic workloads stresses the GPU key–value (KV) cache in a sustained and cumulative manner, often causing severe throughput degradation well before memory capacity is exhausted. We identify this phenomenon as middle-phase thrashing, a previously under-characterized pathology in…

Cited by 0SourceScholar
2026

SpecForge: A Flexible and Efficient Open-Source Training Framework for Speculative Decoding

ICML 2026poster

Speculative decoding mitigates the memory-bound nature of LLM decoding by using a lightweight draft model to propose multiple tokens for parallel verification. However, its adoption has been limited by the lack of high-quality draft models and scalable training infrastructure. We introduce SpecForge…

Cited by 0SourceScholar
2024

GliDe with a CaPE: A Low-Hassle Method to Accelerate Speculative Decoding

ICML 2024poster

Speculative decoding is a relatively new decoding framework that leverages small and efficient draft models to reduce the latency of LLMs. In this study, we introduce GliDe and CaPE, two low-hassle modifications to vanilla speculative decoding to further improve the decoding speed of a frozen LLM. S…

Cited by 19SourcePDFScholar
2023

Sequence Parallelism: Long Sequence Training from System Perspective

ACL 2023long

Transformer achieves promising results on various tasks. However, self-attention suffers from quadratic memory requirements with respect to the sequence length. Existing work focuses on reducing time and space complexity from an algorithm perspective. In this work, we propose sequence parallelism, a…

Cited by 102SourcePDFScholar