当前位置: 首页 > news >正文

1F1B(One-Forward-One-Backward)

1F1B(One-Forward-One-Backward)是流水线并行(Pipeline Parallelism)中最经典的微批次调度策略,主要用于解决传统流水线训练中的显存占用过高气泡率(Bubble Ratio)过大问题。


核心痛点:传统 GPipe 的缺陷

在普通的流水线并行(如 GPipe)中,系统会先一口气把所有微批次(Micro-batch)的前向传播(Forward)全部跑完,再统一跑反向传播(Backward)

  • 显存爆炸:设备必须在显存中同时保存所有 Micro-batch 的激活值(Activations),直到反向传播时才能释放。
  • 气泡严重:卡与卡之间存在大量的空闲等待时间。

1F1B 的工作机制

1F1B 的核心思想是:让前向传播和反向传播在时间线上升高交叉,每跑完一个 Micro-batch 的前向,就紧接着跑一个最老 Micro-batch 的反向。

整个调度过程分为三个阶段:

  • 1. 预热阶段(Warmup)

  • 处于流水线前面的卡先连续跑若干个(通常等于流水线深度p pp)Micro-batch 的前向计算,直到把流水线“填满”。

  • 2. 稳定阶段(1F1B Steady State)

  • 一前一后交替:每执行完1 个 Micro-batch 的 Forward,就立刻接1 个最老的 Micro-batch 的 Backward

  • 显存及时释放:由于 Backward 执行完毕后该 Micro-batch 的激活值就可以立即从显存中销毁,因此显存占用被锁死在一个固定上限(仅需保存约p pp个 Micro-batch 的激活值)。

  • 3. 消退阶段(Cooldown)

  • 所有 Forward 跑完后,把剩余 Micro-batch 的 Backward 执行完毕。


1F1B vs GPipe 性能对比

维度GPipe (Naive PP)1F1B Pipeline
激活值显存峰值O ( m ) O(m)O(m)m mm为总 Micro-batch 数量)O ( p ) O(p)O(p)p pp为流水线 Stage 数量/深度,p ≪ m p \ll mpm
气泡率(Bubble Ratio)p − 1 m \frac{p - 1}{m}mp1p − 1 m \frac{p - 1}{m}mp1(与 GPipe 相当,但显存大幅降低)
调度复杂度简单(顺序执行)中等(需精细管理内部队列和通信)

进阶演进:交错式 1F1B(Interleaved 1F1B)

Megatron-LM 等框架在 1F1B 基础上提出了Interleaved 1F1B

  • 让每张物理 GPU 负责模型中多个非连续的物理层(例如 16 层模型,卡 0 负责第 1 层和第 9 层)。
  • 通过在单卡内部交错调度不同 Virtual Stage 的前向和反向,进一步把流水线气泡率降低到了原本的1 / v 1/v1/vv vv为每张卡上的虚拟 Stage 数量)。
http://www.jsqmd.com/news/1390975/

相关文章:

  • 洛雪音乐开源音源如何免费听全网无损?我的真实体验分享
  • Maven命令实战手册:从核心概念到问题排查的完整指南
  • 建筑外墙玻璃幕墙节点设计20大忌!
  • C#三层架构实现机房卡号充值模块:事务、并发与安全设计
  • 从一次“眼睁睁看着消息被撤回“说起:RevokeMsgPatcher 特征码匹配机制拆解
  • 2026 年 8 月吉林房屋漏水科普:台风暴雨叠加回潮,房屋渗水维修怎么选 - 筑宅安
  • 从“被看见”到“有选择”,尿素循环障碍患者长期治疗迈向新阶段 - 滚动商讯
  • 2026年香港高端眼镜哪家值得推荐 最新选型参考梳理 - 互联网科技品牌测评
  • 私享草原:7位金牌导游为你定制专属旅行 - 纯玩旅游分享
  • Deep-Live-Cam 实时换脸完整实战:一张照片3步开启免费直播换脸
  • WSL 2 互操作性深度解析:无缝融合 Windows 与 Linux 命令行
  • 2026 台州全屋定制实测:大宅全案落地避坑 + 靠谱机构参考 - 互联网科技品牌测评
  • 2026上海二手电脑回收避坑指南,同城上门免费估价干货全分享 - 米諾
  • 抖音视频批量下载零基础实操:免费开源工具douyin-downloader无水印指南
  • 免费资源网站高效使用指南:从收藏到生产力工具
  • 如何把黑苹果配置从玄学变成科学:用 OpCore-Simplify 实现 OpenCore EFI 自动化
  • 2026北京房产遗产继承律师事务所筛选盘点:3家律所对比,合作避坑与挑选指南 - 产业观察报
  • 深入解析8255A可编程并行接口芯片:从原理到嵌入式IO扩展实战
  • 建筑外遮阳技术
  • 做抖音小店无货源副业怎样规避风控?密文一键发货实操细节解析 - 抖掌柜一键下单
  • 2024最新版招商网站建设方案全解析与实操指南,助力政府与企业高效招商
  • 中国技术大败局TBL-20260814-081深度解剖报告V2.1 决策迭代版
  • 眉山网站建设公司如何选择?揭秘专业团队背后的服务逻辑与避坑指南
  • 2026北京房产继承法律服务大盘点:5家专业律所对比及挑选避坑全指南 - 商业大观
  • 数学建模竞赛:从赛题分析到模型构建的实战方法论
  • 右键菜单乱成一锅粥?三步搞定Windows右键菜单清理
  • 阿里云ECS搭建Minecraft服务器:从零到联机全流程实战指南
  • 只用一张照片就能实时换脸?Deep-Live-Cam 免费开源AI工具完整上手指南
  • 魔兽争霸3终极兼容性救星:WarcraftHelper从安装到调优的完整实战指南
  • Unity专业功能解锁工具免费完整指南:UniHacker快速上手全攻略