LLM 训推基础设施构建:Kimi 全链路稳定性与高效资源利用实践
导语
随着大模型规模与业务复杂度的指数级上升,基础设施的挑战已从单点优化转向全局架构治理。线上推理呈现明显的潮汐效应,强化学习(RL)场景下训练与推理的混合部署成为刚需。然而,大规模集群的硬件故障、资源闲置与模式冲突,正严重拖慢工程迭代效率。本文将拆解 Kimi 在 LLM 训推基础设施上的工程实践,聚焦全链路稳定性治理与高效资源利用,给出可直接落地的系统架构与调度方案。
核心问题与挑战
在大规模训推集群的4, A 9, 2B C D E F G H I J K L M N O P Q R S T U V W X Y Z [ \ ] ^ _ ` a b c d e f g h i j k l m n o p q r s t u v w x y z { | } ~ ¡ ¢ £ ¤ ¥ ¦ § ¨ © ª « ¬ ® ¯ ° ± ² ³ ´ µ ¶ · ¸ ¹ º » ¼ ½ ¾ ¿ À Á Â Ã Ä Å Æ Ç È É Ê Ë Ì Í Î Ï Ð*Ð Ñ Ò Ó Ô Õ Ö × Ø Ù Ú Û Ü Ý Þ ß à á â ã ä å æ ç è é ê ë ì@ì í î ï ð ñ ò ó ô õ ö ÷ ø ù ú û ü ý þ ÿ
- 机器频繁故障且恢复慢:大规模集群故障是常态,人工排查慢,直接影响实验效率与训练吞吐。
- 开发机与存储资源浪费:开发机与文件存储存在大量闲置,利用率低下。
- 线上推理潮汐效应:流量峰谷明显,固定配比导致低谷期资源闲置、高峰期资源短缺。
- RL 训推分配不均且难融合:训练和推理跑的是两套代码与优化方案,资源割裂,难以统一调度。
方案与实践
全链路稳定性提升
稳定性是算力节省的前提。我们构建了 PLMC 系统架构,核心在于区分好机器与坏机器,根据任务6, 2B C D E F G H I J K L M N O P Q R S T U V W X Y Z [ \ ] ^ _ ` a b c d e f g h i j k l m n o p q r s t u v w x y z { | } ~ ¡ ¢ £ ¤ ¥ ¦ § ¨ © ª « ¬ ® ¯ ° ± ² ³ ´ µ ¶ · ¸ ¹ º » ¼ ½ ¾ ¿ À Á Â Ã Ä Å Æ Ç È É Ê Ë Ì Í Î Ï Ð Ñ Ò Ó Ô Õ Ö × Ø Ù Ú Û Ü Ý Þ ß à á â ã ä å æ ç è é ê ë ì í î ï ð ñ ò ó ô õ ö ÷ ø ù ú û ü ý þ ÿ
- 故障自愈(PreCheck & 周期巡检):在任务调度前执行 PreCheck,测试集合通讯性能、GPU d2h 等指标;配合周期巡检持续监控,自动剔除坏节点,实现故障自动维护与自愈。
- 调用栈全链路监控:CPU 侧周期记录 py 调用栈,任务崩溃后记录 torch C 调用栈,精准定位异常。
- 智能日志分析:离线日志聚合查询,匹配特殊模式寻找报错实际日志;在线实时分析定位任务异常诱因。
- 连续异步 Checkpoint:实现无损 async checkpoint saver,重要任务不间断开启异步存 ckpt,结合跨集群同步备份,避免故障恢复时的大量算力浪费。
高效资源利用
- **全方位开发8, 2B C D E F G H I J K L M N O P Q R S T U V W X Y Z [ \ ] ^ _ ` a b c d e f g h i j k l m n o p q r s t u v w x y z { | } ~ ¡ ¢ £ ¤ ¥ ¦ § ¨ © ª « ¬ ® ¯ ° ± ² ³ ´ µ ¶ · ¸ ¹ º » ¼ ½ ¾ ¿ À Á Â Ã Ä Å Æ Ç È É Ê Ë Ì Í Î Ï Ð Ñ Ò Ó Ô Õ Ö × Ø Ù Ú Û Ü Ý Þ ß à á â ã ä å æ ç è$è é ê ë ì í î ï ð ñ ò ó ô õ ö ÷ ø ù ú û ü ý þ ÿ
- 跨机房推理模型分发:优化模型跨机房分发机制,支撑全局高效调度与资源流转。
- 训推多级潮汐系统:定义四级资源优先级(如重保服务、Kimi 线上推理等),按优先级动态分配资源,削峰填谷,解决潮汐效应带来的资源闲置。
强化学习中的混合部署
RL 场景下,训练和推理是两种不同的模式,框架与代码各异1, 2B C D E F G H I J K L M N O P Q R S T U V W X Y Z [ \ ] ^ _ ` a b c d e f g h i j k l m n o p q r s t u v w x y z { | } ~ ¡ ¢ £ ¤ ¥ ¦ § ¨ © ª « ¬ ® ¯ ° ± ² ³ ´ µ ¶ · ¸ ¹ º » ¼ ½ ¾ ¿ À Á Â Ã Ä Å Æ Ç È É Ê Ë Ì Í Î Ï Ð Ñ Ò Ó Ô Õ Ö × Ø Ù Ú Û Ü Ý Þ ß à á â ã ä å æ ç è é ê ë ì í î ï ð ñ ò ó ô õ ö ÷ ø ù ú û ü ý þ ÿ
- k1.5 混合部署架构:@, 2B C D E F G H I J K L M N O P Q R S T U V W X Y Z [ \ ] ^ _ ` a b c d e f g h i j k l m n o p q r s t u v w x y z { | } ~ ¡ ¢ £ ¤ ¥ ¦ § ¨ © ª « ¬ ® ¯ ° ± ² ³ ´ µ ¶ · ¸ ¹ º » ¼ ½ ¾ ¿ À Á Â Ã Ä Å Æ Ç È É Ê Ë Ì Í Î Ï Ð Ñ Ò Ó Ô Õ Ö × Ø Ù Ú Û Ü Ý Þ ß à á â ã ä å æ ç è é ê ë ì í î ï ð ñ ò ó ô õ ö ÷”÷ ø ù ú û ü ý þ ÿ
- 不落盘权重传输:训练与推理间直接内存级传输 weight,避免 IO 瓶颈。
- Greedy Rollout 调度:在多机推理和 weights 传输中动态调整 RDMA QoS,通过动态负载均衡最大化利用异构资源。
原则/方法论沉淀
- The more you stabilize, the more you save:稳定性不仅是保障,更是算力节省的根本。故障越少,重算越少,隐性成本越低。
- Efficiency is all you need:极致效率是基础设施的核心目标,从开发机到调度全链路压榨利用率。
- 训推共享资源,避免资源闲置:打破训推资源孤岛,通过混部与潮汐调度让资源时刻处于高负荷运转状态。
总结与行动建议
LLM 基础设施已从“够用即可”转向“稳定性与效率并重”。通过 PLMC 自愈、异步 Ckpt、多级潮汐与 RL 混部,我们实现了算力利用率与迭代效率的双重提升。对于正在构建大模型 Infra 的团队,建议采取以下行动:
- 补齐自愈能力:尽快落地集群 PreCheck 与周期巡检,将故障发现与剔除自动化,降低人工运维负担。
- 评估潮汐特性:梳理业务优先级,落地多级优先级调度,让离线任务填补在线低谷。
- 打通训推孤岛:在 RL 场景优先尝试 Sidecar 隔离与不落盘传输,实现同一集群内的训推资源复用。
开放问题与延伸方向
- 连续异步 ckpt 跨集群同步延迟与带宽占用基准是多少?(需量化异步 ckpt 的网络开销底线)
- Side:9.0) f&f’f)7-! e&!e+e5e?EA e3f)f+f,f.f/f, f2f4f6f8f:f;f<f=f>f?f@fAfBfCfDfEfFfGfHfIfFfJfKfLfMfNfOfPfQfRfSfTfUfVfWfXfYfZf[f@f[f\f]f^f_f`fafbfcfdfefffgfhfifjfkflfmfnfofpfqfrfsftfufvfwfxfyfzf{f|f}f~fffffff ffffffffffffffffffffffffff�Cf f¡f¢f£f¤f¥f¦f§f¨f©fªf«f¬ff®f¯f°f±f²f³f´fµf¶f·f¸f¹fºf»f¼f½f¾f¿fÀfÁfÂfÃfÄfÅfÆfÇfÈfÉfÊfËfÌfÍfÎfÏfÐfÑfÒfÓfÔfÕfÖf×fØfÙfÚfÛfÜfÝfÞfßfàfáfâfãfäfåfæfçfèféfêfëfìfífîfïfðfñfòfófôfõFfõföf÷føfùfúfûfüfýfþfÿ
- 多级潮汐下低优任务频繁被抢占3, 2B C D E F G H I J K L M N O P Q R S T U V W X Y Z [ \ ] ^ _ ` a b c d e f g h i j k l m n o p q r s t u v w x y z { | } ~ ¡ ¢ £ ¤ ¥ ¦ § ¨A¨ © ª « ¬ ® ¯ ° ± ² ³ ´ µ ¶ · ¸ ¹ º » ¼ ½ ¾ ¿ À Á Â Ã Ä Å Æ Ç È É Ê Ë Ì Í Î Ï Ð Ñ Ò Ó Ô Õ Ö × Ø Ù Ú Û Ü Ý Þ ß à á â ã ä å æ ç è é ê ë ì í î ï ð ñ ò ó ô õ ö ÷ ø ù ú û ü ý þ ÿ
- PreCheck 误判导致正常节点被剔除的风险如何规避?(需关注巡检容错与灰度恢复机制)
- Greedy Rollout 在长尾负载下是否容易陷入局部最优?(异构资源调度的全局视野补充)
- 潮汐架构能否扩展对齐跨可用区冷热流量?(更大范围资源复用的可行性)
- 异步 ckpt 是否能低成本支撑训练时序回溯与消融实验?(挖掘 ckpt 副产品的科研价值)
- 探索 eBPF 或 GPU MIG 替代 Sidecar 隔离?(寻找更低开销的隔离方案)
- 网络拥塞时能否降级至 NVMe 临时落盘容灾?(不落盘传输的极端降级策略)
- 复杂调度与自愈系统的整体 ROI 如何量化评估?(避免过度工程化,需量化收益)
- 超长 Context 与超大 MoE 对当前调度与通信拓扑的冲击?(架构演进的前瞻性思考)