BigMac – 小红书开源的多模态大模型流水并行训练框架

最近 AI 工具圈又上新了——BigMac。下面用第三方视角拆一拆:它到底解决什么问题、怎么上手、和同类比差在哪。

BigMac是什么

BigMac 是小红书 dots infra 团队开源的多模态大模型流水并行训练框架,针对 MLLM 训练中计算效率与显存占用难以兼顾的问题,提出依赖安全的嵌套流水线范式。框架用成熟 LLM 流水线为主干,在不打乱执行顺序的前提下嵌入编码器和生成器计算,实现训练速度提升 1.08~1.9 倍的同时保持显存稳定,已作为 dots 多模态训练核心组件投入生产。

核心功能能做什么

BigMac是什么

BigMac 是小红书 dots infra 团队开源的多模态大模型流水并行训练框架,针对 MLLM 训练中计算效率与显存占用难以兼顾的问题,提出依赖安全的嵌套流水线范式。框架用成熟 LLM 流水线为主干,在不打乱执行顺序的前提下嵌入编码器和生成器计算,实现训练速度提升 1.08~1.9 倍的同时保持显存稳定,已作为 dots 多模态训练核心组件投入生产。

BigMac

BigMac的主要功能

  • 依赖安全的嵌套流水线:以 LLM pipeline 为稳定主干,在依赖满足的位置嵌入 encoder / generator 计算,避免跨模块 bubble 干扰。
  • 显存有界的模态激活:编码器和生成器激活显存降至 O(1),LLM 激活行为保持不变,支持更大 batch size 而不 OOM。
  • 全局调度与执行解耦:Scheduler 生成覆盖所有 rank、microbatch 和模块的全局 operator 表;Executor 解释本地序列并分发给对应后端。
  • 流水线无感接口:算法工程师只需描述模块输入输出边界,BigMac 自动处理 schedule、handoff 和跨设备通信,模型代码保持模块化。
  • Schedule-aware 工具链:内置 profiler、simulator 和可视化工具,支持 per-operator 级别 trace 导出(兼容 Perfetto UI),快速定位 bubble、慢算子与通信等待。

BigMac的技术原理

  • 依赖安全的嵌套流水线调度:BigMac将其作为基础时间线。调度器分析 encoder forward / LLM forward / generator forward 及对应 backward 的数据依赖关系,仅在输入已就绪且不会打乱 LLM 执行顺序的”空档”插入模态计算。 LLM 按原 schedule 持续推进,encoder 和 generator 的激活也能在梯度就绪后尽快释放,backward 及时执行,打破”计算高效必高显存、显存高效必多 bubble”的帕累托权衡。
  • 全局 Operator 表与后端解耦:系统将调度策略显式化为一张全局 operator 表,包含所有 pipeline rank 上 LLM / encoder / generator 的 forward/backward operator 及模块边界通信。执行层仅解释本地 operator 序列,LLM operator 复用 Megatron Core,模态 operator 保留自身 data-parallel / FSDP 实现,schedule 调整无需重写模型 runtime。

实际适合谁用

适合想快速体验新能力、做原型或本地部署的开发者与爱好者;涉及生产级稳定性、合规审计的场景,先评估再上车。

和同类比,怎么选

BigMac 侧重「小红书开源的多模态大模型流水并行训练框架」。选哪个看你是个人尝鲜还是企业落地——建议先试免费或开源版,再决定要不要深度接入。

常见问题

BigMac是什么,主要解决什么问题

BigMac 是小红书 dots infra 团队开源的多模态大模型流水并行训练框架,针对 MLLM 训练中计算效率与显存占用难以兼顾的问题,提出依赖安全的嵌套流水线范式。框架用成熟 LLM 流水线为主干,在不打乱执行顺序的前提下嵌入编码器和生成器计算,实现训练速度提升 1.08~1.9 倍的同时保持显存稳定,已作为 dots 多模态训练核心组件投入生产。

BigMac开源吗

BigMac已开源(正文已说明其开源版本与本地部署方式),具体许可协议以官方仓库公示为准。

BigMac怎么上手 / 怎么用

访问官网或对应平台(文内链接)注册或下载即可体验;开源版本按官方仓库说明本地部署,消费级显卡大多能跑。

BigMac和同类工具比,选哪个

BigMac侧重「小红书开源的多模态大模型流水并行训练框架」;选哪个看你是个人尝鲜还是企业落地,建议先试免费或开源版再决定。

BigMac适合谁用,不适合谁

适合想快速体验新能力、做原型或本地部署的开发者与爱好者;涉及生产级稳定性、合规审计的场景,先评估再上车,别直接当关键链路。

⚠️ 说明:本文为第三方工具介绍,非官方推广;功能、价格与许可以官方文档为准;AI 生成/输出内容请自行甄别,关键决策勿过度依赖模型。

声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。