atlas

Research map / Large language models / Instruction and fine-tuning

Parameter Competition Balancing for Model Merging

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Fine-tuning generalization and delta compression · 6 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 PCB-Merging, a training-free, dataless model merging method that drops and rescales task-vector parameters using intra-task importance (softmax over squared magnitudes scaled by task count) and inter-task similarity scores, optionally refined by CMA-ES coefficient search.

Problem

Existing task-vector model merging methods apply uniform coefficients per task and parameter and do not balance parameter competition within and across tasks, limiting merged-model performance.

Approach

Computes a per-parameter balancing matrix combining intra-task importance (softmax of normalized squared task-vector magnitudes, scaled by number of tasks) and inter-task similarity (products of task vectors across tasks), then drops low-scoring parameters and rescales the rest before summing task vectors into the pretrained model.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
apply parameter competition balancing during model mergingT5-Base · 7 NLP tasks, no validationaverage performance: 69.5 -> 73.8Helped
apply parameter competition balancing during model mergingT5-Large · 7 NLP tasks, no validationaverage performance: 73.6 -> 77.1Helped
apply parameter competition balancing during model mergingT0-3B IA3 adapters · 11 PEFT tasks, no validationaverage performance: 64.9 -> 66.1Helped
apply parameter competition balancing during model mergingLlama-2-7B · 3 LLM tasksaverage performance: 34.2 -> 35.1Helped
apply parameter competition balancing during model mergingViT-B/32 · 8 vision tasks, no validationaverage performance: 72.4 -> 75.9Helped
apply parameter competition balancing during model mergingViT-L/14 · 8 vision tasks, no validationaverage performance: 86.0 -> 86.9Helped
replace softmax with sigmoid in intra-balancing (ablation)PCB-Merging · ViT-B/32 and T5-Base task suitesaverage performance: 76.3 -> 76.1; 75.4 -> 75.3No clear effect
replace softmax with ReLU in intra-balancing (ablation)PCB-Merging · ViT-B/32 and T5-Base task suitesaverage performance: 76.3 -> 74.9; 75.4 -> 72.8Hurt
replace softmax with tanh in intra-balancing (ablation)PCB-Merging · ViT-B/32 and T5-Base task suitesaverage performance: 76.3 -> 76.1; 75.4 -> 75.2No clear effect
replace softmax with sigmoid in inter-balancing (ablation)PCB-Merging · ViT-B/32 and T5-Base task suitesaverage performance: 76.3 -> 76.2; 75.4 -> 75.3No clear effect
replace softmax with ReLU in inter-balancing (ablation)PCB-Merging · ViT-B/32 and T5-Base task suitesaverage performance: 76.3 -> 76.1; 75.4 -> 75.2No clear effect
replace softmax with tanh in inter-balancing (ablation)PCB-Merging · ViT-B/32 and T5-Base task suitesaverage performance: 76.3 -> 76.4; 75.4 -> 75.4No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: T0 mixture tasks P3 prompt templates GLUE (MRPC, RTE, CoLA, SST-2) CMMLU GSM8K HumanEval eight vision classification datasets (SUN397, Cars, RESISC45, EuroSAT, SVHN, GTSRB, MNIST, DTD) emotion classification domain datasets (Ober et al.)

Benchmarks: T0-style NLP tasks (paws, qasc, quartz, story_cloze, wiki_qa, winogrande, wsc) (IA)^3 eleven tasks (RTE, CB, ANLI, COPA, HellaSwag, StoryCloze, WSC, Winogrande, WiC) CMMLU GSM8K HumanEval SUN397 Cars RESISC45 EuroSAT SVHN GTSRB MNIST

Metrics: accuracy average normalized accuracy F1 Matthews correlation

Baselines: Averaging Task Arithmetic TIES-Merging Fisher Merging RegMean AdaMerging DARE LoraHub

Builds on: Task Arithmetic TIES-Merging DARE AdaMerging RegMean

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