atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

LoQT: Low-Rank Adapters for Quantized Pretraining

NeurIPS 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Low-rank compression and merging · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes LoQT, which trains low-rank factors on top of NF4-quantized full-rank weights (GaLore-style gradient-projected P, trainable B, periodic merging on an exponentially growing schedule) to enable memory-efficient quantized pretraining and fine-tuning on a 24GB GPU.

Problem

Pretraining large models on consumer hardware is infeasible without sharding, offloading, or per-layer gradient updates, and low-rank adapters like LoRA do not work well for pretraining from scratch; updating quantized weights directly is also hard because small gradient updates vanish in low precis

Approach

Initialize a fixed projection P from the SVD of the gradient (GaLore-style) and a trainable B to minimize quantization error, train only B, periodically merge PB into the NF4-quantized W with error compensation, and use an exponentially increasing merge interval so accumulated updates survive quantization.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use LoQT instead of full-rank trainingLLaMA2-style pretraining · C4, 60Mperplexity: 33.32 $\pm$ 0.22 -> 33.98 $\pm$ 0.15Hurt
Use LoQT instead of full-rank trainingLLaMA2-style pretraining · C4, 130Mperplexity: 24.51 $\pm$ 0.03 -> 24.57 $\pm$ 0.01No clear effect
Use LoQT instead of full-rank trainingLLaMA2-style pretraining · C4, 1Bperplexity: 15.56 -> 15.55No clear effect
Quantize the model weights (LoQT versus PLoRA)LoQT pretraining · C4, 350Mperplexity: 18.85 $\pm$ 0.01 -> 19.12 $\pm$ 0.01Hurt
Use LoQT instead of LoRADeBERTaV3-base fine-tuning · GLUE development averageaverage score: 88.3 -> 88.7Helped
Remove quantization (ablation)LoQT fine-tuning · GLUE development averageaverage score: 88.7 -> 89.1Helped
Omit merging updates into pretrained weights (ablation)LoQT fine-tuning · GLUE development averageaverage score: 88.7 -> 88.9Hurt
Add quantization error compensation (ablation)LoQT pretraining · C4, 130Mperplexity: approximately 3.5 points better perplexityHelped
Use exponentially increasing update intervals (ablation)LoQT pretraining · C4, 130Mperplexity: additional 1.5 pointsHelped
Quantize P as well as W (ablation)LoQT pretraining · C4, 130Mvalidation lossHurt
Use exponentially increasing rather than fixed projection gaps (ablation)LoQT and GaLore · C4 pretrainingperplexityHelped
Reduce adapter rank to 64 (ablation)LoQT · C4, 130MperplexityHurt

Surprising findings

Limitations

Experimental setup

Datasets: C4 GLUE GSM8K Icelandic text dataset (curated subset)

Benchmarks: C4 validation perplexity GLUE GSM8K Icelandic perplexity

Metrics: validation perplexity accuracy Matthews correlation F1 Pearson correlation memory (GB) tokens per second

Baselines: Full-rank training LoRA ReLoRA GaLore LoftQ QLoRA

Builds on: LoRA GaLore ReLoRA LoftQ QLoRA

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy