RewardRank: Optimizing True Learning-to-Rank Utility

Gaurav Bhatt, Kiran Koshy Thekumparampil, Tanmay Gangwani, Tesi Xiao, Leonid Sigal
8/19/2025
cs.IRcs.LG

Official code: GauravBh1010tt/RewardRank (5 stars, last commit 2025-10-22).

Abstract

Traditional ranking systems optimize offline proxy objectives that rely on oversimplified assumptions about user behavior, often neglecting factors such as position bias and item diversity. Consequently, these models fail to improve true counterfactual utilities such as such as click-through rate or purchase probability, when evaluated in online A/B tests. We introduce RewardRank, a data-driven learning-to-rank (LTR) framework for counterfactual utility maximization. RewardRank first learns a reward model that predicts the utility of any ranking directly from logged user interactions, and then trains a ranker to maximize this reward using a differentiable soft permutation operator. To enable rigorous and reproducible evaluation, we further propose two benchmark suites: (i) Parametric Oracle Evaluation (PO-Eval), which employs an open-source click model as a counterfactual oracle on the Baidu-ULTR dataset, and (ii) LLM-as-User Evaluation (LAU-Eval), which simulates realistic user behavior via large language models on the Amazon-KDD-Cup dataset. RewardRank achieves the highest counterfactual utility across both benchmarks and demonstrates that optimizing classical metrics such as NDCG is sub-optimal for maximizing true user utility. Finally, using real user feedback from the Baidu-ULTR dataset, RewardRank establishes a new state of the art in offline relevance performance. Overall, our results show that learning-to-rank can be reformulated as direct optimization of counterfactual utility, achieved in a purely data-driven manner without relying on explicit modeling assumptions such as position bias. Our code is available at: $https://github.com/GauravBh1010tt/RewardRank$

View on arXivView PDF

Code Implementations(1)

Counter-factual reward ranking

50Shell, PythonSep 2, 202412 months ago

Similar Papers

Cite this paper

@article{bhatt2025rewardrank,
  title  = {RewardRank: Optimizing True Learning-to-Rank Utility},
  author = {Gaurav Bhatt and Kiran Koshy Thekumparampil and Tanmay Gangwani and Tesi Xiao and Leonid Sigal},
  year   = {2025},
  eprint = {2508.14180},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2508.14180v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2508.14180.svg)](https://codeofpaper.com/paper/2508.14180)

Discussion