首页|期刊导航|大数据|深度学习模型训练过程检查点访问性能优化方法

深度学习模型训练过程检查点访问性能优化方法OA

Checkpoint accessing performance optimization method for the deep learning model training process

中文摘要英文摘要

随着大模型应用越来越广泛、规模逐渐增大,目前大模型训练面临出错概率高、检查点访问性能差等问题.总结了已有检查点访问性能优化方法的优缺点,提出了一种新的检查点访问性能优化方法.观察检查点数据模式可知,相近检查点的模型权重数据变化较小,适合增量压缩.基于多台互联训练节点实现了增量压缩,并基于真实的深度学习模型训练时产生的检查点数据进行了实验测试.结果表明,在训练周期内,增量压缩对大多数检查点具有较好的压缩效果.此外,提出在增量压缩中使用动态间隔来平衡压缩率与存储开销,并对动量数据特征进行分析.文章对已有方法的分析及对检查点访问性能的优化为大模型训练加速提供了指导.

As LLMs become more widely used and their scale continues to expand,currently LLMs training faces issues such as high error rate and poor performance of checkpoint accessing.This paper reviews the strengths and weaknesses of existing methods for optimizing checkpoint accessing performance and introduces a novel method for optimizing checkpoint accessing performance.Based on the observation of data patterns in checkpoints,where the model weights change between adjacent checkpoints are minimal,making them suitable for delta compression.The proposed method implements delta compression across multiple interconnected training nodes and conducts experimental tests using real checkpoints generated during deep learning model training.The results demonstrate that,during the model training,delta compression has good compression effect for most checkpoints.Furthermore,the paper introduces dynamic intervals in delta compression to balance compression ratio and storage overhead,while also analyzing the characteristics of momentum datas.The analysis of existing methods and the optimization of checkpoint accessing performance offer insights for accelerating LLMs training.

滕云;张广艳;孙大为;田海东;常锐

中国地质大学(北京)人工智能学院,北京 100083清华大学计算机科学与技术系,北京 100084中国地质大学(北京)人工智能学院,北京 100083中兴通讯股份有限公司,江苏 南京 210012中兴通讯股份有限公司,江苏 南京 210012

信息技术与安全科学

大模型检查点数据压缩性能提升

LLMcheckpointdata compressionperformance improvement

《大数据》 2026 (2)

75-84,10

国家自然科学基金项目(No.62025203) The National Natural Science Foundation of China(No.62025203)

10.11959/j.issn.2096-0271.2026029

评论