让AI更聪明地听歌推荐,该用内藏知识时用,该调工具时调。
WeMusic-Agent: Efficient Conversational Music Recommendation via Knowledge Internalization and Agentic Boundary Learning
- 通过内化音乐知识与学习调用外部工具的边界,让模型自主决策何时用内部知识、何时调工具。
- 在真实微信听歌数据上,推荐相关性、个性化和多样性均显著优于现有模型。
- 开源了首个基于真实数据的对话式音乐推荐基准,支持多维度评估。
对话式个性化音乐推荐需深刻理解用户偏好与音乐上下文,但现有方法常难以兼顾领域专知与灵活工具调用。本文提出 WeMusic-Agent 框架,用于高效训练基于大模型的对话式音乐推荐系统。该框架融合知识内化与代理边界学习,使模型能智能判断何时使用内化知识、何时调用外部工具(如音乐检索API、推荐系统)。基于此,我们构建了 WeMusic-Agent-M1 模型,通过在500亿条音乐相关语料上持续预训练,内化丰富音乐知识,同时具备必要时调用外部工具的能力。此外,针对对话式音乐推荐缺乏开源基准的问题,我们基于微信听歌真实数据构建了一个个性化推荐评测基准,支持对推荐结果的相关性、个性化和多样性进行多维度评估。在真实数据上的实验表明,WeMusic-Agent 显著优于现有模型。
原文摘要 · Abstract (English)
Personalized music recommendation in conversational scenarios usually requires a deep understanding of user preferences and nuanced musical context, yet existing methods often struggle with balancing specialized domain knowledge and flexible tool integration. This paper proposes WeMusic-Agent, a training framework for efficient LLM-based conversational music recommendation. By integrating the knowledge internalization and agentic boundary learning, the framework aims to teach the model to intelligently decide when to leverage internalized knowledge and when to call specialized tools (e.g., music retrieval APIs, music recommendation systems). Under this framework, we present WeMusic-Agent-M1, an agentic model that internalizes extensive musical knowledge via continued pretraining on 50B music-related corpus while acquiring the ability to invoke external tools when necessary. Additionally, considering the lack of open-source benchmarks for conversational music recommendation, we also construct a benchmark for personalized music recommendations derived from real-world data in WeChat Listen. This benchmark enables comprehensive evaluation across multiple dimensions, including relevance, personalization, and diversity of the recommendations. Experiments on real-world data demonstrate that WeMusic-Agent achieves significant improvements over existing models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。