中文

Planning over Matrix-Factorization MDPs for Candidate Generation

信息检索 2026-07-02 v1

摘要

For a recommender service, we view the customer journey as a chain of item recommendations: a useful item changes the user's state and therefore what should be retrieved next. Standard matrix-factorization retrieval ignores this -- it builds one user vector and returns the top-KK items by a static score, treating them as independent. We ask a narrow question: when is it worth planning over the user-state dynamics that fold-in induces? To answer it we propose casting top-KK retrieval as an MDP over the implicit-ALS posterior (A1,u)(A^{-1},u), where an action is an item and the transition is a closed-form rank-one fold-in, and the trajectory reward combines a relevance similarity with a posterior-alignment term. Under the same fixed embeddings we compare static retrieval, one-step planning, and horizon-KK MCTS across five datasets and two protocols: a per-user leave-last-nn split and a stricter global time split. Dynamics-aware planning tends to overcome static retrieval on all datasets under leave-last-nn, and the gains hold on MovieLens-1M and the VK-LSVD slices under the global time split. A single step of lookahead already captures most of the gain, so the lightweight planning layer turns static top-KK scoring into a short decision and improves retrieval over fixed collaborative-filtering embeddings, with no retraining and no change to the representation. These gains depend on measuring relevance with cosine rather than inner-product similarity, which is otherwise entangled with item popularity.

引用

@article{arxiv.2607.02115,
  title  = {Planning over Matrix-Factorization MDPs for Candidate Generation},
  author = {Mikhail Trapeznikov and Maksim Utushkin},
  journal= {arXiv preprint arXiv:2607.02115},
  year   = {2026}
}

备注

Accepted to the 5th Workshop on End-to-End Customer Journey Optimization at KDD 2026. 6 pages, 3 figures, 2 tables