← Search

Hailin Zhang

7 accepted papers

2026

RaBitQCache: Rotated Binary Quantization for KVCache in Long Context LLM Inference

ICML 2026poster

Long-context Large Language Model inference is severely bottlenecked by the massive Key-Value (KV) cache, yet existing sparse attention methods often suffer from static fixed-budget (Top-k) retrieval or rely on proxy scores that are computationally expensive and biased. To address these limitations,…

Cited by 0SourceScholar
2026

SALE : Low-bit Estimation for Efficient Sparse Attention in Long-context LLM Prefilling

ICML 2026poster

Many advanced Large Language Model (LLM) applications require long-context processing, but the self-attention module becomes a bottleneck during the prefilling stage of inference due to its quadratic time complexity with respect to sequence length. Existing sparse attention methods accelerate attent…

Cited by 0SourcecodeScholar
2026

Stabilizing MoE Reinforcement Learning by Aligning Training and Inference Routers

ICML 2026poster

Reinforcement learning (RL) has emerged as a crucial approach for enhancing the capabilities of large language models. However, in Mixture-of-Experts (MoE) models, the routing mechanism often introduces instability, even leading to catastrophic RL training collapse. We analyze the training-inference…

Cited by 0SourceScholar
2024

Surge Phenomenon in Optimal Learning Rate and Batch Size Scaling

NeurIPS 2024poster

In current deep learning tasks, Adam-style optimizers—such as Adam, Adagrad, RMSprop, Adafactor, and Lion—have been widely used as alternatives to SGD-style optimizers. These optimizers typically update model parameters using the sign of gradients, resulting in more stable convergence curves. The l…

Cited by 6SourcePDFScholar
2023

Model-enhanced Vector Index

NeurIPS 2023poster

Embedding-based retrieval methods construct vector indices to search for document representations that are most similar to the query representations. They are widely used in document retrieval due to low latency and decent recall performance. Recent research indicates that deep retrieval solutions o…

2022

Knowledge Distillation With the Reused Teacher Classifier

CVPR 2022poster

Knowledge distillation aims to compress a powerful yet cumbersome teacher model into a lightweight student model without much sacrifice of performance. For this purpose, various approaches have been proposed over the past few years, generally with elaborately designed knowledge representations, whic…

Cited by 246PDFcodeScholar