TurboGR: An Accelerated Training System for Large-Scale Generative Recommendation
Not on the map. Why?
Abstract (arXiv)
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).
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 14 of its 19 references that are in the map. First posted on arXiv 13 May 2026.
Builds on
14 of 19 references.
References · 14
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2018Self-Attentive Sequential Recommendationcited by 1,169
- 2024Actions Speak Louder than Words: Trillion-Parameter Sequential Transducers for Generative Recommendationscited by 184
- 2024Wukong: Towards a Scaling Law for Large-Scale Recommendationcited by 72
- 2023Generative Recommendation: Towards Next-generation Recommender Paradigmcited by 69
- 2021Personalized News Recommendation: Methods and Challengescited by 30
- 2022Building a Performance Model for Deep Learning Recommendation Model Training on GPUscited by 10
- 2025FuXi-α: Scaling Recommendation Model with Feature Interaction Enhanced Transformercited by 10
- 2023Analyzing User Engagement with TikTok's Short Format Video Recommendations using Data Donationscited by 8
- 2025A Survey of Real-World Recommender Systems: Challenges, Constraints, and Industrial Perspectivescited by 7
- 2024Scenario-Wise Rec: A Multi-Scenario Recommendation Benchmarkcited by 3
Built on by
Not cited yet.