Author

Wang-Song Yin

2 papers indexed here

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

#machine learning Preprint Sep 2026

CacheRepair: Learning to Repair Cross-Chunk Context in RAG for KV Cache Fusion

This work introduces CacheRepair, a lightweight network that learns the difference between independently computed KV caches and those produced by processing the chunks together, and shows that repair reduces KV errors both near chunk boundaries and throughout chunk interiors.

Gen Wang, Wang-Song Yin, Yeerzhati Abudunuer et al. · 0 citations

#artificial intelligence Preprint Sep 2026

Empowering Hybrid Attention Models on NPUs

HA-NPU is presented, the first system to enable efficient hybrid attention LLM inference on edge NPUs without modifying the underlying algorithms, and enhances execution efficiency by reorganizing the dataflow of the LA components across three levels.

Yin-Yuan Zhang, Da-Liang Xu, Xiao-Long Huang et al. · 0 citations