← Search

Shaoyi Zheng

3 accepted papers

2026

HilbertA: Hilbert-Curve–Aligned Sparse Attention for 2D Structured Data

ICML 2026poster

Designing sparse attention for 2-dimensional image data in diffusion models and vision-language models requires reconciling spatial locality with hardware-efficient execution, a fundamental trade-off that existing methods struggle to resolve. Prior approaches preserve 2D structure through handcrafte…

Cited by 0SourceScholar
2026

InfoFlow KV: Information-Flow-Aware KV Recomputation for Long Context

ICML 2026poster

Retrieval-augmented generation (RAG) for long-context question answering is bottlenecked by inference-time prefilling over large retrieved contexts. A common strategy is to precompute key–value (KV) caches for individual documents and selectively recompute a small subset of tokens to restore global …

Cited by 0SourceScholar