用状态空间模型实现高效开源文本生成音乐,训练成本大幅降低。
Training-Efficient Text-to-Music Generation with State-Space Modeling
- 用状态空间模型替代Transformer,减少参数量和计算开销。
- 仅用9%算力、2%数据量,性能接近MusicGen-small基准。
- 模型更小仍保持竞争力,适合资源有限的研究者使用。
近年来文本到音乐生成(TTM)取得了高质量成果,但通常需要大量计算资源和专有数据。为提升训练的可负担性和开放性,亟需一个参数量少、数据效率高的开源生成模型。本文将生成模型的可训练参数量控制在约300M(与MusicGen-small相当),并用新兴的状态空间模型(SSM)替代其Transformer主干。我们探索了多种序列建模的SSM变体,比较了单阶段纯SSM设计与两阶段SSM/扩散混合设计。所有模型均从零开始,在457小时的CC授权公开音乐数据集上训练,确保完全开放。实验发现:第一,SSM在训练效率上优于Transformer;第二,尽管仅使用9%的浮点运算量和2%的训练数据规模,模型在客观指标和基于MusicCaps的主观听感测试中仍表现良好;第三,当模型缩小至原尺寸的1/4时,在相同训练迭代次数下,性能仍优于同规模的Transformer基线。为推动TTM研究普及,项目已开源:处理后的文本标签、模型检查点及源代码可在GitHub获取。
原文摘要 · Abstract (English)
Recent advances in text-to-music generation (TTM) have yielded high-quality results, but often at the cost of extensive compute and the use of large proprietary internal data. To improve the affordability and openness of TTM training, an open-source generative model backbone that is more training- and data-efficient is needed. In this paper, we constrain the number of trainable parameters in the generative model to match that of the MusicGen-small benchmark (with about 300M parameters), and replace its Transformer backbone with the emerging class of state-space models (SSMs). Specifically, we explore different SSM variants for sequence modeling, and compare a single-stage SSM-based design with a decomposable two-stage SSM/diffusion hybrid design. All proposed models are trained from scratch on a purely public dataset comprising 457 hours of CC-licensed music, ensuring full openness. Our experimental findings are three-fold. First, we show that SSMs exhibit superior training efficiency compared to the Transformer counterpart. Second, despite using only 9% of the FLOPs and 2% of the training data size compared to the MusicGen-small benchmark, our model achieves competitive performance in both objective metrics and subjective listening tests based on MusicCaps captions. Finally, our scaling-down experiment demonstrates that SSMs can maintain competitive performance relative to the Transformer baseline even at the same training budget (measured in iterations), when the model size is reduced to four times smaller. To facilitate the democratization of TTM research, the processed captions, model checkpoints, and source code are available on GitHub via the project page: https://lonian6.github.io/ssmttm/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。