Papers
Topics
Authors
Recent
Search
2000 character limit reached

GRASP LoRA: Guided Adapter Sparsity via GRPO

Updated 17 January 2026
  • The paper proposes GRASP LoRA, a method that leverages GRPO for dynamically optimizing adapter sparsity, achieving up to 7× faster fine-tuning on cross-lingual tasks.
  • GRASP LoRA is a parameter-efficient approach that merges English and target language LoRA adapters and employs magnitude-based pruning with learnable global prune ratios.
  • Experimental results demonstrate improved metrics in summarization and QA tasks while significantly reducing computational costs, highlighting its practical efficiency.

GRASP LoRA (GRPO Guided Adapter Sparsity Policy) is a parameter-efficient fine-tuning methodology designed for cross-lingual transfer of LLMs under limited computational and data resources. Unlike conventional adapter pruning pipelines that rely on grid search over sparsity ratios—an approach both resource-intensive and coarse—GRASP LoRA transforms the global sparsity ratio into a learnable control variable optimized online by a Group-Relative Policy Optimization (GRPO) controller using minimal development data (Hassan et al., 10 Jan 2026).

1. GRPO Controller: Mathematical Formulation and Optimization

At the core of GRASP LoRA is a stochastic policy for the global prune ratio ρ[pmin,pmax][0,1]\rho \in [p_{\min}, p_{\max}] \subseteq [0, 1], parameterized by a univariate Gaussian: ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0. Every KK optimizer steps, the controller samples CC candidate prune ratios {ρi}i=1C\{\rho_i\}_{i=1}^C as ziN(μ,σ2)z_i \sim \mathcal{N}(\mu, \sigma^2) and ρi=clamp(zi,pmin,pmax)\rho_i = \mathrm{clamp}(z_i, p_{\min}, p_{\max}). Magnitude-thresholded binary masks M(ρi)M(\rho_i) are constructed over the merged LoRA weights W~\widetilde{W}, each inducing a pruned subnetwork. Candidate losses i\ell_i and a baseline loss ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0.0 are evaluated on a fixed micro development slice of ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0.1 target-language examples: ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0.2 serves as the reward signal. The policy is optimized via the GRPO surrogate: ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0.3 where ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0.4 regularizes towards the current ratio and ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0.5 encourages entropy. Score-function gradients ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0.6, ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0.7 are computed according to centered advantages, and parameter updates are constrained within admissible bounds. Commitment to a new prune ratio occurs only if no micro-dev loss increase is observed, with bounded step size ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0.8.

2. End-to-End Algorithmic Workflow

GRASP LoRA interleaves adapter fine-tuning, policy-guided pruning, and evaluation in three main phases:

  1. Adapter Training and Merging
    • English LoRA adapters are trained on high-resource English data with the backbone model frozen.
    • Target-language LoRA adapters are trained on low-resource target language data, again with a frozen backbone.
    • The adapters are merged by summing their low-rank update matrices at each projection site: ρπμ,σ=N(μ,σ2),μ[pmin,pmax],  σ>0.\rho \sim \pi_{\mu,\sigma} = \mathcal{N}(\mu, \sigma^2), \qquad \mu\in[p_{\min},p_{\max}],\; \sigma>0.9.
  2. Sparsity Policy Learning (Controller Rounds)
    • Initialize KK0, with controller parameters set to KK1.
    • In each controller round:
      • Fine-tune the merged adapters on target data under the current mask KK2.
      • Every KK3 steps, probe KK4 candidate prune ratios, evaluate their corresponding micro-dev loss, and update the controller policy using Eqs. (1)-(2).
      • Commit to a new KK5 if improvement is observed, update masks, and clear optimizer states for new zeros.
    • Upon completion, select KK6 via post-hoc validation loss minimization.
  3. Final Pruning and Fine-tuning
    • Reload the frozen backbone and pre-controller merged adapters.
    • Build the final mask KK7 and fine-tune the masked model on the full target data until early stopping on a held-out dev set.

3. Adapter Merging and Magnitude-based Pruning

LoRA adapter merging is performed by summing the low-rank update matrices for source and target languages: KK8 Pruning is implemented tensor-wise: for prune ratio KK9, the mask is determined by retaining the top CC0 fraction of (magnitude-ordered) entries per tensor. Let CC1 be the number of parameters in tensor CC2, CC3, and CC4 the CC5th order statistic of CC6. Then,

CC7

This mask is applied to all adapters on the (frozen) backbone model.

4. Experimental Protocol and Hyperparameters

