DARE: Aligning LLM Agents with the R Statistical Ecosystem via Distribution-Aware Retrieval

Maojun Sun^†, Yue Wu^†, Yifei Xie^†, Ruijian Han, Binyan Jiang, Defeng Sun, Yancheng Yuan, Jian Huang.

Under review at the Journal of the American Statistical Association.

† Equal contribution (co-first authors).

Paper

Overview

Distribution-aware retrieval integrates statistical data profiles into R package embeddings, achieving 93.47% NDCG@10 versus 79.32% for the strongest baseline and improving downstream agents on statistical analysis tasks.

All publications & manuscripts →