Probabilistic Rank and Reward: A Scalable Model for Slate Recommendation
Imad Aouali, Achraf Ait Sidi Hammou, Otmane Sakhi, David Rohde, Flavian Vasile
TL;DR
Probabilistic Rank and Reward (PRR) introduces a scalable model for slate recommendations where a user can interact with at most one item from a slate of size $K$. It jointly models reward and rank via a categorical distribution with engagement features $\mathbf{y}$ and user-interest features $\mathbf{z}$, enabling fast decision making through maximum inner product search (MIPS). The decision rule reduces to a MIPS problem by removing nuisance factors, yielding end-to-end $\mathcal{O}(\log P)$ retrieval suitable for large catalogs and low-latency domains. Empirical results on synthetic and session-completion tasks show PRR outperforms off-policy reward baselines (IPS/DM) and variants, while offering superior computational efficiency and scalability. The work highlights a practical, model-based approach that bridges reward optimization and ranking in slate settings and points to extensions with priors and richer interaction modeling.
Abstract
We introduce Probabilistic Rank and Reward (PRR), a scalable probabilistic model for personalized slate recommendation. Our approach allows off-policy estimation of the reward in the scenario where the user interacts with at most one item from a slate of K items. We show that the probability of a slate being successful can be learned efficiently by combining the reward, whether the user successfully interacted with the slate, and the rank, the item that was selected within the slate. PRR outperforms existing off-policy reward optimizing methods and is far more scalable to large action spaces. Moreover, PRR allows fast delivery of recommendations powered by maximum inner product search (MIPS), making it suitable in low latency domains such as computational advertising.
