公众号:小红书技术· 小红书技术REDtech·· 2025-12-11AI 评分52
小红书技术提出细粒度激活卸载:大模型训练显存与吞吐性能联合最优方案
大模型训练的高效内存解决方案:流水线感知的细粒度激活卸载,实现显存开销与吞吐性能的联合最优
AI 导读
细粒度激活卸载(Fine-grained Activation Offloading)采用模块级粒度,将计算与内存传输重叠,并兼容全并行设计,在主流大模型训练中同时降低峰值显存开销并提升吞吐性能,实现显存开销与吞吐性能的联合最优,获得显著性能收益和显存收益。
来源:公众号:小红书技术 · mp.weixin.qq.com