English

Clustering Context in Off-Policy Evaluation

Machine Learning 2025-03-03 v1 Artificial Intelligence Machine Learning

Abstract

Off-policy evaluation can leverage logged data to estimate the effectiveness of new policies in e-commerce, search engines, media streaming services, or automatic diagnostic tools in healthcare. However, the performance of baseline off-policy estimators like IPS deteriorates when the logging policy significantly differs from the evaluation policy. Recent work proposes sharing information across similar actions to mitigate this problem. In this work, we propose an alternative estimator that shares information across similar contexts using clustering. We study the theoretical properties of the proposed estimator, characterizing its bias and variance under different conditions. We also compare the performance of the proposed estimator and existing approaches in various synthetic problems, as well as a real-world recommendation dataset. Our experimental results confirm that clustering contexts improves estimation accuracy, especially in deficient information settings.

Keywords

Cite

@article{arxiv.2502.21304,
  title  = {Clustering Context in Off-Policy Evaluation},
  author = {Daniel Guzman-Olivares and Philipp Schmidt and Jacek Golebiowski and Artur Bekasov},
  journal= {arXiv preprint arXiv:2502.21304},
  year   = {2025}
}

Comments

35 pages, 25 figures, 2 tables. AISTATS 2025

R2 v1 2026-06-28T22:02:16.677Z