vector-search-as-nearest-neighbor-matching-rag-based-policy-learning-in-causal-inference-7eeeb91f·1 events·first seen Aliases: Vector Search As Nearest Neighbor Matching: RAG-based Policy Learning in Causal Inference
A new arXiv preprint proposes one-step and two-step methods for policy learning using retrieval-augmented generation, formulated under the potential outcome framework from causal inference. The two-step method connects action-specific vector search to nearest-neighbor matching, with regret decomposed into candidate-generation and within-candidate choice components. Theoretical bounds on regret are derived using prediction-error guarantees for nearest-neighbor estimators and transformers. The work bridges RAG methodology with formal causal inference theory, offering a principled treatment of action selection in RAG-based systems.