atlas

Research map / Vision-language and multimodal / Multimodal LLMs

Octavius: Mitigating Task Interference in MLLMs via LoRA-MoE

ICLR 2024 · Vision-language and multimodal › Multimodal LLMs · approach: Visual instruction tuning · 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 Octavius, an MLLM whose LLM decoder uses LoRA-MoE with instance-based (question-conditioned) gating over LoRA experts, plus a Object-As-Scene 3D point cloud encoder, to mitigate task interference when training on 2D and 3D instruction data.

Problem

Simultaneously instruction-tuning an MLLM on multiple tasks and modalities (2D detection, captioning, VQA; 3D captioning, VQA, classification) causes task interference ("tug-of-war") that degrades performance under PEFT.

Approach

Replace a single LoRA in each projection of the LLM with a set of LoRA experts routed by a sparse top-k gate computed from the instance's question (instance-based gate rather than token-based gate), so different tasks use different experts; combined with a scene-level 3D encoder that aggregates Point-BERT RoI features via query-based cross-attention.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace single LoRA with sparse top-2 LoRA-MoELAMM · PASCAL VOC, IoU=0.5Recall: 7.61 -> 39.04Helped
Replace single LoRA with sparse top-2 LoRA-MoELAMM · ScienceQAaccuracy: 40.31 -> 46.95Helped
Add LoRA-MoEOctavius point-cloud setup · ScanQACIDEr: 168.21 -> 167.31Hurt
Add LoRA-MoE during joint 2D and 3D trainingOctavius · ScanQACIDEr: 182.00 -> 181.44Hurt
Add LoRA-MoE during joint 2D and 3D trainingOctavius · PASCAL VOC, IoU=0.5Recall: 2.64 -> 34.30Helped
Use a dense weighted sum gate instead of sparse top-2 routing (ablation)LAMM LoRA-MoE · PASCAL VOC, IoU=0.5Recall: 39.04 -> 9.78Hurt
Assign different experts individually to each 2D task (ablation)LAMM LoRA-MoE · ScienceQAaccuracy: 46.95 -> 48.54Helped
Reduce LoRA expert rank from 32 to 16 (ablation)LAMM sparse top-2 LoRA-MoE · PASCAL VOC, IoU=0.5Recall: 39.04 -> 32.81Hurt
Reduce LoRA expert rank from 32 to 8 (ablation)LAMM sparse top-2 LoRA-MoE · PASCAL VOC, IoU=0.5Recall: 39.04 -> 25.44Hurt
Remove positional embedding (ablation)Object-As-Scene · ScanNetaccuracy: 47.40 -> 42.60Hurt
Add image features to language-feature fusion (ablation)Object-As-Scene · ScanNetaccuracy: 47.40 -> 61.60Helped
Increase learnable queries from 16 to 64 (ablation)Object-As-Scene · Scan2CapCIDEr: 35.11 -> 41.45Helped

Surprising findings

Limitations

Experimental setup

Datasets: MS-COCO Bamboo LAMM v2 ScanNet Scan2Inst PASCAL VOC ScienceQA LLaVA instruction data ShapeNet NR3D Flickr30K CIFAR-10

Benchmarks: PASCAL VOC ScienceQA Flickr30K CIFAR-10 CelebA ShapeNet NR3D Scan2Cap ScanQA ScanNet

Metrics: Recall@0.5 Precision@0.5 Acc@1 CIDEr BLEU-1 BLEU-4 METEOR ROUGE-L

Baselines: LAMM LLaVA-LoRA 3D-LLM (Flamingo) Ours w/o MoE

Builds on: LAMM LLaVA LoRA Mixture-of-Experts ULIP Point-BERT 3D-LLM

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