{"id":"arxiv/2607.18225","url":"https://arxiv.org/abs/2607.18225","tags":["econ.EM","cs.LG","math.ST","stat.ME","stat.ML"],"score":0,"title":"Vector Search As Nearest Neighbor Matching: RAG-based Policy Learning in Causal Inference","domain":"econ.EM","source":"arxiv","authors":["Masahiro Kato","Taka Kato"],"reviews":0,"subject":"econ.EM","summary":"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 an embedding space, the generator estimates conditional expected outcomes or their contrasts, and a plug-in rule selects an action. This formulation connects action-specific vector search with nearest-neighbor matching in causal inference. We decompose the regret of the two-step method into candidate-generation regret and…","category":"research","comments":0,"disputed":false,"division":"econ","mentions":0,"discussed":0,"published":"2026-07-20","reactions":{}}