An Efficient Heterogeneous Co-Design for Fine-Tuning on a Single GPU
Not on the map. Why?
Abstract (arXiv)
Fine-tuning Large Language Models (LLMs) has become essential for domain adaptation, but its memory-intensive property exceeds the capabilities of most GPUs. To address this challenge and democratize LLM fine-tuning, we present SlideFormer, a novel system designed for single-GPU environments. Our innovations are: (1) A lightweight asynchronous engine that treats the GPU as a sliding window and overlaps GPU computation with CPU updates and multi-tier I/O. (2) A highly efficient heterogeneous memory management scheme significantly reduces peak memory usage. (3) Optimized Triton kernels to solve key bottlenecks and integrated advanced I/O. This collaborative design enables fine-tuning of the latest 123B+ models on a single RTX 4090, supporting up to 8x larger batch sizes and 6x larger models. In evaluations, SlideFormer achieves 1.40x to 6.27x higher throughput while roughly halving CPU/GPU memory usage compared to baselines, sustaining >95% peak performance on both NVIDIA and AMD GPUs.The code is available at https://github.com/RegiaYoung/SlideFormer.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 16 of its 35 references that are in the map. First posted on arXiv 17 Mar 2026.
Builds on
16 of 35 references.
References · 16
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,747
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,278
- 2017Mixed Precision Trainingcited by 1,135
- 2016Training Deep Nets with Sublinear Memory Costcited by 825
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 2021ZeRO-Offload: Democratizing Billion-Scale Model Trainingcited by 339
- 2021ZeRO-Infinity: Breaking the GPU Memory Wall for Extreme Scale Deep Learningcited by 253
4 more
- 2024LoRA vs Full Fine-tuning: An Illusion of Equivalencecited by 102
- 2021Colossal-AI: A Unified Deep Learning System For Large-Scale Parallel Trainingcited by 96
- 2024Smart-Infinity: Fast Large Language Model Training using Near-Storage Processing on a Real Systemcited by 14
- 2024LoHan: Low-Cost High-Performance Framework to Fine-Tune 100B Model on a Consumer GPUcited by 10
Built on by
Not cited yet.