arXiv:2503.24193cs.IR2025-03被引 12

用生成式检索直接从提示词推荐歌曲,更准更快。

Text2Tracks: Prompt-based Music Recommendation via Generative Retrieval

  • 将歌曲推荐转为生成式检索,直接输出歌曲ID
  • 语义化歌曲标识符效果远超传统标题标识
  • 适合需要精准自然语言推荐的音乐系统

近年来,大语言模型(LLMs)使用户能通过自然语言提示(如“推荐一些适合慢舞的老经典”)提出高度具体的音乐推荐请求。当前方法采用自回归方式由LLM逐词生成歌曲标题,但存在三大局限:一是通用分词器对歌曲标题不优化;二是需额外实体解析层匹配标题与真实歌曲标识;三是解码步数随标题长度线性增长,影响推理速度。本文提出将提示词驱动的音乐推荐重构为生成式检索任务,设计新型高效的歌曲标识符表示,显著优于常见策略。我们构建Text2Tracks模型,学习从用户提示直接映射到相关歌曲ID。在包含语言输入的播放列表数据集上离线评估发现:(1) 歌曲标识符的设计是决定Text2Tracks效果的关键因素,语义化标识符明显优于依赖歌曲标题的常用策略;(2) 在选择合适标识符的前提下,Text2Tracks优于基于稀疏和密集检索训练的歌曲推荐方案。

原文摘要 · Abstract (English)

In recent years, Large Language Models (LLMs) have enabled users to provide highly specific music recommendation requests using natural language prompts (e.g. "Can you recommend some old classics for slow dancing?"). In this setup, the recommended tracks are predicted by the LLM in an autoregressive way, i.e. the LLM generates the track titles one token at a time. While intuitive, this approach has several limitation. First, it is based on a general purpose tokenization that is optimized for words rather than for track titles. Second, it necessitates an additional entity resolution layer that matches the track title to the actual track identifier. Third, the number of decoding steps scales linearly with the length of the track title, slowing down inference. In this paper, we propose to address the task of prompt-based music recommendation as a generative retrieval task. Within this setting, we introduce novel, effective, and efficient representations of track identifiers that significantly outperform commonly used strategies. We introduce Text2Tracks, a generative retrieval model that learns a mapping from a user's music recommendation prompt to the relevant track IDs directly. Through an offline evaluation on a dataset of playlists with language inputs, we find that (1) the strategy to create IDs for music tracks is the most important factor for the effectiveness of Text2Tracks and semantic IDs significantly outperform commonly used strategies that rely on song titles as identifiers (2) provided with the right choice of track identifiers, Text2Tracks outperforms sparse and dense retrieval solutions trained to retrieve tracks from language prompts.

音乐推荐生成检索大模型应用

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。