← Search

Size Zheng

6 accepted papers

2026

CoCoQuant: Breaking the Bandwidth Wall via Co-Optimized Communication and Computation Quantization

ICML 2026poster

The rapid scaling of large language models (LLMs) has made distributed inference indispensable, yet end-to-end latency is increasingly dominated by communication, forming a critical bandwidth wall that fundamentally limits the practical gains of existing quantization techniques. Existing approaches …

Cited by 0SourceScholar
2026

DITRON: Distributed Multi-level Tiling Compiler for Parallel Tensor Programs

ICML 2026poster

The scaling of large language models (LLMs) is currently bottlenecked by the rigidity of distributed programming. While high-performance libraries like CuBLAS and NCCL provide optimized primitives, they lack the flexibility required for rapidly evolving model architectures. Conversely, existing tens…

Cited by 0SourceScholar
2026

Forge: Compiling a Unified Abstraction into Scalable Kernels for Linear Attention

ICLR 2026poster

The quadratic complexity of softmax attention poses a major bottleneck for long-context modeling, motivating a surge of linear attention variants with linear complexity. Unlike softmax attention, which benefits from optimized kernels, linear attention lacks general-purpose, hardware-efficient suppor…

Cited by 0SourceScholar
2025

MxMoE: Mixed-precision Quantization for MoE with Accuracy and Performance Co-Design

ICML 2025poster

Mixture-of-Experts (MoE) models face deployment challenges due to their large parameter counts and computational demands. We explore quantization for MoE models and highlight two key insights: 1) linear blocks exhibit varying quantization sensitivity, and 2) divergent expert activation frequencies c…

2025

ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference

ICML 2025spotlight

With the widespread deployment of long-context large language models (LLMs), there has been a growing demand for efficient support of high-throughput inference. However, as the key-value (KV) cache expands with the sequence length, the increasing memory footprint and the need to access it for decodi…

2024

ArkVale: Efficient Generative LLM Inference with Recallable Key-Value Eviction

NeurIPS 2024poster

Large Language Models (LLMs) are widely used in today's tasks of natural language processing. To support applications like multi-turn chats, document understanding, and content generation, models with long context lengths are growing in importance. However, managing long contexts brings substantial…

Cited by 2SourcePDFScholar