Yambda-5B -- A Large-Scale Multi-modal Dataset for Ranking And Retrieval
Not on the map. Why?
Abstract (arXiv)
We present Yambda-5B, a large-scale open dataset sourced from the Yandex Music streaming platform. Yambda-5B contains 4.79 billion user-item interactions from 1 million users across 9.39 million tracks. The dataset includes two primary types of interactions: implicit feedback (listening events) and explicit feedback (likes, dislikes, unlikes and undislikes). In addition, we provide audio embeddings for most tracks, generated by a convolutional neural network trained on audio spectrograms. A key distinguishing feature of Yambda-5B is the inclusion of the is_organic flag, which separates organic user actions from recommendation-driven events. This distinction is critical for developing and evaluating machine learning algorithms, as Yandex Music relies on recommender systems to personalize track selection for users. To support rigorous benchmarking, we introduce an evaluation protocol based on a Global Temporal Split, allowing recommendation algorithms to be assessed in conditions that closely mirror real-world use. We report benchmark results for standard baselines (ItemKNN, iALS) and advanced models (SANSA, SASRec) using a variety of evaluation metrics. By releasing Yambda-5B to the community, we aim to provide a readily accessible, industrial-scale resource to advance research, foster innovation, and promote reproducible results in recommender systems.
Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 18 of its 36 references that are in the map. First posted on arXiv 28 May 2025.
Builds on
18 of 36 references.
References · 18
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,501
- 2014Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modelingcited by 2,367
- 2018Self-Attentive Sequential Recommendationcited by 1,184
- 2019BERT4Rec: Sequential Recommendation with Bidirectional Encoder Representations from Transformercited by 964
- 2015Image-based Recommendations on Styles and Substitutescited by 706
- 2015Session-based Recommendations with Recurrent Neural Networkscited by 695
- 2017Deep & Cross Network for Ad Click Predictionscited by 393
- 2020DCN V2: Improved Deep & Cross Network and Practical Lessons for Web-scale Learning to Rank Systemscited by 327
- 2024Bridging Language and Items for Retrieval and Recommendation: Benchmarking LLMs as Semantic Encoderscited by 224
6 more
- 2019Embarrassingly Shallow Autoencoders for Sparse Datacited by 94
- 2021Contrastive Learning of Musical Representationscited by 82
- 2024Wukong: Towards a Scaling Law for Large-Scale Recommendationcited by 74
- 2022Understanding Scaling Laws for Recommendation Modelscited by 38
- 2023Scaling Law of Large Sequential Recommendation Modelscited by 23
- 2020A Re-visit of the Popularity Baseline in Recommender Systemscited by 18
Built on by
Citing papers per year: 2026: 5.
5 papers.
- 2026Variable-Length Semantic IDs for Recommender Systemscited by 4
- 2026Recommendation Retrievers Need Verifiers: Universal Generative Reranking for Sequential Recommendationscited by 1
- 2026Position-Aware Sequential Attention for Accurate Next Item Recommendationscited by 0
- 2026Cross-Representation Knowledge Transfer for Improved Sequential Recommendationscited by 0
- 2026Cut Binary Cross Entropy: Efficient Large-Vocabulary Loss and Gradient Kernels for Sequential Recommendationcited by 0
- 2026Cut Binary Cross Entropy: Efficient Large-Vocabulary Loss and Gradient Kernels for Sequential Recommendation4 Oct 2026
- 2026Recommendation Retrievers Need Verifiers: Universal Generative Reranking for Sequential Recommendations10 Sep 2026
- 2026Cross-Representation Knowledge Transfer for Improved Sequential Recommendations26 Feb 2026
- 2026Position-Aware Sequential Attention for Accurate Next Item Recommendations24 Feb 2026
- 2026Variable-Length Semantic IDs for Recommender Systems18 Feb 2026