随着人工智能技术的不断发展,大模型在各个领域的应用日益广泛。然而,大模型的训练一直面临着算力、数据、算法等多方面的挑战。近年来,我国在国产大模型训练效率上实现了飞跃式提升,本文将揭秘这一突破背后的关键因素。
一、算力突破:国产芯片助力大模型训练
国产GPU发展迅速:摩尔线程开源的MT-MegatronLM与MT-TransformerEngine两大AI框架,在国产全功能GPU上实现了混合并行训练和推理,大幅提升了大模型训练的效率与稳定性。这些框架通过深度融合FP8混合训练策略和高性能算子库,使得国产GPU在训练效率上取得了显著突破。
全国产化算力平台:中国电信人工智能研究院(TeleAI)成功完成国内首个基于全国产化万卡集群训练的万亿参数大模型,标志着国产大模型训练真正实现全国产化替代。这一成果得益于我国在芯片、服务器等硬件领域的持续投入。
二、算法创新:优化训练流程
深度优化训练策略:TeleAI在TeleChat2-115B的训练过程中,采用了大量小模型进行Scaling,验证不同模型结构的有效性。同时,基于小模型实验结果反馈,采用回归预测模型,得到较优数据配比,大幅提升大参数模型最优方案的搜寻效率。
后训练优化:TeleAI在Post-Training(后训练)阶段,针对数学、代码和逻辑推理等内容合成了大量问答数据,用于SFT(监督式微调)第一阶段模型训练。通过迭代式更新策略,提升答案质量,并利用拒绝采样获取优质SFT数据及RM(奖励模型)代表性数据,进一步提升了模型训练效果。
三、数据资源:高质量数据助力模型训练
海量数据支持:火山引擎发布的大模型训练视频预处理方案,借助英特尔(Intel)的CPU等资源,有效应对模型训练的算力成本挑战。该方案可以对海量视频数据进行高质量预处理,提高模型训练效率。
数据标准化:对训练视频进行预处理,统一视频的数据格式、提高数据质量、实现数据标准化,减少数据量以及处理标注信息,从而提高模型训练效果。
四、生态建设:推动国产大模型发展
开源框架:摩尔线程开源的MT-MegatronLM与MT-TransformerEngine框架,为开发者构建自有的AI技术栈提供了底层支撑,推动了国产大模型生态建设。
产学研合作:科大讯飞与华为、中国电信等企业合作,攻克了万卡高速互联组网、计算通信隐藏、训练推理强交互、高吞吐推理优化以及国产算子优化等一系列难题,实现了国产算力的极致效率。
总之,我国在国产大模型训练效率上取得的飞跃式提升,得益于算力突破、算法创新、数据资源丰富以及生态建设等多方面因素。未来,随着我国在人工智能领域的持续投入,国产大模型将在更多领域发挥重要作用。
