用遮蔽策略提升音乐表征学习,效率与精度双突破
Myna: Masking-Based Contrastive Learning of Musical Representations
- 用视觉变换器+频谱令牌遮蔽,替代传统数据增强
- 单卡训练达4096批量,性能超越6200万参数模型
- 适合追求高效高精度音乐分析的研究者使用
我们提出Myna,一种基于对比学习的自监督音乐表征学习方法。其核心创新包括:(1)采用视觉变换器(ViT)处理梅尔频谱图作为主干网络;(2)引入新型数据增强策略——令牌遮蔽,遮蔽90%的频谱令牌。该设计显著提升效率:每GPU批量从此前方法(CLMR、MULE)的48或120提升至4096。通过避免传统增强,模型保留音高敏感性,提升调性检测等任务表现。垂直分块设计有助于捕捉关键特征。混合模型Myna-22M-Hybrid同时处理16x16和128x2块,在单卡上训练,平均性能超过MULE(62M),媲美在16/64张GPU上训练的MERT-95M,且优于公开可用的MERT-95M-public,成为当前公开数据训练的最佳模型。代码与模型已开源。
原文摘要 · Abstract (English)
We present Myna, a simple yet effective approach for self-supervised musical representation learning. Built on a contrastive learning framework, Myna introduces two key innovations: (1) the use of a Vision Transformer (ViT) on mel-spectrograms as the backbone and (2) a novel data augmentation strategy, token masking, that masks 90 percent of spectrogram tokens. These innovations deliver both effectiveness and efficiency: (i) Token masking enables a significant increase in per-GPU batch size, from 48 or 120 in prior methods (CLMR, MULE) to 4096. (ii) By avoiding traditional augmentations, Myna retains pitch sensitivity, enhancing performance in tasks like key detection. (iii) The use of vertical patches allows the model to better capture critical features for key detection. Our hybrid model, Myna-22M-Hybrid, processes both 16x16 and 128x2 patches, achieving state-of-the-art results. Trained on a single GPU, it outperforms MULE (62M) on average and rivals MERT-95M, which was trained on 16 and 64 GPUs, respectively. Additionally, it surpasses MERT-95M-public, establishing itself as the best-performing model trained on publicly available data. We release our code and models to promote reproducibility and facilitate future research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。