SLAM-LLM
SLAM-LLM 是一款专为语音、语言、音频及音乐处理打造的深度学习工具包,旨在帮助开发者高效训练定制化的多模态大语言模型。它主要解决了传统框架在处理复杂音频任务时面临的训练门槛高、大规模数据支持不足以及资源消耗过大等痛点,让从实验验证到工业级部署变得更加顺畅。
这款工具特别适合人工智能研究人员、算法工程师以及对多模态交互感兴趣的开发者使用。无论是学术探索还是构建千万小时级的工业应用,SLAM-LLM 都能提供详尽的训练指南和高性能推理模型。
其技术亮点十分突出:不仅完整复现了仅需单阶段训练即可实现音色可控对话的 SLAM-Omni 系统,还针对超大规模数据集进行了深度优化。通过支持动态帧批处理、迭代式数据集加载以及 DeepSpeed 分布式训练,它能显著降低显存占用并将训练时间缩短至原来的四分之一。此外,工具包内置了丰富的任务配方,统一支持自动语音识别、语音翻译等多任务场景,并具备多机多卡分布式推理能力,是构建下一代智能语音交互系统的得力助手。
使用场景
某智能客服团队正致力于构建一个支持中英双语、能识别用户情绪并控制回复音色的端到端语音对话系统,以替代传统的“语音转文字 + 文本大模型 + 文字转语音”串联架构。
没有 SLAM-LLM 时
- 流程割裂且延迟高:需要分别维护 ASR、LLM 和 TTS 三个独立模型,数据在模块间反复转换,导致响应延迟严重,无法实现流畅的多轮实时对话。
- 音色控制极其困难:若想实现根据用户情绪动态调整回复音色(如安慰时用温柔声线),需额外训练复杂的后处理模块,开发周期长达数月。
- 训练资源消耗巨大:面对十万小时级的工业级语音数据,传统固定批次(Batch Size)训练方式显存占用极高,且启动缓慢,普通算力集群难以承载。
- 多任务协同成本高:语音识别(ASR)与语音翻译(ST)等任务需要维护多套数据格式和代码库,迭代更新时极易出现兼容性问题。
使用 SLAM-LLM 后
- 端到端单阶段训练:利用 SLAM-LLM 复现 SLAM-Omni 架构,仅需单阶段训练即可构建高质量的端到端对话系统,大幅降低延迟,实现自然的中英多轮交互。
- 原生音色可控能力:直接调用框架内置的音色控制机制,无需额外开发即可让模型根据语境自动切换说话风格,显著提升了用户体验。
- 高效处理海量数据:借助动态帧批处理(Dynamic Frame Batching)和 DeepSpeed 支持,训练十万小时数据的时间缩短了 75%,显存占用显著降低,使大规模训练在有限算力下成为可能。
- 统一多任务框架:通过统一的数据格式支持 ASR、ST 等多任务混合训练,一套代码即可管理所有语音语言任务,研发效率提升数倍。
SLAM-LLM 将原本繁琐割裂的语音 AI 开发流程整合为高效统一的端到端解决方案,让开发者能以更低成本快速落地具备情感交互能力的智能语音应用。
运行环境要求
- Linux
必需 NVIDIA GPU,支持 CUDA 11.8+,显存需求视任务而定(支持 DeepSpeed 以降低显存占用,支持多机多卡分布式训练/推理)
未说明(大规模工业级训练建议大内存,Docker 运行示例中设置共享内存为 256GB)

