atlas

Research map / Large language models / Language, safety and interpretability

Making LLaMA SEE and Draw with SEED Tokenizer

ICLR 2024 · Large language models › Language, safety and interpretability · approach: LLM unlearning and benchmarks · 8 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.

Introduces SEED, a discrete image tokenizer with 1D causal dependency and high-level semantics, and SEED-LLaMA, an LLM trained with next-token prediction over interleaved image-text tokens for multimodal comprehension and generation.

Problem

Existing visual tokenizers (e.g., VQ-VAE) capture low-level detail and use 2D dependencies incompatible with left-to-right LLM autoregression, so MLLMs struggle to unify comprehension and generation with the original next-word-prediction recipe.

Approach

A VQ tokenizer whose Causal Q-Former produces 32 causally dependent query embeddings trained with contrastive learning for high-level semantics, then quantized and decoded to a unCLIP-SD generation embedding; the discrete codes are added as new vocabulary words to a pretrained LLM trained with unified next-token prediction.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply multimodal instruction tuningSEED-LLaMA, 8B · VizWizVQA accuracy: 21.5 -> 55.1Helped
Apply multimodal instruction tuningSEED-LLaMA, 8B · COCO text-to-imageCLIP similarity: 69.07 -> 70.68Helped
Use the larger LLaMA2-Chat-13B backbone instead of Vicuna-7BSEED-LLaMA-I · SEED-BenchMCQ accuracy: 51.5 -> 53.7Helped
Align generation embeddings with unCLIP-SD image embeddings rather than SD text embeddings (ablation)SEED tokenizer · COCO image-to-image reconstructionCLIP similarity: 68.23 -> 77.35Helped
Use causal visual codes instead of bilateral visual codes (ablation)SEED-LLaMA* · COCO text-to-image generationsuccessful image generations: 2134 -> 4997Helped
Add full fine-tuning after LoRA tuning (ablation)SEED-LLaMA · COCO image captioningCIDEr: 124.5 -> 125.0Helped
Add full fine-tuning after LoRA tuning (ablation)SEED-LLaMA · COCO text-to-imageCLIP similarity: 68.87 -> 69.07Helped
Quantize causal embeddings into discrete causal codes (ablation)SEED tokenizer · Flickr30K image-text retrievalR@m: 93.6 -> 90.8Hurt
Quantize causal embeddings into discrete causal codes (ablation)SEED tokenizer · COCO image-text retrievalR@m: 82.5 -> 78.2Hurt

Surprising findings

Limitations

Experimental setup

Datasets: CC3M Unsplash LAION-COCO MS-COCO MMC4 OBELISC WebVid JourneyDB DiffusionDB LAION-Aesthetics VIST InstructPix2Pix

Benchmarks: COCO Caption VQAv2 OKVQA VizWiz SEED-Bench MSVDQA MSRVTTQA NExTQA COCO text-to-image Flickr30K text-to-image Flickr30K retrieval COCO retrieval

Metrics: CIDEr VQA accuracy MCQ accuracy top-1 accuracy WUPS CLIP score Recall@K

Baselines: Flamingo BLIP-2 InstructBLIP Kosmos-1 Kosmos-2 CM3Leon Emu Emu-I GILL IDEFICS unCLIP SD

Builds on: BLIP-2 unCLIP Stable Diffusion VQ-VAE LoRA

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