用二维频谱图学习音乐表征,提升音高和节奏识别能力
Frequency-Aware Self-Supervised Music Representation Learning

- 直接在二维频谱图上训练,通过预测被遮蔽块的嵌入来学习
- 在MARBLE基准上多项任务超越传统一维模型,线性探测表现更好
- 适合研究音乐结构、声学特征或想改进音频表征的开发者
自监督学习(SSL)已成为音乐信息检索(MIR)的重要范式。现有模型通常将音频视为一维序列,处理时域波形或展平的时频谱图,忽略了时频表示中的空间与结构信息,也违背了音乐制作中的基本直觉。音乐在基于MIDI的工作流中天然以时频网格形式存在,这一结构与二维谱图高度对应,并使许多MIR任务变得简单。受此启发,我们提出PupuJEPA,一种直接在二维谱图上训练的视觉联合嵌入预测架构(JEPA)。不同于对一维序列使用掩码语言建模,PupuJEPA通过未遮蔽上下文预测被遮蔽的二维谱图块的隐向量来学习鲁棒表征。为适配音乐信号,我们对模型结构、训练方案和推理范式进行了领域特定优化,消融实验验证其有效性。在MARBLE基准上的评估显示,PupuJEPA在多种MIR任务的线性探测中优于基于一维序列的SSL模型。注意力图案例分析进一步表明,该模型捕捉到了时频域内具有音乐意义的模式。
原文摘要 · Abstract (English)
Self-supervised learning (SSL) has emerged as an essential paradigm for music information retrieval (MIR). While current SSL models achieve state-of-the-art performance across various MIR tasks, they typically treat audio as 1D sequences, either operating on time-domain waveforms or on flattened time-frequency-domain spectrograms. This discards the rich spatial and structural information in time-frequency representations and overlooks a fundamental intuition in music production. In particular, music is naturally represented as time-frequency grids in MIDI-based workflows, a structure that tightly corresponds to 2D spectrograms and inherently makes many MIR tasks trivial. Motivated by this intuition, we propose PupuJEPA, a visual Joint-Embedding Predictive Architecture (JEPA) that is trained directly on 2D spectrograms. Instead of applying masked language modeling (MLM) to 1D sequences, PupuJEPA learns robust representations by predicting the latent embeddings of masked 2D spectrogram patches from unmasked contexts. To optimally adapt such a visual framework to music signals, we also apply domain-specific modifications to model architecture, training scheme, and inference paradigm, with comprehensive ablation studies showing their effectiveness. Evaluations on the MARBLE benchmark show that PupuJEPA outperforms the 1D sequence-based SSL models across multiple MIR tasks in linear probing. Additionally, case studies of the attention maps also confirm that PupuJEPA captures musically meaningful patterns within the 2D time-frequency domain. Codes and checkpoints are available at: https://www.yichenggu.com/PupuJEPA/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。