The evaluation covers cross-lingual transfer for summarization and extractive QA:

  • Datasets:
    • XL-Sum (English→Arabic, English→Chinese): English train/dev 10k/1k; Arabic/Chinese train 50, dev 50, micro 16, test 100.
    • MLQA (extractive QA): English train 3k; Arabic/Chinese train 50, micro 16, test 100.
  • Model and PEFT Setup:
    • Backbone: Llama 3 8B (frozen).
    • LoRA applied to Q and V projections; rank 8, CC8, dropout 0.05.
  • Optimization and Controller:
    • Adapter fine-tuning: 10 epochs, lr CC9, AdamW, batch size 1, max input 2200 tokens.
    • GRPO settings: prune range {ρi}i=1C\{\rho_i\}_{i=1}^C0, {ρi}i=1C\{\rho_i\}_{i=1}^C1, probe interval {ρi}i=1C\{\rho_i\}_{i=1}^C2, {ρi}i=1C\{\rho_i\}_{i=1}^C3 candidates, micro-dev {ρi}i=1C\{\rho_i\}_{i=1}^C4, {ρi}i=1C\{\rho_i\}_{i=1}^C5, controller lr {ρi}i=1C\{\rho_i\}_{i=1}^C6.
    • Regularization {ρi}i=1C\{\rho_i\}_{i=1}^C7 is tuned: for Arabic XL-Sum {ρi}i=1C\{\rho_i\}_{i=1}^C8, Chinese XL-Sum {ρi}i=1C\{\rho_i\}_{i=1}^C9, Arabic MLQA ziN(μ,σ2)z_i \sim \mathcal{N}(\mu, \sigma^2)0, Chinese MLQA ziN(μ,σ2)z_i \sim \mathcal{N}(\mu, \sigma^2)1.
  • Evaluation Metrics:
    • Summarization: BERTScore-F1, BLEU-4, ROUGE-L (and additional variants).
    • QA: BERTScore-F1, Exact Match, token F1 (plus BLEU/ROUGE/chrF for spans).
  • Prompt Structure: Unchanged across languages; e.g., for summarization, "Article:{article} → Summary:"; for QA, "Context:{context} Question:{question} → Answer:".

5. Empirical Results and Performance Analysis

GRASP LoRA demonstrates consistent improvements over strong merge-and-prune grid search baselines:

Task Baseline Prune GRASP Prune BERT-F1 Δ BLEU-4 Δ ROUGE-L Δ EM Δ F1 Δ Time Δ
XL-Sum Arabic 70% 67.49% +0.88 +1.75 +2.13 3.90× faster
XL-Sum Chinese 50% 56.94% +1.62 +1.73 +1.45 5.66× faster
MLQA Arabic 40% 48.97% +0.56 +2.67 +2.22 6.40× faster
MLQA Chinese 10% 23.73% +1.98 +1.50 +0.67 7.45× faster

Additional findings include:

  • The approach reduces end-to-end runtime by a factor of 4–7× compared to grid search baselines.
  • Improvements are robust with respect to micro-dev size, with ziN(μ,σ2)z_i \sim \mathcal{N}(\mu, \sigma^2)2 and BERT-F1 stable across ziN(μ,σ2)z_i \sim \mathcal{N}(\mu, \sigma^2)3.
  • Regularization ablations show that removing entropy or mean anchoring leads to excessive pruning (~79%) and a 1–2 point drop in evaluation metrics.
  • Qualitative analysis shows superior semantic faithfulness in summarization and more accurate answer extraction in QA compared to baselines.

6. Implementation Considerations and Practical Implications

Key features for faithful and efficient deployment include:

  • Use of a small, fixed micro-dev slice (16 examples) for all controller evaluations, independent of the early-stop dev set.
  • All controller reward computations, pruning evaluations, and commitment decisions are logged, enabling post-hoc ziN(μ,σ2)z_i \sim \mathcal{N}(\mu, \sigma^2)4 selection if needed.
  • Identical prompt templates and consistent adapter architectures facilitate experimentation across unrelated linguistic domains.
  • The learnable sparsity policy makes it feasible to select fractional sparsity optima impractical under conventional discrete grid search, especially in low-resource settings.

A plausible implication is that GRASP LoRA extends reliable adapter reuse to previously intractable low-resource regimes by decoupling sparsity hyperparameter tuning from costly grid search. The method offers a systematic pathway for tuning adapter sparsity using only minimal dev resources, providing improved model quality, content coverage, and answer quality relative to strong baselines and yielding major reductions in both computational and annotation costs (Hassan et al., 10 Jan 2026).

Definition Search Book Streamline Icon: https://streamlinehq.com
References (1)

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to GRASP LoRA (GRPO Guided Adapter Sparsity Policy).