atlas

Research map / Vision-language and multimodal / Multimodal LLMs

Wings: Learning Multimodal LLMs without Text-only Forgetting

NeurIPS 2024 · Vision-language and multimodal › Multimodal LLMs · approach: Unified any-modality 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 Wings, an MLLM with parallel visual and textual low-rank residual attention learners routed by attention weights, to prevent text-only forgetting while retaining multimodal performance.

Problem

Multimodal LLMs catastrophically forget text-only instruction-following abilities of their initial LLM when trained on multimodal instructions, and existing fixes require extra text/interleaved data or separate models.

Approach

Identifies a cross-layer attention shift (Pearson correlation of layer-level attention before vs. after the image) linked to text-only forgetting, and adds parallel visual and textual Low-Rank Residual Attention (LoRRA) learners in each attention block, combined by a router driven by the main attention weights.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Wings visual and textual learners with attention-based routing instead of Qwen + SigLIPQwen1.5 MLLM · MMLUaccuracy: 50.83 -> 60.53Helped
Use Wings visual and textual learners with attention-based routing instead of Qwen + SigLIPQwen1.5 MLLM · CMMLUaccuracy: 60.46 -> 69.82Helped
Use Wings visual and textual learners with attention-based routing instead of Qwen + SigLIPQwen1.5 MLLM · Race-Highaccuracy: 57.72 -> 69.62Helped
Use Wings visual and textual learners with attention-based routing instead of Qwen + SigLIPQwen1.5 MLLM · GSM8Kaccuracy: 51.37 -> 52.08Helped
Use Wings visual and textual learners with attention-based routing instead of Qwen + SigLIPQwen1.5 MLLM · MMMU-VALaccuracy: 35.11 -> 39.89Helped
Use Wings visual and textual learners with attention-based routing instead of Qwen + SigLIPQwen1.5 MLLM · MMBenchaccuracy: 70.94 -> 70.53Hurt
Use Wings visual and textual learners with attention-based routing instead of Qwen + SigLIPQwen1.5 MLLM · ScienceQAaccuracy: 74.89 -> 78.76Helped
Use a lower learning rate (ablation)Wings · text-only tasksperformanceHelped
Use a lower learning rate (ablation)Wings · multimodal questionsperformanceHurt
Use a higher learning rate (ablation)Wings · multimodal questionsperformanceHelped
Use a higher learning rate (ablation)Wings · text-only tasksperformanceHurt
Train visual learners alone, without textual learners and routing (ablation)Wings · text-only tasksperformanceHelped

Surprising findings

Limitations

Experimental setup

Datasets: LLaVA-v1.5 training data ALLaVA SynthDog ArXivQA Stanford Alpaca Alpaca GPT-4 LIMA UltraChat WebQA BELLE-0.5M MMLU CMMLU

Benchmarks: MMLU CMMLU ARC-Easy ARC-Challenge WinoGrande OpenbookQA RACE-Middle RACE-High WSC CHID HellaSwag SIQA

Metrics: accuracy forgetting delta vs. initial LLM MME score

Baselines: LLaVA-v1.5 LLaVA-Next DeepSeek-VL Qwen-VL Monkey MiniCPM-V ShareGPT4V InstructBLIP IDEFICS O-Flamingo GPT-4V Gemini Pro Vision

Builds on: LLaVA LoRA Mixture-of-Experts

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