← Search

Yiwu Yao

10 accepted papers

2026

ASTRAEA: A Token-wise Acceleration Framework for Video Diffusion Transformers

ICLR 2026poster

Video diffusion transformers (vDiTs) have made tremendous progress in text-to-video generation, but their high computational demands pose a major challenge for practical deployment. While existing studies propose acceleration methods to reduce workload at various granularities, they often rely on he…

Cited by 0SourceScholar
2026

FreeKV: Boosting KV Cache Retrieval for Efficient LLM Inference

ICLR 2026poster

Large language models (LLMs) have been widely deployed with rapidly expanding context windows to support increasingly demanding applications. However, long contexts pose significant deployment challenges, primarily due to the KV cache whose size grows proportionally with context length. While KV cac…

Cited by 0SourcecodeScholar
2026

TimeRipples: Accelerating vDiTs by Understanding the Spatio-Temporal Correlations in Latent Space

CVPR 2026

The recent surge in video generation has shown the growing demand for high-quality video synthesis using large vision models. Existing video generation models are predominantly based on the video diffusion transformer (vDiT), however, they suffer from substantial inference delay due to self-attentio

Cited by 0SourceScholar
2025

DartQuant: Efficient Rotational Distribution Calibration for LLM Quantization

NeurIPS 2025poster

Quantization plays a crucial role in accelerating the inference of large-scale models, and rotational matrices have been shown to effectively improve quantization performance by smoothing outliers. However, end-to-end fine-tuning of rotational optimization algorithms incurs high computational costs…

Cited by 0SourceScholar
2025

Dynamic Low-Rank Sparse Adaptation for Large Language Models

ICLR 2025poster

Despite the efficacy of network sparsity in alleviating the deployment strain of Large Language Models (LLMs), it endures significant performance degradation. Applying Low-Rank Adaptation (LoRA) to fine-tune the sparse LLMs offers an intuitive approach to counter this predicament, while it holds sho…

2025

KVTuner: Sensitivity-Aware Layer-Wise Mixed-Precision KV Cache Quantization for Efficient and Nearly Lossless LLM Inference

ICML 2025poster

KV cache quantization can improve Large Language Models (LLMs) inference throughput and latency in long contexts and large batch-size scenarios while preserving LLMs effectiveness. However, current methods have three unsolved issues: overlooking layer-wise sensitivity to KV cache quantization, high…

2025

RazorAttention: Efficient KV Cache Compression Through Retrieval Heads

ICLR 2025poster

The memory and computational demands of Key-Value (KV) cache present significant challenges for deploying long-context language models. Previous approaches attempt to mitigate this issue by selectively dropping tokens, which irreversibly erases critical information that might be needed for future qu…

Cited by 23SourcePDFScholar
2024

Dynamic Sparse No Training: Training-Free Fine-tuning for Sparse LLMs

ICLR 2024poster

The ever-increasing large language models (LLMs), though opening a potential path for the upcoming artificial general intelligence, sadly drops a daunting obstacle on the way towards their on-device deployment. As one of the most well-established pre-LLMs approaches in reducing model complexity, net…