Kyutai推出MuScriptor开源多乐器转录模型,实现高精度MIDI生成

1 阅读5分钟开源

模型概述

MuScriptor 是 Kyutai 与 Mirelo 合作研发的解码器唯一(decoder‑only)Transformer,专注于将混合乐器音频自动转录为标准 MIDI。模型首先将 5 秒音频片段转为 mel‑spectrogram,随后以自回归方式预测基于 MT3 方案的 pitch、timing、instrument 三类 token,等同于把转录任务视作语言模型预测。当前发布三种规模:

  • small:103M 参数
  • medium(默认):307M 参数
  • large:1.4B 参数

推理代码采用 MIT 许可证,模型权重采用 CC BY‑NC 4.0,商业使用受限。

三阶段训练管线

MuScriptor 的核心创新在于 数据驱动 而非架构改动,训练分为三步:

  1. DSynth 预训练 – 使用约 145 万合成 MIDI,训练时通过实时合成管线生成音频,加入调音、变速、力度、乐器随机化等增强;超过 250 种 SoundFont 与随机失调提供几乎无限的音色组合。
  2. DReal 精调 – 基于内部收集的 17 万真实录音(总时长 11,000 小时),通过插值与动态时间规整实现音频‑符号对齐,过滤对齐误差大的样本。
  3. DRL 后训练 – 采用类似 GRPO 的强化学习方法,对 300 条人工验证轨道进行 REINFORCE + group‑relative advantage 正规化,奖励函数为 onset、frame、offset 三个 F1 的加权和,进一步提升起止时点的准确性。

基准评估

在公开数据集 DTest(372 条持有准确标注的曲目)上,MuScriptor 与基线 YourMT3+ 的对比如下(large 模型):

模型Onset F1Frame F1Offset F1Drums F1Multi F1
YourMT3+32.545.517.841.421.9
MuScriptor · DSynth34.548.916.120.16.2
+ DReal54.469.342.334.341.6
+ DRL60.473.349.048.248.2

可以看到,真实数据的加入提升约 20 分,强化学习进一步压缩漏检并 sharpen 起止时点。跨数据集测试同样显示,帧 F1 在 Dagstuhl ChoirSet 上从 51.0 提升至 80.7,虽在合唱等复杂风格上仍有提升空间。

使用指南

pip install muscriptor  # 或 uv add muscriptor
from pathlib import Path
from muscriptor import TranscriptionModel

# 自动下载并加载默认 medium 版本
model = TranscriptionModel.load_model()

# 实时流式转录,支持乐器条件
for event in model.transcribe("audio.wav", instruments=["acoustic_piano", "drums"]):
    print(event)  # NoteStartEvent / NoteEndEvent / ProgressEvent

# 直接保存为 MIDI 文件
Path("out.mid").write_bytes(model.transcribe_to_midi("audio.wav"))

模型默认 cfg_coef 为 1,已完成 RL 后训练。large 规模模型推荐使用 GPU(至少 12GB 显存)以保证实时速度。

应用场景与局限

  • 音乐制作:快速提取混音中的贝斯、钢琴等声部,便于在 DAW 中二次编曲。
  • 乐理研究:将历史录音转为可编辑乐谱,辅助音乐学分析。
  • 音乐信息检索:为 MIR(Music Information Retrieval)任务提供高质量的音符序列,提升和弦、调式识别的准确性。
  • 教育工具:同步展示钢琴卷轴,帮助学生对照音频练习。

局限

  • 权重采用 CC BY‑NC,商业部署需额外授权。
  • Tokenizer 未保留力度信息,且无法表示同音同乐器的重叠音符。
  • 在合唱、复杂和声等风格上起止/结束点仍表现欠佳。
  • 大模型对显存要求高,5 秒片段限制了长程上下文捕获。

“MuScriptor 证明了开放模型在多乐器转录领域同样可以达到科研前沿的性能。” – Kyutai 团队技术负责人

总体而言,MuScriptor 为音频‑符号转换提供了开源且高效的方案,既可供研究者复现,也能直接服务于音乐创作与分析工作流。

本文是对第三方新闻源的主观解读。消息可能出现过时、不准确、歧义或错误的地方,仅供参考使用。点击此处查看消息源。