快速开始
SLAM-LLM
SLAM-LLM 是一个深度学习工具包,允许研究人员和开发者训练自定义的多模态大语言模型(MLLM),专注于Speech、Language、Audio、Music 处理。我们提供了详细的训练配方以及用于推理的高性能检查点。
目录
新闻
- [更新于2026年1月15日] 我们的SLAM-LLM框架已被IEEE《信号处理精选主题期刊》(JSTSP)接收,其中我们详细介绍了该框架的设计以及一些未发表的关于语音、音频和音乐MLLM的实验与见解。
- [更新于2025年4月24日] 我们现已支持大规模工业级训练,适用于约10万小时规模的数据集。其主要特点包括:
- 多任务训练支持: 通过统一的数据格式,支持ASR和ST等多种任务。
- 动态提示选择: 支持从多个提示中随机选取。
- 迭代数据集: 使用迭代数据集格式,以减少大型数据集的启动时间。
- DeepSpeed训练: 支持DeepSpeed训练,显著降低内存占用。
- 多机多GPU推理: 支持跨多台机器和多块GPU的分布式推理,从而缩短评估时间。
- 动态帧批处理: 根据音频大小动态组合帧,而非采用固定批大小,大幅缩短训练和评估时间(对于10万小时的数据,可将训练时间缩短四分之三)。
- [更新于2025年4月24日] 我们已支持DeepSpeed训练,请参阅此处的“使用DeepSpeed进行微调”说明。
- [更新于2025年1月22日] 🔥🔥🔥 已完全复现了SLAM-Omni的整个流程(包括所有数据准备、模型训练和推理)。
- [更新于2024年11月17日] 已支持基于LLM的情境感知ASR的配方。
- [更新于2024年11月5日] 已支持使用emotion2vec作为编码器的语音情感字幕生成(SEC)配方。
- [更新于2024年10月12日] 已支持使用EAT作为编码器的SLAM-AAC配方。
- [更新于2024年9月28日] 已支持CoT-ST的配方。
- [更新于2024年9月25日] 已支持DRCap的配方。
- [更新于2024年6月12日] 已支持MaLa-ASR的配方。
- 【征集示例】 我们诚挚邀请开发者和研究人员基于SLAM-LLM开发新应用、开展学术研究,并提交您的示例!我们也欢迎工程相关的PR(例如改进和加速多节点训练)。
- [更新于2024年5月22日] 请加入我们的Slack群组或微信交流群。我们将在此同步最新动态及答疑信息。
- [更新于2024年5月21日] 已支持空间音频理解的配方。
- [更新于2024年5月20日] 已支持音乐字幕生成(MC)的配方。
- [更新于2024年5月8日] 已支持视觉语音识别(VSR)的配方。
- [更新于2024年5月4日] 已支持零样本文本转语音(TTS)的配方。
- [更新于2024年4月28日] 已支持自动音频字幕生成(AAC)的配方。
- [更新于2024年3月31日] 已支持自动语音识别(ASR)的配方。
安装
git clone https://github.com/huggingface/transformers.git
cd transformers
git checkout tags/v4.35.2
pip install -e .
cd ..
git clone https://github.com/huggingface/peft.git
cd peft
git checkout tags/v0.6.0
pip install -e .
cd ..
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
git clone https://github.com/ddlBoJack/SLAM-LLM.git
cd SLAM-LLM
pip install -e .
对于部分示例,您可能需要使用fairseq,命令如下:
# 在安装SLAM-LLM之前,需先安装fairseq
git clone https://github.com/pytorch/fairseq
cd fairseq
pip install --editable ./
我们还提供了一个方便的Docker镜像:
# 构建Docker镜像
docker build -t slam-llm:latest .
# 使用GPU运行Docker镜像
docker run -it --gpus all --name slam --shm-size=256g slam-llm:latest /bin/bash
使用
食谱列表
我们提供了多种基于大语言模型的语音、音频和音乐任务的参考实现:
语音任务
自动语音识别 (ASR)
上下文自动语音识别 (CASR)
语音到文本翻译 (S2TT)
文本到语音 (TTS)
语音交互系统
音频任务
空间音频理解
音乐任务
配置优先级
我们提供了如下层次化的配置继承关系:
命令行(shell文件) > Hydra配置(yaml文件) > dataclass配置(Python文件)
特性
- 易于扩展到新模型和新任务。
- 提供详细的训练食谱和高性能推理检查点。
- 支持混合精度训练,在NVIDIA Tensor Core上以更少的显存更快地训练。
- 多GPU训练支持数据并行和模型并行,兼容DDP、FSDP和deepspeed(仍在完善中)。
- 基于Hydra和dataclass的灵活配置,允许代码、命令行和文件配置的组合。
致谢
- 我们借鉴了Llama-Recipes中的训练流程代码。
- 我们借鉴了Fairseq中的deepspeed配置代码。
- 感谢各位贡献者提供的多样化食谱。
引用
SLAM-LLM框架:
@article{ma2026slam,
title={SLAM-LLM: 一种模块化、开源的多模态大语言模型框架及语音、语言、音频和音乐处理的最佳实践},
author={Ma, Ziyang and Yang, Guanrou and Chen, Wenxi and Gao, Zhifu and Du, Yexing and Li, Xiquan and Zheng, Zhisheng and Zhu, Haina et al.},
journal={IEEE信号处理专题期刊},
year={2026}
}
语音任务
SLAM-ASR:
@article{ma2025speech,
title={语音识别与大语言模型的结合:基准测试、模型与探索},
author={Ma, Ziyang and Yang, Guanrou and Yang, Yifan and Gao, Zhifu and Wang, Jiaming and Du, Zhihao and Yu, Fan and Chen, Qian and Zheng, Siqi and Zhang, Shiliang et al.},
journal={AAAI会议论文集},
year={2025}
}
Mala-ASR:
@article{yang2024mala,
title={MaLa-ASR:多媒体辅助的大语言模型ASR},
author={Yang, Guanrou and Ma, Ziyang and Yu, Fan and Gao, Zhifu and Zhang, Shiliang and Chen, Xie},
journal={INTERSPEECH会议论文集},
year={2024}
}
基于大语言模型的上下文ASR:
@article{yang2024ctc,
title={CTC辅助的大语言模型上下文ASR},
author={Yang, Guanrou and Ma, Ziyang and Gao, Zhifu and Zhang, Shiliang and Chen, Xie},
journal={SLT会议论文集},
year={2024}
}
SLAM-Omni:
@article{chen2024slam,
title={SLAM-Omni:单阶段训练的音色可控语音交互系统},
author={Chen, Wenxi and Ma, Ziyang and Yan, Ruiqi and Liang, Yuzhe and Li, Xiquan and Xu, Ruiyang and Niu, Zhikang and Zhu, Yanqiao and Yang, Yifan and Liu, Zhanxun et al.},
journal={ACL会议论文集},
year={2025}
}
CoT-ST:
@article{du2024cot,
title={CoT-ST:通过多模态思维链增强大语言模型语音翻译},
author={Du, Yexing and Ma, Ziyang and Yang, Yifan and Deng, Keqi and Chen, Xie and Yang, Bo and Xiang, Yang and Liu, Ming and Qin, Bing},
journal={ACL会议论文集},
year={2025}
}
音频任务
SLAM-AAC:
@article{chen2025slam,
title={SLAM-AAC:利用释义增强和CLAP精炼通过大语言模型提升音频字幕生成},
author={Chen, Wenxi and Ma, Ziyang and Li, Xiquan and Xu, Xuenan and Liang, Yuzhe and Zheng, Zhisheng and Yu, Kai and Chen, Xie},
journal={ICASSP会议论文集},
year={2025}
}
DRCap:
@article{li2025drcap,
title={DRCap:通过检索增强生成解码CLAP潜在表示实现零样本音频字幕生成},
author={Li, Xiquan and Chen, Wenxi and Ma, Ziyang and Xu, Xuenan and Liang, Yuzhe and Zheng, Zhisheng and Kong, Qiuqiang and Chen, Xie},
journal={ICASSP会议论文集},
year={2025}
}
BAT:
@article{zheng2024bat,
title={BAT:利用大语言模型学习空间声音推理},
author={Zheng, Zhisheng and Peng, Puyuan and Ma, Ziyang and Chen, Xie and Choi, Eunsol and Harwath, David},
journal={ICML会议论文集},
year={2024}
}
常见问题
相似工具推荐
everything-claude-code
everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上
LLMs-from-scratch
LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备
spec-kit
Spec Kit 是一款专为提升软件开发效率而设计的开源工具包,旨在帮助团队快速落地“规格驱动开发”(Spec-Driven Development)模式。传统开发中,需求文档往往与代码实现脱节,导致沟通成本高且结果不可控;而 Spec Kit 通过将规格说明书转化为可执行的指令,让 AI 直接依据明确的业务场景生成高质量代码,从而减少从零开始的随意编码,确保产出结果的可预测性。 该工具特别适合希望利用 AI 辅助编程的开发者、技术负责人及初创团队。无论是启动全新项目还是在现有工程中引入规范化流程,用户只需通过简单的命令行操作,即可初始化项目并集成主流的 AI 编程助手。其核心技术亮点在于“规格即代码”的理念,支持社区扩展与预设模板,允许用户根据特定技术栈定制开发流程。此外,Spec Kit 强调官方维护的安全性,提供稳定的版本管理,帮助开发者在享受 AI 红利的同时,依然牢牢掌握架构设计的主动权,真正实现从“凭感觉写代码”到“按规格建系统”的转变。
NextChat
NextChat 是一款轻量且极速的 AI 助手,旨在为用户提供流畅、跨平台的大模型交互体验。它完美解决了用户在多设备间切换时难以保持对话连续性,以及面对众多 AI 模型不知如何统一管理的痛点。无论是日常办公、学习辅助还是创意激发,NextChat 都能让用户随时随地通过网页、iOS、Android、Windows、MacOS 或 Linux 端无缝接入智能服务。 这款工具非常适合普通用户、学生、职场人士以及需要私有化部署的企业团队使用。对于开发者而言,它也提供了便捷的自托管方案,支持一键部署到 Vercel 或 Zeabur 等平台。 NextChat 的核心亮点在于其广泛的模型兼容性,原生支持 Claude、DeepSeek、GPT-4 及 Gemini Pro 等主流大模型,让用户在一个界面即可自由切换不同 AI 能力。此外,它还率先支持 MCP(Model Context Protocol)协议,增强了上下文处理能力。针对企业用户,NextChat 提供专业版解决方案,具备品牌定制、细粒度权限控制、内部知识库整合及安全审计等功能,满足公司对数据隐私和个性化管理的高标准要求。
ML-For-Beginners
ML-For-Beginners 是由微软推出的一套系统化机器学习入门课程,旨在帮助零基础用户轻松掌握经典机器学习知识。这套课程将学习路径规划为 12 周,包含 26 节精炼课程和 52 道配套测验,内容涵盖从基础概念到实际应用的完整流程,有效解决了初学者面对庞大知识体系时无从下手、缺乏结构化指导的痛点。 无论是希望转型的开发者、需要补充算法背景的研究人员,还是对人工智能充满好奇的普通爱好者,都能从中受益。课程不仅提供了清晰的理论讲解,还强调动手实践,让用户在循序渐进中建立扎实的技能基础。其独特的亮点在于强大的多语言支持,通过自动化机制提供了包括简体中文在内的 50 多种语言版本,极大地降低了全球不同背景用户的学习门槛。此外,项目采用开源协作模式,社区活跃且内容持续更新,确保学习者能获取前沿且准确的技术资讯。如果你正寻找一条清晰、友好且专业的机器学习入门之路,ML-For-Beginners 将是理想的起点。
funNLP
funNLP 是一个专为中文自然语言处理(NLP)打造的超级资源库,被誉为"NLP 民工的乐园”。它并非单一的软件工具,而是一个汇集了海量开源项目、数据集、预训练模型和实用代码的综合性平台。 面对中文 NLP 领域资源分散、入门门槛高以及特定场景数据匮乏的痛点,funNLP 提供了“一站式”解决方案。这里不仅涵盖了分词、命名实体识别、情感分析、文本摘要等基础任务的标准工具,还独特地收录了丰富的垂直领域资源,如法律、医疗、金融行业的专用词库与数据集,甚至包含古诗词生成、歌词创作等趣味应用。其核心亮点在于极高的全面性与实用性,从基础的字典词典到前沿的 BERT、GPT-2 模型代码,再到高质量的标注数据和竞赛方案,应有尽有。 无论是刚刚踏入 NLP 领域的学生、需要快速验证想法的算法工程师,还是从事人工智能研究的学者,都能在这里找到急需的“武器弹药”。对于开发者而言,它能大幅减少寻找数据和复现模型的时间;对于研究者,它提供了丰富的基准测试资源和前沿技术参考。funNLP 以开放共享的精神,极大地降低了中文自然语言处理的开发与研究成本,是中文 AI 社区不可或缺的宝藏仓库。