about
AutoHete: An Automatic and Efficient Heterogeneous Training System for LLMs (arxiv.org)
44 points by PaulHoule on Mar 15, 2025 | hide | past | pdf | discuss on HN

In plain words: A system trains large language models on graphics cards with little memory by automatically deciding what to move to the main processor and re-compute, and scheduling tasks to overlap. It runs 1.32 to 1.91 times faster than the best similar systems.

Abstract

Transformer-based large language models (LLMs) have demonstrated exceptional capabilities in sequence modeling and text generation, with improvements scaling proportionally with model size. However, the limitations of GPU memory have restricted LLM training accessibility for many researchers. Existing heterogeneous training methods significantly expand the scale of trainable models but introduce substantial communication overheads and CPU workloads. In this work, we propose AutoHete, an automatic and efficient heterogeneous training system compatible with both single-GPU and multi-GPU environments. AutoHete dynamically adjusts activation checkpointing, parameter offloading, and optimizer offloading based on the specific hardware configuration and LLM training needs. Additionally, we design a priority-based scheduling mechanism that maximizes the overlap between operations across training iterations, enhancing throughput. Compared to state-of-the-art heterogeneous training systems, AutoHete delivers a 1.32x~1.91x throughput improvement across various model sizes and training configurations.

Zihao Zeng, Chubo Liu, Xin He, Juan Hu, Yong Jiang, Fei Huang, Kenli Li, Wei Yang Bryan Lim
arXiv:2503.01890 · cs.LG · submitted Feb 27, 2025
abstract · pdf · html

add comment on HN