arXiv:2511.14124cs.DCcs.LG2025-11中稿 · presentation to th…被引 2

10Cache智能调度显存、内存与固态存储,显著提升大模型训练速度与资源利用率。

10Cache: Heterogeneous Resource-Aware Tensor Caching and Migration for LLM Training

  • 基于张量执行顺序预估,动态规划缓存与迁移策略。
  • 训练速度最高提升2倍,显存命中率提高86.6倍,内存利用率提升超2倍。
  • 适合云环境大规模部署,降低对高端GPU依赖,节省训练成本。

在云端训练大语言模型(LLM)面临日益严重的内存瓶颈,受限于GPU容量和高昂成本。尽管现有方法将张量数据卸载至CPU和NVMe可缓解问题,但普遍存在高迁移延迟和内存利用不充分的问题,导致训练时间延长与云成本上升。为此,我们提出10Cache——一种面向异构资源的张量缓存与迁移系统,通过智能协调GPU、CPU与NVMe三级内存,加速大模型训练。10Cache通过分析张量执行顺序构建预取策略,基于张量大小分布分配固定内存缓冲区,并复用缓冲区以减少分配开销。针对云规模部署设计,显著提升内存效率并减少对高端GPU的依赖。在多种典型LLM工作负载下,相比当前最优卸载方法,10Cache实现最高2倍的训练速度提升,显存命中率提升达86.6倍,CPU与GPU内存利用率分别提高2.15倍和1.33倍。结果表明,10Cache是优化云环境中大模型训练吞吐量与资源效率的实用且可扩展方案。

原文摘要 · Abstract (English)

Training large language models (LLMs) in the cloud faces growing memory bottlenecks due to the limited capacity and high cost of GPUs. While GPU memory offloading to CPU and NVMe has made large-scale training more feasible, existing approaches suffer from high tensor migration latency and suboptimal device memory utilization, ultimately increasing training time and cloud costs. To address these challenges, we present 10Cache, a resource-aware tensor caching and migration system that accelerates LLM training by intelligently coordinating memory usage across GPU, CPU, and NVMe tiers. 10Cache profiles tensor execution order to construct prefetch policies, allocates memory buffers in pinned memory based on tensor size distributions, and reuses memory buffers to minimize allocation overhead. Designed for cloud-scale deployments, 10Cache improves memory efficiency and reduces reliance on high-end GPUs. Across diverse LLM workloads, it achieves up to 2x speedup in training time, improves GPU cache hit rate by up to 86.6x, and increases CPU/GPU memory utilization by up to 2.15x and 1.33x, respectively, compared to state-of-the-art offloading methods. These results demonstrate that 10Cache is a practical and scalable solution for optimizing LLM training throughput and resource efficiency in cloud environments.

大模型训练内存优化异构调度云部署

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