mlsanityData to 14 Oct 2026

Long-Context Fine-Tuning with Limited VRAM

arXiv 2607.15105 ↗First posted 16 Jul 2026cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Parameter-efficient fine-tuning reduces model and optimizer memory, but dense attention still makes long training sequences expensive. We combine Hierarchical Global Attention (HGA) with segment-wise backpropagation and tiered KV storage. Only the active segment remains differentiable in VRAM; older KV is detached into RAM or NVMe, and HGA loads a bounded set of exact historical tokens for each query block. On Qwen3-8B with 4-bit QLoRA and PG19, dense training on a 16 GB Quadro RTX 5000 fits 2,048 tokens but fails at 4,096, whereas HGA reaches 16,384 tokens with 15.28 GB peak VRAM. Under evaluation the same adapter runs through 131,072 tokens on this card; VRAM is not constant but grows gently with the resident chunk summaries, so RAM and NVMe capacity set the practical limit beyond these lengths. At the shared 2K training length, HGA-trained and dense-trained adapters obtain 2.7405 and 2.7383 nat under the same dense-attention readout, while the stock model obtains 2.9541. At this boundary HGA training is already marginally faster (217.75 vs. 207.02 tokens/s), and the HGA-to-dense throughput ratio improves from 1K to 2K; because HGA keeps the attended historical set per token approximately constant while dense work per token grows, we expect this lead to widen as context grows. Dense attention is used for the main quality and retrieval comparisons so that they measure the learned weights and remain compatible with standard generation frameworks. HGA can also be used for retrieval and generation; an optimized production-grade serving implementation is under development.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 5 of its 10 references that are in the map. First posted on arXiv 16 Jul 2026.

Builds on

5 of 10 references.

References · 5

  1. 2025Qwen3 Technical Reportcited by 8,820
  2. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  3. 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
  4. 2019Compressive Transformers for Long-Range Sequence Modellingcited by 726
  5. 2023LongLoRA: Efficient Fine-tuning of Long-Context Large Language Modelscited by 244

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2024A Little Goes a Long Way: Efficient Long Context Training and Inference with Partial Contexts
  2. 2026Kwai Summary Attention Technical Report
  3. 2026Learning how to Forget: Fine-tuning for Long-Context Sparse Attention
  4. 2026Unifying Sparse Attention with Hierarchical Memory for Scalable Long-Context LLM Serving
  5. 2026Learning When to Attend: Conditional Memory Access for Long-Context LLMs
  6. 2026Out of the Memory Barrier: A Highly Memory Efficient Training System for LLMs with Million-Token Contexts
  7. 2026Virtual Head Attention
  8. 2025HGCA: Hybrid GPU-CPU Attention for Long Context LLM Inference