A Statistical Framework for Data-dependent Retrieval-Augmented Models

Soumya Basu, Ankit Singh Rawat, Manzil Zaheer

Modern ML systems increasingly augment input instances with additional relevant information to enhance final prediction. Despite growing interest in such retrieval-augmented models…