arXiv:2602.04816cs.OScs.CL2026-02

让一台机器训练1200亿参数大模型,靠的是把内存当主战场。

Horizon-LM: A RAM-Centric Architecture for LLM Training

  • 用主机内存存模型权重,GPU只临时干活,不长期驻留数据
  • 单张H200+1.5TB内存可稳定训练120B模型,吞吐比传统方案高12.2倍
  • 适合个人或小团队做微调、对齐等不需要多卡的训练任务

大语言模型规模的快速扩张已超越单卡硬件的发展速度,模型规模日益受内存容量而非算力限制。现有训练系统虽通过分布式并行与跨CPU/存储层级的卸载扩展GPU内存,但依然采用以GPU为中心的执行范式,即GPU持续驻留模型副本和完整的自动求导图。这导致模型扩展仍高度依赖多卡集群、复杂的分布式运行时和不可预测的主机内存消耗,为单节点后训练任务(如指令微调、对齐、领域适配)设置了重大障碍。我们提出Horizon-LM,一种以内存为中心的训练系统,重新定义了CPU与GPU在大模型优化中的角色。Horizon-LM将主机内存作为权威参数存储,采用CPU主控、GPU模板的执行模式,使GPU仅作为瞬时计算引擎。通过移除持久的GPU驻留模块和自动求导图,采用显式重计算与手动梯度传播,并引入流水线双缓冲执行引擎,Horizon-LM实现了模型规模与GPU数量解耦,将内存使用限制在理论参数占用范围内。在单张H200 GPU搭配1.5TB主机内存的环境下,Horizon-LM可稳定训练高达1200亿参数的模型;在标准单A100机器上,其训练吞吐量比启用了CPU卸载的DeepSpeed ZeRO-3高出12.2倍,同时保持数值正确性。在多种平台和规模下,Horizon-LM均实现高设备利用率和可预测的内存增长,证明主机内存而非GPU内存才是单节点大规模模型训练的真实可行性边界。

原文摘要 · Abstract (English)

The rapid growth of large language models (LLMs) has outpaced the evolution of single-GPU hardware, making model scale increasingly constrained by memory capacity rather than computation. While modern training systems extend GPU memory through distributed parallelism and offloading across CPU and storage tiers, they fundamentally retain a GPU-centric execution paradigm in which GPUs host persistent model replicas and full autograd graphs. As a result, scaling large models remains tightly coupled to multi-GPU clusters, complex distributed runtimes, and unpredictable host memory consumption, creating substantial barriers for node-scale post-training workloads such as instruction tuning, alignment, and domain adaptation. We present Horizon-LM, a memory-centric training system that redefines the roles of CPU and GPU for large-model optimization. Horizon-LM treats host memory as the authoritative parameter store and uses GPUs solely as transient compute engines through a CPU-master, GPU-template execution model. By eliminating persistent GPU-resident modules and autograd graphs, employing explicit recomputation with manual gradient propagation, and introducing a pipelined double-buffered execution engine, Horizon-LM decouples model scale from GPU count and bounds memory usage to the theoretical parameter footprint. On a single H200 GPU with 1.5\,TB host RAM, Horizon-LM reliably trains models up to 120B parameters. On a standard single A100 machine, Horizon-LM achieves up to 12.2$\times$ higher training throughput than DeepSpeed ZeRO-3 with CPU offloading while preserving numerical correctness. Across platforms and scales, Horizon-LM sustains high device utilization and predictable memory growth, demonstrating that host memory, not GPU memory, defines the true feasibility boundary for node-scale large-model training.

大模型训练内存优化单机训练高效推理

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