atlas

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

Model Tailor: Mitigating Catastrophic Forgetting in Multi-modal Large Language Models

ICML 2024 · Continual learning › Continual fine-tuning of LLMs · approach: Continual pretraining of language models · 2 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 Model Tailor, a post-training, sparse-mask method that keeps pre-trained MLLM weights and replaces about 10% of fine-tuned parameters (chosen by fusing salience and OBS-style sensitivity scores, with Hessian-based compensation) to mitigate catastrophic forgetting in InstructBLIP and LLaVA-1.5.

Problem

Fine-tuning multi-modal LLMs on new target tasks degrades their performance on original pre-trained tasks (catastrophic forgetting), and full fine-tuning and LoRA do not solve this well.

Approach

Select a sparse subset of fine-tuned parameters (the "model patch") using a hybrid of salience (|sft - pre|) and second-order OBS sensitivity scores, then compensate the selected parameters with Hessian-inverse-based updates (the "patch decorator") while reverting all other parameters to pre-trained values.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
apply Model Tailor instead of GraftingInstructBLIP (Vicuna-7B) · Flickr30k targetH-score: 92.78 -> 93.67Helped
apply Model Tailor instead of DAREInstructBLIP (Vicuna-7B) · GQA targetH-score: 66.46 -> 71.00Helped
apply Model Tailor instead of GraftingLLaVA-1.5 (Vicuna-7B) · Flickr30k targetH-score: 60.03 -> 66.94Helped
apply Model Tailor instead of GraftingLLaVA-1.5 (Vicuna-7B) · OKVQA targetH-score: 41.25 -> 47.71Helped
add patch decoration (weight compensation) (ablation)Model Tailor on InstructBLIP · Flickr30k targetH-score: 91.96 -> 93.67Helped
add patch decoration (weight compensation) (ablation)Model Tailor on LLaVA-1.5 · Flickr30k targetH-score: 64.24 -> 66.94Helped
increase sparsity level (ablation)Model Tailor · pre-trained tasksperformanceHelped
increase sparsity level (ablation)Model Tailor · target tasksperformanceHurt
increase salience-score weight ω (ablation)Model Tailor mask selection · target tasksperformanceHelped
increase sensitivity-score weight by reducing ω (ablation)Model Tailor mask selection · pre-trained tasksperformanceHelped
combine Model Tailor with LoRALLaVA-1.5 fine-tuned with LoRA · Flickr30kperformance: +18.9-pointHelped
combine Model Tailor with LoRALLaVA-1.5 fine-tuned with LoRA · multiple datasetsH-score: +9.9%Helped

Surprising findings

Limitations

Experimental setup

Datasets: Flickr30k GQA OKVQA COCO Caption NoCaps AOKVQA VQAv2 VizWiz ScienceQA TextVQA POPE MM-Bench

Benchmarks: COCO Caption NoCaps OKVQA AOKVQA VQAv2 GQA Flickr30k VizWiz ScienceQA TextVQA POPE MM-Bench

Metrics: average performance H-score (harmonic mean of original and target task averages)

Baselines: Standard fine-tuning Model Grafting DARE

Builds on: Optimal Brain Surgeon Lottery Ticket Hypothesis SparseGPT Model Grafting DARE

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