Long-Context Demonstration Selection Using State Space Models

Ziniu Zhang, Zhenshuo Zhang, Ruoxuan Xiong +2

We study the problem of demonstration selection, which involves selecting a subset of examples for prepending to a query to a language model. This problem is closely related to in-…