Taste-aware music retrieval from audio embeddings
📄 Taste-aware music retrieval from audio embeddings #音乐检索 #预训练 #多任务学习 #可解释性 6.9/10 | 创新 1/2 | 严谨 1.1/1.5 | 实验 0.9/1.5 | 清晰 0.8/1 | 影响 0.4/1.5 | 开源 1.3/1.5 | 复现 0.4/0.5 | 工程 1/1.5 ✅ 6.9/10 | 前50% | #音乐检索 | #预训练 | #多任务学习 #可解释性 | arxiv 👥 作者与机构 第一作者:Matteo Spanio(帕多瓦大学,CSC实验室) 通讯作者:未说明 作者列表:Matteo Spanio(帕多瓦大学,CSC实验室)、Antonio Rodà(帕多瓦大学,CSC实验室) 💡 毒舌点评 本文把一个已有心理学根基的“声‑味对应”任务做实成了可复现、可检索的MIR基准,单编码器就能把误差压到单人评估者的一半,这个结论很漂亮。但40项的测试集让几乎所有的融合收益都落在统计噪声里,检索实验的CLAP-text完全失效更像是prompt问题而非方法本质优势,对“辣味”这一缺乏心理学支撑的轴也没有充分辩护。 📌 核心摘要 论文将“从音频预测味觉”形式化为一个内容化音乐信息检索(MIR)基准,使用经过感知验证的多源语料库,预测甜、苦、咸、酸、辣五种味觉强度。 方法核心是冻结预训练音频编码器提取嵌入,经时间池化后送入一个共享的、带sigmoid输出的多层感知机进行多任务回归,损失为掩码MSE,并引入门控晚期融合来组合多编码器。 相比先前需微调五个独立AST回归器的工作,本文用一个多任务头取代五个独立头,并配合掩码损失和sigmoid输出,在极简的冻结编码器设置下大幅降低绝对误差,并额外增加了检索评估和心理学驱动的可解释性分析。 最佳系统(单VGGish或融合组合)在测试集上达到宏观RMSE 0.134,在真实音乐子集上误差(RMSE 0.13)不到单个人类评估者与共识偏差(RMSE 0.28)的一半;门控晚期融合将宏观Pearson \(r\) 从0.666提升到0.724;在309项检索池中,基于味觉向量检索的精度完全饱和,而CLAP-text基线几乎随机。 方法 宏观RMSE↓ 宏观MAE↓ 宏观Pearson \(r\) ↑ VGGish (单编码器) 0.134 0.109 0.666 VGGish+MULE (融合) 0.134 0.111 0.724 SOTA (AST 5头) 0.219 0.175 0.556 配置 RMSE↓ MAE↓ 宏观\(r\) ↑ SOTA (微调AST, 无界MSE) 0.219 0.175 0.556 +冻结AST, 每味MLP, 掩码MSE, sigmoid 0.143 0.115 0.663 +共享多任务头 0.143 0.116 0.658 +门控晚期融合 (VGGish+MULE) 0.134 0.111 0.724 实际意义在于为音乐推荐系统提供了一个可解释的“味觉”语义轴,可支持“相似但更甜”这类检索,且模型误差已低于普通标注者,有替代或辅助人工评分的潜力。 主要局限是样本量极小(训练269项,测试40项),导致统计效力不足,尤其是融合效果的显著性难以保证;辣味轴缺乏类似甜/苦那样的跨模态对应实证;跨文化泛化未验证。 🔗 开源详情 代码:https://github.com/CSCPadova/wav2taste 模型权重:训练好的 taste 预测头(task-specific heads)随代码仓库提供(位于 https://github.com/CSCPadova/wav2taste);使用的预训练音频编码器权重来自各开源项目(见下文“论文中引用的开源项目”)。 数据集:https://huggingface.co/datasets/csc-unipd/sonic-seasoning (sonic-seasoning 统一音乐‑味觉语料库) Demo:论文中未提及 复现材料:代码仓库提供训练与评估脚本;论文中给出训练超参数(AdamW,lr \(=10^{-3}\),weight decay \(=10^{-4}\),batch size \(=32\),max \(50\) epochs,patience \(10\) on validation macro \(r\),multi-task MLP head hidden \(256\),dropout \(0.2\),sigmoid output,\(5\) seeds \(\{11,22,33,44,55\}\)),冻结编码器缓存策略,分析探针配置(ridge \(\alpha=1.0\),\(5\)-fold CV)等。 论文中引用的开源项目: HEAR benchmark:https://hearbenchmark.com VGGish:https://github.com/tensorflow/models/tree/master/research/audioset/vggish PANNs:https://github.com/qiuqiangkong/audioset_tagging_cnn AST (Audio Spectrogram Transformer):https://github.com/YuanGongND/ast HuBERT:https://github.com/pytorch/fairseq MERT:https://huggingface.co/m-a-p/MERT-v1-330M CLAP:https://github.com/LAION-AI/CLAP EnCodec:https://github.com/facebookresearch/encodec MULE:https://github.com/mule-project/mule (推测地址,论文引用[22]) librosa:https://github.com/librosa/librosa MusicGen (Audiocraft):https://github.com/facebookresearch/audiocraft FMA (Free Music Archive):https://github.com/mdeff/fma Omar-RQ:结合论文上下文应为基于 EnCodec 的离散自监督模型,常与 EnCodec 关联,未找到独立官方仓库(可能指 Omar 等人提出的残差量化变体,与 EnCodec 同源) 🏗️ 方法概述和架构 本文构建了一个“从音频嵌入预测味觉”的冻结编码器‑多任务回归框架,并在此基础上附加门控晚期融合、可解释性探测和检索评估。 ...