arXiv:2507.12996cs.SD2025-07被引 2

用双类标记融合对比与等变学习,提升音乐信息检索的多任务表现

Multi-Class-Token Transformer for Multitask Self-supervised Music Information Retrieval

  • 设计双类标记ViT-1D模型,分别优化对比学习和等变学习任务
  • 在多个音乐识别任务上优于单任务模型,平均性能更优
  • 参数量仅为MERT的1/18,适合资源受限场景

对比学习和等变学习是音频内容分析中有效的自监督学习方法。然而它们在音乐信息检索(MIR)中的应用面临两难:前者在标签任务(如乐器识别)上表现更好,但在结构化预测(如调性估计)上较弱;后者在特定任务上可媲美有监督方法,但泛化能力差。本文提出一种融合两者优势的方法,同时训练深度神经网络于两种预训练任务。新架构为一维频谱图块的视觉变换器(ViT-1D),配备两个专用类标记,分别用于不同自监督任务,但共享同一模型:即自监督多类标记多任务(MT2)。前一类标记优化圆五度上的交叉功率谱密度(CPSD)以实现等变学习,后一标记优化归一化温度缩放交叉熵(NT-Xent)以实现对比学习。MT2结合两种任务的优势,在多个任务上持续超越仅采用对比或等变学习的单类标记模型。进一步平均两个类标记的输出,可在多个任务上获得更高性能,凸显两类表示的互补性。此外,使用相同的单层线性探测方法,MT2在所有任务上优于MERT,除节拍跟踪外;且参数量仅为其1/18,得益于其多任务能力。本自监督基准验证了该多类标记多任务学习方法在MIR应用中的通用性。

原文摘要 · Abstract (English)

Contrastive learning and equivariant learning are effective methods for self-supervised learning (SSL) for audio content analysis. Yet, their application to music information retrieval (MIR) faces a dilemma: the former is more effective on tagging (e.g., instrument recognition) but less effective on structured prediction (e.g., tonality estimation); The latter can match supervised methods on the specific task it is designed for, but it does not generalize well to other tasks. In this article, we adopt a best-of-both-worlds approach by training a deep neural network on both kinds of pretext tasks at once. The proposed new architecture is a Vision Transformer with 1-D spectrogram patches (ViT-1D), equipped with two class tokens, which are specialized to different self-supervised pretext tasks but optimized through the same model: hence the qualification of self-supervised multi-class-token multitask (MT2). The former class token optimizes cross-power spectral density (CPSD) for equivariant learning over the circle of fifths, while the latter optimizes normalized temperature-scaled cross-entropy (NT-Xent) for contrastive learning. MT2 combines the strengths of both pretext tasks and outperforms consistently both single-class-token ViT-1D models trained with either contrastive or equivariant learning. Averaging the two class tokens further improves performance on several tasks, highlighting the complementary nature of the representations learned by each class token. Furthermore, using the same single-linear-layer probing method on the features of last layer, MT2 outperforms MERT on all tasks except for beat tracking; achieving this with 18x fewer parameters thanks to its multitasking capabilities. Our SSL benchmark demonstrates the versatility of our multi-class-token multitask learning approach for MIR applications.

自监督学习音乐信息检索多任务Transformer

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