Super-Tuning: From Activation-Aware Pruning to Sparse Fine-Tuning

· Source: cs.CL updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Software Development & Engineering · Depth: Expert, extended

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

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

Topics

Code references

Best for: Research Scientist, NLP Engineer, AI Scientist, Machine Learning Engineer, AI Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.