arXiv:2605.16184cs.DCcs.LG2026-05被引 1

让大模型二阶优化训练变高效,靠的是动态调度优化器状态和异步计算。

Runtime-Orchestrated Second-Order Optimization for Scalable LLM Training

  • 将优化器状态分发到GPU、CPU和硬盘,按需调度
  • 在单卡上实现10亿参数模型的二阶训练,多卡加速收敛
  • 适合需要高效训练大模型的研究者和工程师

二阶优化方法能提升大语言模型训练效率,但因需维护大型矩阵型优化器状态,系统开销大而难应用。我们提出Asteria,一个运行时系统,通过将二阶优化逻辑与核心训练路径解耦,解决此瓶颈。Asteria不将所有预条件器状态保留在加速器上,而是根据架构约束和实时压力,动态分配至GPU内存、CPU内存及可选的NVMe存储。它利用训练钩子提前准备影子状态,使昂贵的逆根计算可在主机异步进行,同时GPU继续计算。在分布式训练中,Asteria采用有界延迟协议,限制同步频率,通过拓扑感知协调保持优化有效性。我们在内存受限和分布式训练场景下评估了Asteria。在单块GB10 GPU与128GB统一内存的DGX Spark平台上,支持10亿参数语言模型的二阶训练。在多节点GH200系统上,降低了可见优化器开销,减少重复性延迟峰值,加快了实际时间下的收敛速度,并在70亿参数模型上维持SOAP与KL-Shampoo的优化优势。结果表明,二阶大模型训练的实用性不只依赖简化优化器,更在于重思运行时层面的优化器状态管理、后台计算与分布式同步策略。

原文摘要 · Abstract (English)

Second-order methods offer an attractive path toward more sample-efficient LLM training, but their practical use is often blocked by the systems cost of maintaining and updating large matrix-based optimizer states. We introduce \textbf{Asteria}, a runtime system designed to remove this bottleneck by separating second-order optimization logic from the critical GPU training path. Rather than keeping all preconditioner state on the accelerator, Asteria dynamically distributes optimizer state across GPU memory, CPU memory, and optional NVMe storage according to architectural constraints and runtime pressure. It further uses training hooks to prepare shadow states in advance, allowing expensive inverse-root computations to proceed asynchronously on the host while GPU computation continues. For distributed training, Asteria employs a bounded-staleness protocol that limits synchronization frequency while preserving optimizer effectiveness through topology-aware coordination. We evaluate Asteria on both memory-constrained and distributed training settings. On a DGX Spark platform with a single GB10 GPU and 128GB unified memory, Asteria supports second-order training for a 1B-parameter language model. On multi-node GH200 systems, it lowers visible optimizer overhead, reduces recurring latency spikes, accelerates convergence in wall-clock time, and maintains the optimization advantages of SOAP and KL-Shampoo in a 7B-parameter language model. Our results suggest that second-order LLM training can be made practical not by simplifying the optimizer alone, but by rethinking how optimizer state, background computation, and distributed synchronization are managed at the runtime level.

大模型训练二阶优化系统优化分布式

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。