Organizations: University of Science and Technology of China, Hefei, China · Huawei, Shanghai, China · Shanghai Jiao Tong University, Shanghai, China · Beijing University of Posts and Telecommunications, Beijing, China
Generative models offer a promising paradigm for the final stage reranking in multi-stage recommender systems, with the ability to capture inter-item dependencies within reranked lists. However, their practical deployment still faces two key challenges: (1) an inherent conflict between achieving high generation quality and ensuring low-latency inference, making it difficult to balance the two, and (2) insufficient interaction between user and item features in existing methods. To address these challenges, we propose a novel Personalized Semi-Autoregressive with online knowledge Distillation (PSAD) framework for reranking. In this framework, the teacher model adopts a semi-autoregressive generator to balance generation quality and efficiency, while its ranking knowledge is distilled online into a lightweight scoring network during joint training, enabling real-time and efficient inference. Furthermore, we propose a User Profile Network (UPN) that injects user intent and models interest dynamics, enabling deeper interactions between users and items. Extensive experiments conducted on three large-scale public datasets demonstrate that PSAD significantly outperforms state-of-the-art baselines in both ranking performance and inference efficiency.
Figures & tables
Figure 1: Two challenges faced by generative reranking methods.
Figure 2: The overall framework of PSAD and its sub-modules.
Dataset
#Users
#Items
#Records
Ad
1.06×106
8.27×105
2.04×106
PRM Public
7.44×105
7.25×106
1.44×107
Avito
1.32×106
2.36×107
5.36×107
Table 1: Dataset statistics.
Datasets
Ad
PRM Public
Avtio
Metric
N@5
N@10
M@5
M@10
N@5
N@10
M@5
M@10
N@5
M@5
miDNN
0.6819
0.6952
0.6025
0.6063
0.2620
0.3341
0.2842
0.2954
0.6737
0.6838
DLCM
0.6857
0.6989
0.6064
0.6100
0.2664
0.3395
0.2918
0.3023
0.6912
0.7057
PRM
0.6872
0.7015
0.6080
0.6117
0.2689
0.3430
0.2937
0.3042
0.7047
0.7174
Seq2Slate
0.6916
0.7062
0.6119
0.6155
0.2709
0.3449
0.2959
0.3061
0.7245
0.7360
CRUM
0.6899
0.7037
0.6084
0.6112
0.2703
0.3445
0.2952
0.3056
0.7110
0.7266
Table 2: Overall performance comparison of all methods in terms of NDCG@K and MAP@K (K=5, 10). (p-value < 0.05)
Batch Size
64
128
256
Model
Train(↓)
Inf.(↓)
Train(↓)
Inf.(↓)
Train(↓)
Inf.(↓)
miDNN
0.0895
0.0764
0.1425
0.1256
0.2132
0.1893
DLCM
0.2799
0.1497
0.4301
0.2409
0.6827
0.3940
PRM
0.1238
0.1014
0.1852
0.1360
0.2806
0.2037
Seq2Slate
0.3142
0.2176
0.5114
0.3993
0.8384
0.5873
CRUM
0.2021
0.1589
0.3185
0.2850
0.4415
0.3048
Table 3: The training and inference time comparison between PSAD and baseline methods on the PRM Public dataset. The comparison is conducted under different batch sizes, and the time is calculated by averaging the results over 10 steps. The time unit is seconds.
Datasets
Ad
PRM Public
Avito
Metric
N@10
M@10
N@10
M@10
N@5
M@5
w/o sa
0.7095
0.6192
0.3455
0.3054
0.7145
0.7295
w/o ce
0.7133
0.6220
0.3482
0.3096
0.7231
0.7360
w/o sace
0.7042
0.6139
0.3441
0.3052
0.7103
0.7217
w/o ppe
0.7168
0.6248
0.3523
0.3128
0.7384
0.7546
w/o pg
0.7136
0.6224
0.3507
0.3110
0.7402
0.7543
Table 4: Ablation analysis of PSAD-G
Figure 3: Hyper-Parameter Performance
Dataset
Ad
PRM
Teacher Model
N@10(↑)
Train(↓)
N@10(↑)
Train(↓)
AR(online)
0.7078
0.4896
0.3434
0.7372
NAR(online)
0.7046
0.3004
0.3429
0.4165
SAR(offline)
0.7083
0.6547
0.3481
0.8933
SAR(online)
0.7091
0.3752
0.3464
0.5128
Table 5: Comparison of different distillation methods in terms of performance (NDCG@10) and training time. The batch size is set to 256, and the time is reported as the average over 10 training steps. The time unit is seconds.
Figure 4: Performance comparison of item–user feature interaction mechanisms across user sequences with varying activity levels.
Kuaishou serves over 400 million daily active users, processing hundreds of millions of search queries daily against a repository of tens of billions of short videos. As the final decision layer, the reranking stage determines user experience by optimizing whole-page utility. While traditional score-and-sort methods fail to capture combinatorial dependencies, Generative Reranking offers a superior paradigm by directly modeling the permutation probability. However, deploying Generative Reranking in such a high-stakes environment faces a fundamental dual dilemma: 1) the structural trade-off where Autoregressive (AR) models offer superior Sequential modeling but suffer from prohibitive latency, versus Non-Autoregressive (NAR) models that enable efficiency but lack dependency capturing; 2) the optimization gap where Supervised Learning faces challenges in directly optimizing whole-page utility, while Reinforcement Learning (RL) struggles with instability in high-throughput data streams. To resolve this, we propose Dual-Rerank, a unified framework designed for industrial reranking that bridges the structural gap via Sequential Knowledge Distillation and addresses the optimization gap using List-wise Decoupled Reranking Optimization (LDRO) for stable online RL. Extensive A/B testing on production traffic demonstrates that Dual-Rerank achieves State-of-the-Art performance, significantly improving User satisfaction and Watch Time while drastically reducing inference latency compared to AR baselines.
Chao Zhang, Shuai Lin, ChengLei Dai +5
Kuaishou Technology, Hangzhou, China · Kuaishou Technology, Beijing, China · Unaffiliated, Beijing, China
In multi-stage recommender systems, reranking optimizes overall utility by capturing intra-list contextual dependencies, yet its central challenge lies in exploring optimal sequences within an exponentially large permutation space. Recent studies have shifted towards end-to-end generative frameworks, which typically leverage list-wise rewards or preference alignment to guide generator training. However, these methods still face two critical issues. First is the heuristic label bias. Existing methods often construct training targets based on simple rules, such as promoting clicked items to the top, while ignoring causal dependencies within the list context. Second is the credit assignment problem. Sparse list-level posterior rewards fail to directly guide intermediate steps in sequence generation, leading to ambiguous optimization directions. To address these issues, we propose DeGRe (Dense-supervised Generative Reranking), a generative reranking framework that bridges the gap between offline exploration and online efficiency through dense supervision. The core of DeGRe lies in its offline-online decoupled design. During the offline phase, we introduce a Lookahead Evaluator based on cumulative regression, which leverages beam search to actively mine high-value lookahead sequences in the unexposed space. During training, we transform the step-wise value estimations from the evaluator into dense supervision signals and distill them into a lightweight Online Generator. This mechanism enables the generator to internalize lookahead planning capabilities, requiring only a single efficient greedy decoding pass during online inference to approximate the global optimum. Experiments demonstrate that DeGRe outperforms baseline models on public benchmarks and industrial datasets. We have successfully deployed DeGRe on Taobao Flash Shopping, significantly improving online recommendations.
Chaotian Song, Jingyao Zhang, Chenghao Chen +6
College of Software, Zhejiang University Hangzhou, China · Rajax Network Technology, Taobao Shangou of Alibaba Hangzhou, China · Rajax Network Technology, Taobao Shangou of Alibaba Beijing, China +1
Recent studies increasingly explore Large Language Models (LLMs) as a new paradigm for recommendation systems due to their scalability and world knowledge. However, existing work has three key limitations: (1) most efforts focus on retrieval and ranking, while the reranking phase, critical for refining final recommendations, is largely overlooked; (2) LLMs are typically used in zero-shot or supervised fine-tuning settings, leaving their reasoning abilities, especially those enhanced through reinforcement learning (RL) and high-quality reasoning data, underexploited; (3) items are commonly represented by non-semantic IDs, creating major scalability challenges in industrial systems with billions of identifiers. To address these gaps, we propose the Generative Reasoning Reranker (GR2), an end-to-end framework with a three-stage training pipeline tailored for reranking. First, a pretrained LLM is mid-trained on semantic IDs encoded from non-semantic IDs via a tokenizer achieving ≥99% uniqueness. Next, a stronger larger-scale LLM generates high-quality reasoning traces through carefully designed prompting and rejection sampling, which are used for supervised fine-tuning to impart foundational reasoning skills. Finally, we apply Decoupled Clip and Dynamic sAmpling Policy Optimization (DAPO), enabling scalable RL supervision with verifiable rewards designed specifically for reranking. Experiments on two real-world datasets demonstrate GR2's effectiveness: it surpasses the state-of-the-art OneRec-Think by 2.4% in Recall@5 and 1.3% in NDCG@5. Ablations confirm that advanced reasoning traces yield substantial gains across metrics. We further find that RL reward design is crucial in reranking: LLMs tend to exploit reward hacking by preserving item order, motivating conditional verifiable rewards to mitigate this behavior and optimize reranking performance.