Yambda-5B是千万级用户、47.9亿次交互的音乐推荐数据集,支持真实场景下的算法评测。
Yambda-5B -- A Large-Scale Multi-modal Dataset for Ranking And Retrieval
- 基于真实音乐平台数据,区分自然行为与推荐驱动事件。
- 包含47.9亿次隐式/显式反馈,覆盖939万首曲目和100万用户。
- 提供音频嵌入与时间分割评估协议,适合推荐系统研究者使用。
我们提出 Yambda-5B,一个来自 Yandex Music 平台的大规模开放数据集。该数据集包含来自 100 万用户对 939 万首曲目的 47.9 亿次用户-物品交互记录,涵盖隐式反馈(播放事件)和显式反馈(点赞、点踩、取消点赞、取消点踩)。此外,我们为大多数曲目提供了由卷积神经网络在音频频谱图上训练生成的音频嵌入。数据集关键特性是包含 is_organic 标志,可区分用户自然行为与推荐系统引发的交互,这对开发和评估机器学习算法至关重要。为支持严谨基准测试,我们引入基于全局时间划分的评估协议,使推荐算法能在更贴近真实场景的条件下被评估。我们报告了标准基线(ItemKNN、iALS)和先进模型(SANSA、SASRec)在多种评估指标下的基准结果。通过公开 Yambda-5B,我们旨在为社区提供一个可直接使用的工业级资源,推动推荐系统研究发展,促进创新与可复现性。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。