arXiv AI

SAKURA: Serving Attention Key-Value State Under Rate Adaptation for Private Image Editing on Phones

SAKURA is a cloud-edge system that delivers reusable attention key‑value state to mobile devices for diffusion‑based image editing while respecting a bandwidth budget. It compresses the state with a token‑importance‑aware method, dynamically selects cache representations based on network conditions, and overlaps transfer with on‑device computation. Experiments on a Snapdragon 8 Elite Gen 5 phone show that SAKURA cuts latency by up to 3.27× and reduces cache storage by 3.06× compared to uncompressed caches, all while maintaining image quality.

Hugging Face Trending Papers
Jun 11

MiniPIC: Flexible Position-Independent Caching in <100LOC

Retrieval-augmented and agentic workloads repeatedly prefill recurring predictable structured inputs (which we call "spans") such as documents and code files. Yet, prefix caching in engines such as vLLM cannot reuse their KV entries unless they share identical prefixes with another request, while Position-Independent Caching (PIC) implementations within production-grade inference servers typically either require substantial server code changes or keep KV state outside the server, incurring host-to-device transfer overhead.

arXiv AI
Oct 9

SparKV: Overhead-Aware KV Cache Loading for Efficient On-Device LLM Inference

SparKV is an adaptive framework for on-device large language model inference that intelligently decides whether to stream or compute each key-value (KV) cache chunk, overlapping communication and computation to cut latency. It dynamically adjusts offline-generated schedules at runtime to cope with changing wireless connectivity and edge resource availability. Experiments show SparKV can reduce Time-to-First-Token by 1.3×–5.1× and lower per-request energy consumption by 1.5×–3.3× with minimal impact on response quality.

By Hongyao Liu, Liuqun Zhai, Junyi Wang, Zhengru Fang, Jingshu Chen, Jun Huang
arXiv AI
Oct 31

GAAC: Lossless KV Compression for Bandwidth-Efficient LLM Inference on Personal Devices

The paper introduces GAAC, a lossless key‑value (KV) compression method for large language model inference on personal devices. GAAC encodes BF16 exponents with per‑channel base‑plus‑offset codes, adapts code widths across small groups, and reconstructs KV tiles on the fly to avoid full‑cache decompression. Experiments show 1.4× compression for BF16 KV, speedups of up to 1.27× in decode attention, and a 26% reduction in resident KV memory while boosting throughput on NVIDIA and Apple hardware.

By Yipin Guo, Arun M George, Siddharth Joshi
arXiv Machine Learning
Oct 19

DIFFLET: A Fast Inference Engine for Diffusion Transformer Models on AWS Trainium

DIFFLET is a serving engine for diffusion transformer models on AWS Trainium that allows five model pipelines to share a single compilation backend. It splits joint attention across cores by attention head and uses a host controller to decide whether to run the denoiser or reuse previous outputs at each denoising step. Across five image and video model families, DIFFLET’s fused attention accelerates denoiser steps by 1.2–4.3× compared to PyTorch/XLA, and resident serving is 1.5–3.4× faster than standalone runs that reload weights.

By Bin Ma, Min Yu, Vijay Kodamalla, Jialin Liu, Dong Li
arXiv AI
Jul 16

Full-Pipeline Inference Optimization for MiMo-V2.5 Series: Pushing Hybrid SWA Efficiency to the Limit

arXiv:2607. 13095v1 Announce Type: cross Abstract: We present a full-pipeline inference optimization for the MiMo-V2.

By Xiaomi MiMo Team, Anqi Liu, Aoxin Ma, Bo Chen, Bo Yang, Chen Wang, Chen Zhang, Chengda Tang, Chengwei Wang, Chiheng Lou, Depeng Yan, Fuli Luo, Gang Wang, Hailin Zhang, Jiale Sun, Kang Zhou, Rui Huang, Shaohui Liu, Shen Huang, Shijie Cao, Shuaishuai Fan, Tianling Zhou, Xiangwei Deng, Xueyang Xie, Xuli Wang, Yingchun Lai, Yu Yang, Yuan Zhang, Zhen Tang, Zhonghua Deng, Zihan Jiang
arXiv AI
Oct 27

Seeing the Options: Typed Decisions over Visual Candidates with Reusable Option Memory

The paper introduces Iris, a framework that extends typed decision‑making models to multimodal settings where candidates can be images, text, or hybrids. Iris preserves key properties such as candidate‑order invariance, scalability, and cache reuse, and achieves high accuracy on a visual‑candidate benchmark while reducing latency through efficient encoding strategies. The authors also demonstrate significant speedups on GPU systems and outperform larger baseline models on MetaEncoder benchmarks.

By Fengxin Li