atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

Customizable Combination of Parameter-Efficient Modules for Multi-Task Learning

ICLR 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Low-rank compression and merging · 1 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 Customized Polytropon (C-Poly), a multi-task PEFT method that splits LoRA skill modules into task-common shared skills and one task-specific skill per task, with a jointly learned skill allocation matrix, evaluated on SuperGLUE and Super-NaturalInstructions.

Problem

Parameter-efficient fine-tuning is underexplored for multi-task learning, where negative transfer and the seesaw effect hurt performance when tasks share parameters.

Approach

Explicitly divides modular LoRA skills into a task-common skillset (routed by a learned task-by-skill allocation matrix) and a task-specific skillset (one exclusive skill per task, initialized as an identity/diagonal allocation), combining both per task, with Gumbel-sigmoid relaxation for the binary allocation.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add task-specific skills alongside task-common skillsPoly-style multi-task PEFT · SuperGLUEaccuracy (AVG): 56.99 -> 62.26Helped
Add task-specific skills alongside task-common skills (ablation)Poly-style multi-task PEFT · SuperNI (100 tasks), T5-LargeRouge1: 43.04 -> 49.34Helped
Increase task-specific skills from 1 to 2 while reducing common skills from 3 to 2 (ablation)C-Poly · SuperNI (100 tasks), T5-LargeRouge1: 49.34 -> 48.79Hurt
Increase task-specific skills from 1 to 3 while reducing common skills from 3 to 1 (ablation)C-Poly · SuperNI (100 tasks), T5-LargeRouge1: 49.34 -> 47.46Hurt
Increase task-specific skills from 1 to 2 while reducing common skills from 3 to 2 (ablation)C-Poly · SuperNI (100 tasks), FLAN-T5-LargeRouge1: 68.69 -> 68.47Hurt
Increase task-specific skills from 1 to 3 while reducing common skills from 3 to 1 (ablation)C-Poly · SuperNI (100 tasks), GLM-10BRouge1: 49.53 -> 46.85Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Super-NaturalInstructions SuperGLUE

Benchmarks: SuperGLUE Super-NaturalInstructions

Metrics: accuracy Rouge-1 Rouge-L Rouge-LSum

Baselines: LoRA MoE-LoRA Polytropon Multi-Head Routing Full fine-tuning

Builds on: Polytropon LoRA MoLoRA Multi-Head Routing Customized Gate Control (CGC)

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