arXiv stat.ML July 21, 2026 · Papers

Vector Search As Nearest Neighbor Matching: RAG-based Policy Learning in Causal Inference

arXiv:2607.18225v1 Announce Type: cross Abstract: We propose one-step and two-step methods for policy learning with retrieval-augmented generation (RAG). We formulate RAG-based action selection under the potential outcome framework. In the two-step method, vector search retrieves action-specific neighboring evidence in