MindCluster MindIO Async Checkpoint Persistence(下文简称MindIO ACP)加速大模型CheckPoint功能主要针对大模型训练中的CheckPoint的保存及加载进行加速,CheckPoint的数据先写入训练服务器的内存系统中,再异步写入后端的可靠性存储设备中。本文档主要介绍纵向加速部分,包含CheckPoint在本系统中的写入及读取过程。
LLM(Large Language Model)是全球当前科技界竞争的焦点,LLM模型的训练往往需要长达数十天、甚至数月。CheckPoint是模型中断训练后恢复的关键点,CheckPoint的密集程度、保存和恢复的性能较为关键,它可以提高训练系统的有效吞吐率。MindIO ACP针对CheckPoint的加速方案,支持昇腾产品在LLM模型领域扩展市场空间,支持客户进行POC和上线测试。
该方案提升昇腾平台上LLM模型的训练吞吐量,性能超越Microsoft Azure Nebula方案。
MindIO ACP加速LLM CheckPoint保存和加载的4个关键点如下:
MindIO ACP仅保存训练过程中的CheckPoint数据,暂不支持敏感数据的保存和处理。若涉及敏感数据存储,请在前序流程完成相关脱敏操作,避免造成信息安全问题。