Skip to yearly menu bar Skip to main content


Oral

Generalization Error Bounds for Two-stage Recommender Systems with Tree Structure

Jin Zhang · Ze Liu · Defu Lian · Enhong Chen

East Meeting Room 1-3
[ ] [ Visit Oral Session 1D: Learning Theory ]
Wed 11 Dec 10 a.m. — 10:20 a.m. PST

Abstract:

Two-stage recommender systems play a crucial role in efficiently identifying relevant items and personalizing recommendations from a vast array of options. This paper, based on an error decomposition framework, analyzes the generalization error for two-stage recommender systems with a tree structure, which consist of an efficient tree-based retriever and a more precise yet time-consuming ranker. We use the Rademacher complexity to establish the generalization upper bound for various tree-based retrievers using beam search, as well as for different ranker models under a shifted training distribution. Both theoretical insights and practical experiments on real-world datasets indicate that increasing the branches in tree-based retrievers and harmonizing distributions across stages can enhance the generalization performance of two-stage recommender systems.

Chat is not available.