atlas

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

Model Sensitivity Aware Continual Learning

NeurIPS 2024 · Continual learning › Continual fine-tuning of LLMs · approach: Low-rank and model-merging adaptation · 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 MACL, a plug-in continual learning method that minimizes worst-case loss over a Gaussian parameter-distribution neighborhood (KL-constrained) using natural gradient in expectation-parameter space, reducing parameter sensitivity to balance forgetting and new-task accuracy.

Problem

Continual learning faces a trade-off between retaining old-task knowledge and learning new tasks well; existing methods often sacrifice one for the other.

Approach

Treats CL performance as dependent on model parameter sensitivity and optimizes the worst-case Gaussian parameter distribution within a KL neighborhood of the current distribution, using natural gradient descent in expectation-parameter space (avoiding explicit FIM) with a diagonal Hessian approximation; the method is a plug-in to existing CL methods.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add MACL to LODELODE · CIFAR-10 Class-ILaccuracy: 75.45 ± 0.90 -> 76.41 ± 0.67Helped
add MACL to LODELODE · CIFAR-10 Task-ILaccuracy: 94.41 ± 0.22 -> 94.32 ± 0.24Hurt
add MACL to LODELODE · CIFAR-100 Class-ILaccuracy: 38.95 ± 0.93 -> 40.67 ± 0.89Helped
add MACL to LODELODE · CIFAR-100 Task-ILaccuracy: 78.92 ± 0.67 -> 40.03 ± 0.51Hurt
add MACL to ER-ACEER-ACE · Tiny-ImageNet Class-ILaccuracy: 20.43 ± 0.97 -> 21.89 ± 0.83Helped
add MACL to LODELODE · Tiny-ImageNet Task-ILaccuracy: 60.18 ± 0.65 -> 61.79 ± 0.86Helped
add MACL to DER++DER++ · CIFAR-100 Task-ILaccuracy: 75.64 ± 0.60 -> 77.53 ± 0.89Helped
add MACL to DER++DER++ · CIFAR-100 Class-ILaccuracy: 36.37 ± 0.85 -> 39.42 ± 0.82Helped
add MACL to GPMGPM · CIFAR-100 Task-ILaccuracy: 72.48 ± 0.40 -> 74.51 ± 0.36Helped
add MACL to DER++DER++ · CIFAR-10 Task-ILaccuracy: 93.88 ± 0.50 -> 94.72 ± 0.65Helped
add MACL to DER++DER++ · Tiny-ImageNet Class-ILaccuracy: 19.38 ± 1.41 -> 20.17 ± 1.56Helped
add MACL to DER++DER++ · Tiny-ImageNet Task-ILaccuracy: 51.91 ± 0.68 -> 54.03 ± 0.79Helped

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-10 CIFAR-100 Tiny-ImageNet 5-datasets CUB200 ImageNet-R

Benchmarks: CIFAR-10 CIFAR-100 Tiny-ImageNet 5-datasets CUB200 ImageNet-R

Metrics: overall accuracy new task accuracy backward transfer

Baselines: EWC SI LwF NCL HAT GPM ER DER++ ER-ACE LODE

Builds on: DER++ SAM EWC UDIL CODA-Prompt

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