arXiv · 2503.08640
Efficient Many-Shot In-Context Learning with Dynamic Block-Sparse Attention
Abstract
Many-shot in-context learning has recently shown promise as an alternative to finetuning, with the major advantage that the same model can be served for multiple tasks. However, this shifts the computational burden from training-time to inference-time, making deployment of many-shot ICL challenging to justify in-practice. This cost is further increased if a custom demonstration set is retrieved for each inference example. We present Dynamic Block-Sparse Attention, a training-free framework for retrieval-based many-shot in-context learning. By combining carefully designed block-sparse attention and retrieval of cached groups of demonstrations, we achieve comparable per-example latency to finetuning while maintaining on average >95% of the best method's accuracy across strong ICL and finetuning baselines. We hope that this will further enable the deployment of many-shot ICL at scale.
Explore related subjects
Keep this discovery
Emily Xiao, Chin-Jou Li, Yilin Zhang, Graham Neubig, Amanda Bertsch. 2025-03-11. Efficient Many-Shot In-Context Learning with Dynamic Block-Sparse Attention. https://arxiv.org/abs/2503.08640
Cite the original work for its findings. Save a collection to share your selection of sources.