atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

Parameter-Efficient Multi-Task Model Fusion with Partial Linearization

ICLR 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Parameter-efficient transfer analysis · 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 L-LoRA, which linearizes only the LoRA adapter modules (first-order Taylor in trainable parameters) so that task arithmetic and other merging methods fuse PEFT task vectors better than standard LoRA while keeping near-LoRA cost.

Problem

Parameter-efficient fine-tuned models merge poorly for multi-task fusion because of representational interference between tasks; PEFT fusion lags far behind full fine-tuning fusion.

Approach

Partially linearize only the adapter (LoRA) parameters via a first-order Taylor expansion around their initialization while keeping the pre-trained backbone nonlinear, so that task vectors in the tangent space are more orthogonal and fuse better with task arithmetic and related methods, at low extra cost.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Partially linearize LoRA modules instead of using standard LoRACLIP-ViT-B-16 multi-task fusion · Vision tasks, all combinationsaverage normalized score: Simple average: 0.58±0.18 -> 0.66±0.02; Task Arithmetic: 0.64±0.03 -> 0.77±0.05; Ties-merging: 0.63±0.03 -> 0.73±0.04; LHelped
Partially linearize LoRA modules instead of using standard LoRAFlan-T5-Base multi-task fusion · NLP tasks, all combinationsaverage normalized score: Simple average: 0.06±0.07 -> 0.05±0.07; Task Arithmetic: 0.21±0.07 -> 0.26±0.05; Ties-merging: 0.21±0.08 -> 0.24±0.06; LMixed
Partially linearize LoRA modules instead of using standard LoRACLIP and Flan-T5 task-specific models · Vision and NLP task pairstask-vector cosine similarityHelped
Partially linearize LoRA modules instead of using standard LoRASingle-task fine-tuning · Vision and NLP downstream taskstask-specific performanceHurt

Surprising findings

Limitations

Experimental setup

Datasets: Stanford Cars DTD EuroSAT GTSRB RESISC45 SUN397 SVHN CoLA MNLI MRPC QQP RTE

Benchmarks: Stanford Cars DTD EuroSAT GTSRB RESISC45 SUN397 SVHN CoLA MNLI MRPC QQP RTE

Metrics: top-1 accuracy Spearman's rho normalized score weight disentanglement error

Baselines: simple average task arithmetic ties-merging LoraHub LoRA fine-tuning full fine-tuning

Builds on: task arithmetic ties-merging LoraHub LoRA neural tangent kernel linearization

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