atlas

Research map / Continual learning / Continual fine-tuning of LLMs

Overcoming Generic Knowledge Loss with Selective Parameter Update

CVPR 2024 · Continual learning › Continual fine-tuning of LLMs · approach: Continual pretraining of language models · 10 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 Selective Parameter Update (SPU), which localizes continual updates of CLIP to the first MLP layer of each transformer block and sparsely updates the ~10% of those parameters with highest gradient-based scores, preserving generic zero-shot knowledge.

Problem

Continually updating foundation vision-language models (CLIP) on new class-incremental tasks causes loss of their generic pre-trained knowledge, which existing continual learning methods largely ignore.

Approach

Localize updates to the first MLP layer of each transformer block (viewed as key-value memories), then select a sparse subset of parameters by a training-free gradient-magnitude score computed before training, and update only those with replay.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
selectively update first MLP-layer parameters using gradient-based scoresSPU on CLIP-ViT/B-16 · 6 CIL sequences (average)Acc. In.: 21.34 vs. 18.48Helped
selectively update first MLP-layer parametersSPU on CLIP-ViT/B-16 · 6 CIL sequences (average)C. Drop: 0.94 vs. 17.56Helped
update attention layers instead of first MLP layers (ablation)SPU on CLIP-ViT/B-16 · 6 CIL sequences (average)Acc. In.: 21.34 -> 20.21Hurt
update second MLP layers instead of first MLP layers (ablation)SPU on CLIP-ViT/B-16 · 6 CIL sequences (average)C. Drop: 0.94 -> 1.85Hurt
update both MLP layers instead of only first MLP layers (ablation)SPU on CLIP-ViT/B-16 · 6 CIL sequences (average)C. Drop: 0.94 -> 1.51Hurt
select parameters randomly instead of using the gradient-based score (ablation)SPU on CLIP-ViT/B-16 · 6 CIL sequences (average)Acc. In.: 21.34 -> 17.73Hurt
use selection rate 0.01 instead of 0.10 (ablation)SPU on CLIP-ViT/B-16 · 6 CIL sequences (average)Acc. In.: 21.34 -> 17.70Hurt
use selection rate 0.50 instead of 0.10 (ablation)SPU on CLIP-ViT/B-16 · 6 CIL sequences (average)Avg. F.: 4.51 -> 7.76Hurt
reduce replay-buffer size from 4% to 1% (ablation)SPU on CLIP-ViT/B-16 · 6 CIL sequences (average)Acc. In.: 21.34 -> 16.18Hurt
reduce replay-buffer size from 4% to 2% (ablation)SPU on CLIP-ViT/B-16 · 6 CIL sequences (average)Acc. In.: 21.34 -> 18.63Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Birdsnap CUB-200-2011 FGVC-Aircraft Stanford Cars CIFAR100 GTSRB ImageNet

Benchmarks: Birdsnap CUB-200-2011 FGVC-Aircraft Stanford Cars CIFAR100 GTSRB ImageNet (control set)

Metrics: Average accuracy increment (Acc. In.) Average forgetting (Avg. F.) Control set accuracy drop (C. Drop) Accuracy

Baselines: FLYP FLYP+ER MAS LwF PRD LoRA-EWC L2P DualPrompt SLCA ZSCL SparseCL SPG

Builds on: FLYP Experience replay (ER) Causal tracing (ROME-style analysis) Key-value memory view of MLP layers Piggyback

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