JKO-RAG: Distributional Retrieval as Wasserstein Free-Energy Gradient Flow

Levi Segal, Murari Ambati

RAG pipelines return a \emph{ranked list} of passages. We argue this is a mismatch: the downstream language model conditions on a \emph{set}, and the selection problem is fundament…