Super-Tuning: From Activation-Aware Pruning to Sparse Fine-Tuning
Summary
Super and Supra are novel parameter-efficient fine-tuning (PEFT) methods addressing the high memory, compute, and storage costs of large language model (LLM) fine-tuning. Super identifies a small trainable weight subset using a Wanda-style activation-weighted magnitude score from a calibration pass, avoiding gradient computations. Supra is a hybrid adapter combining Super's sparse updates with LoRA, maintaining a matched trainable-parameter budget through a simple splitting rule. Experiments on Llama-3.2-1B and Meta-Llama-3-8B with the Math17K dataset show that the best Super/Supra variants achieve the highest average accuracy among tested configurations. Specifically, Supra (BottomK, \u03bb=0.8) reached 62.23% on Llama-3.2-1B, outperforming LoRA by 1.16 percentage points. Supra-Mag (BottomK, \u03bb=0.3) achieved 79.12% on Meta-Llama-3-8B, surpassing SIFT (RandK) by 1.29 points. These results suggest pruning-inspired orderings provide effective fixed sparse supports for PEFT.
Key takeaway
For AI Engineers optimizing LLM deployment costs, consider integrating Super or Supra into your PEFT strategy. These methods offer competitive or superior arithmetic reasoning accuracy on Llama-family models (e.g., 62.23% on Llama-3.2-1B) compared to LoRA, while maintaining matched trainable parameter budgets. You should explore BottomK sparse supports, potentially combined with low-rank adapters, to efficiently adapt models without complex gradient-based mask computations.
Key insights
Pruning-inspired activation-aware or magnitude-only scores effectively identify fixed sparse supports for LLM fine-tuning.
Principles
- Low-score weights are effective for adaptation.
- Sparse and low-rank updates can be complementary.
- Training-free mask selection is viable.
Method
Super generates a fixed sparse mask using a Wanda-style activation-weighted magnitude score from a calibration pass. Supra combines this sparse mask with LoRA, splitting a matched parameter budget.
In practice
- Apply BottomK selection for sparse supports.
- Combine sparse updates with LoRA for hybrid PEFT.
- Use C4 calibration for Wanda-style masks.
Topics
- Parameter-Efficient Fine-Tuning
- Sparse Adaptation
- Low-Rank Adaptation
- Large Language Models
- Wanda Pruning
- Arithmetic Reasoning
Code references
Best for: Research Scientist, NLP Engineer, AI Scientist, Machine Learning Engineer, AI Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.