← Search

Wenhang Shi

4 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

Training Prompt Matters: State-Adaptive Optimization for Robust Fine-Tuning

ICML 2026poster

While prompt engineering is instrumental in maximizing the capabilities of Large Language Models (LLMs) during inference, the role of prompts during training remains critically underexplored. Prevailing fine-tuning paradigms typically treat training prompts as mere surface forms, assuming that seman…

Cited by 0SourceScholar
2025

Joint Knowledge Editing for Information Enrichment and Probability Promotion

AAAI 2025technical

Knowledge stored in large language models requires timely updates to reflect the dynamic nature of real-world information. To update the knowledge, most knowledge editing methods focus on the low layers, since recent probes into the knowledge recall process reveal that the answer information is enri…

2025

No Loss, No Gain: Gated Refinement and Adaptive Compression for Prompt Optimization

NeurIPS 2025poster

Prompt engineering is crucial for leveraging the full potential of large language models (LLMs). While automatic prompt optimization offers a scalable alternative to costly manual design, generating effective prompts remains challenging. Existing methods often struggle to stably generate improved pr…

Cited by 0SourcecodeScholar