Recommender systems (RecSys) are increasingly emphasizing scaling, leveraging larger architectures and more interaction data to improve personalization. Yet, despite the optimizer's pivotal role in training, modern RecSys pipelines almost universally default to Adam/AdamW, with limited scrutiny of whether these choices are truly optimal for recommendation. In this work, we revisit optimizer design for scalable recommendation and introduce MuonRec, the first framework that brings the recently proposed Muon optimizer to RecSys training. Muon performs orthogonalized momentum updates for 2D weight matrices via Newton-Schulz iteration, promoting diverse update directions and improving optimization efficiency. We develop an open-source training recipe for recommendation models and evaluate it across both traditional sequential recommenders and modern generative recommenders. Extensive experiments demonstrate that MuonRec reduces converged training steps by an average of 32.4% while simultaneously improving final ranking quality. Specifically, MuonRec yields consistent relative gains in NDCG@10, averaging 12.6% across all settings, with particularly pronounced improvements in generative recommendation models. These results consistently outperform strong Adam/AdamW baselines, positioning Muon as a promising new optimizer standard for RecSys training. Our code is available.
Figures & tables
Figure 1. Training dynamics of TIGER ( Rajput et al., 2024 ) on Industrial (left) and Office (right) datasets. The top row displays training loss, while the bottom row reports the NDCG@10 curves. The Muon optimizer demonstrates significantly faster convergence and superior final performance compared to Adam . Training dynamics of the TIGER model on Industrial (top row) and Office (bottom row) datasets. The left column displays the Training Loss, while the right column reports the NDCG@10 curves. The Muon optimizer (red) demonstrates significantly faster convergence and superior final performance compared to Adam (blue), validating its efficiency and effectiveness for generative recommendation.
Type
Model
Opt.
Industrial Dataset
Official Dataset
Step ↓
R@1
R@3
R@5
R@10
N@3
N@5
N@10
Step ↓
R@1
R@3
R@5
R@10
N@3
N@5
N@10
Trad.
GRU4Rec
Adam
110
0.0516
0.0746
0.0863
0.1054
0.0651
0.0699
0.0761
142
0.0506
0.0816
0.0960
0.1210
0.0687
0.0746
0.0826
Muon
44
0.0598
0.0869
0.0971
0.1163
0.0753
0.0794
0.0855
86
0.0705
0.1007
0.1099
0.1311
0.0881
0.0919
0.0987
Improv.
60.0%
15.9%
16.5%
12.5%
10.3%
15.7%
13.6%
12.4%
39.4%
39.3%
23.4%
14.5%
8.3%
28.2%
23.2%
19.5%
SASRec
Adam
117
0.0576
0.0805
0.0931
0.1141
0.0713
0.0764
0.0831
153
0.0717
0.0935
0.1060
0.1262
0.0847
0.0898
0.0963
Muon
103
0.0569
0.0810
0.0944
0.1127
0.0713
0.0769
0.0827
117
0.0732
0.0962
0.1087
0.1270
0.0866
0.0918
0.0977
Table 1. Performance comparison with Adam on Industrial and Official datasets. For MiniOneRec models, we report the converged training step during the SFT+RL phase. Best results are in bold . N@ K : NDCG@ K . R@ K : Recall@ K .
Opt.
Step ↓
Recall
NDCG
SFT
RL
R@1
R@3
R@5
R@10
N@3
N@5
N@10
Dataset: Industrial
Adam
Adam
3345
0.0781
0.1046
0.1160
0.1372
0.0933
0.0980
0.1048
Adam
Muon
3015
0.0748
0.1026
0.1176
0.1410
0.0907
0.0968
0.1045
Muon
Adam
2685
0.0796
0.1079
0.1231
0.1423
0.0959
0.1021
0.1083
Muon
Muon
2355
0.0788
0.1066
0.1253
0.1460
0.0949
0.1026
0.1092
Table 2. Optimizer combinations for SFT and RL stages. Step denotes the total converged training steps of both stages. Bold : best results; underlined : second best results.
Figure 2. Impact of LR on NDCG@10 and training loss.
Generative Retrieval (GR) offers a promising paradigm for recommendation through next-token prediction (NTP). However, scaling it to large-scale industrial systems introduces three challenges: (i) within a single request, the identical model inputs may produce inconsistent outputs due to the pagination request mechanism; (ii) the prohibitive cost of encoding long user behavior sequences with multi-token item representations based on semantic IDs, and (iii) aligning the generative policy with nuanced user preference signals. We present GenRec, a preference-oriented generative framework deployed on the JD App that addresses above challenges within a single decoder-only architecture. For training objective, we propose Page-wise NTP task, which supervises over an entire interaction page rather than each interacted item individually, providing denser gradient signal and resolving the one-to-many ambiguity of point-wise training. On the prefilling side, an asymmetric linear Token Merger compresses multi-token Semantic IDs in the prompt while preserving full-resolution decoding, reducing input length by ~2X with negligible accuracy loss. To further align outputs with user satisfaction, we introduce GRPO-SR, a reinforcement learning method that pairs Group Relative Policy Optimization with NLL regularization for training stability, and employs Hybrid Rewards combining a dense reward model with a relevance gate to mitigate reward hacking. In month-long online A/B tests serving production traffic, GenRec achieves 9.5% improvement in click count and 8.7% in transaction count over the existing pipeline.
Yanyan Zou, Junbo Qi, Lunsong Huang +7
JD.com Beijing, China · Waseda University Tokyo, Japan
Generative recommendation (GR) has emerged as a promising paradigm that replaces fragmented, scenario-specific architectures with unified Transformer-based models, exhibiting scaling-law behavior where recommendation quality improves systematically with increased model capacity and training data. However, deploying GR at scale on Ascend NPUs faces fundamental system-level challenges. These challenges are further exacerbated on Ascend NPUs due to the absence of high-performance implementations for jagged operators and the architectural mismatch between irregular sparse primitives and NPU's dense-computation-optimized design. In this paper, we present \model, an Ascend-affinity training system for generative recommendation that systematically addresses these bottlenecks through three core innovations: (i) Ascend-affinity jagged acceleration, including fusion operators that eliminate padding redundancy and dynamic load balancing that reduces inter-device imbalance from 47% to 2.4%; (ii) distributed communication optimization, comprising hierarchical sparse parallelism, semi-asynchronous training with proven convergence guarantees, and fine-grained pipeline orchestration that sustains 94% NPU utilization; and (iii) negative sampling optimization via asynchronous offloading, jaggedness-aware FP16 quantization, and intra-batch logit sharing that expand the effective negative space without additional embedding lookups. Evaluated on the KuaiRand-27K dataset, \model supports training at up to 0.2B parameters and achieves 54.71% MFU with near-linear scalability (0.97).
Sequential recommender systems typically infer user preferences through single-pass encoding of interaction histories without iterative refinement, relying on increasingly deep architectures to capture complex patterns. In this work, we revisit sequential recommendation from a recursive inference perspective: can user preferences be modeled as a persistent latent state that is recursively refined? We propose RecRec (Recursive Recommendation), a lightweight model that maintains a compact latent state and updates it through a shared recursive module conditioned on interaction evidence. Unlike prior recursive models, RecRec introduces an evidence-anchored correction mechanism that stabilizes refinement by grounding each update in the original interaction context, preventing semantic drift during deep recursive reasoning. Experiments on three benchmark datasets under standard evaluation protocols show that RecRec matches or outperforms state-of-the-art sequential, graph-based, and reasoning-enhanced recommenders while using only 3.9M to 14M parameters. Ablation studies demonstrate that both recursive refinement and the evidence-anchored correction gate contribute significantly to performance, highlighting the effectiveness of recursive latent inference as a scalable alternative to deeper or language-based architectures. Code is available at https://anonymous.4open.science/r/RecRec-6B67/README.md.