atlas

Research map / Image generation / Text-to-image diffusion

InstructCV: Instruction-Tuned Text-to-Image Diffusion Models as Vision Generalists

ICLR 2024 · Image generation › Text-to-image diffusion · approach: Compositional text-to-image guidance · 0 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.

Casts computer vision tasks (segmentation, detection, depth, classification) as instruction-guided text-to-image generation by instruction-tuning Stable Diffusion on a pooled multi-task dataset with LLM-rephrased prompts.

Problem

Standard vision recognition tasks rely on task-specific architectures and losses, and prior generalist models use visual or learned prompts that are uninterpretable and generalize poorly to new categories or phrasings.

Approach

Represent each vision task output as an RGB image (masks, drawn boxes, depth-coded images, color blocks) and instruction-tune a pretrained Stable Diffusion conditioned on the input image and a natural language task instruction, with instructions paraphrased by a T5-based LLM to increase diversity.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use fixed prompts instead of rephrased prompts during training (ablation)InstructCV · NYUv2RMSE: 0.297 -> 0.275Helped
Use fixed prompts instead of rephrased prompts during training (ablation)InstructCV · ADE-20KmIoU: 47.235 -> 52.3Helped
Use fixed prompts instead of rephrased prompts during training (ablation)InstructCV · Oxford-Petsaccuracy: 82.135 -> 80.4Hurt
Use rephrased prompts during training (ablation)InstructCV · ADE-20K with unseen user-written instructionsmIoU: 36.315 -> 48.739Helped
Use rephrased prompts during training (ablation)InstructCV · ADE-20K with varied unseen instructionsmIoU: The FP model had a 51.8% drop; the RP model had a 6% reduction.Helped
Use InstructCV-RP for open-vocabulary segmentationInstructCV · FSS-1000mIoU: +3.5 mIoU over Generalist Painter; +7.3 in mIoU over InpaintingHelped
Use InstructCV-FP rather than the strongest task-specific baselineInstructCV · NYUv2RMSE: 10% improvement in RMSEHelped
Evaluate task-specific depth models on external SUNRGB-D dataDepthFormer and BinsFormer · SUNRGB-DRMSE: BinsFormer: ≈ 31.2%; DepthFormer: ≈ 26.0%Hurt
Evaluate InstructCV on external depth dataInstructCV · SUNRGB-DRMSE: performance improved by 6%Helped
Evaluate Pix2SeqV2 on external VOC detection dataPix2SeqV2 · VOCmAP@0.5: 32.2%Hurt
Use InstructCV instead of Pix2SeqV2 on external VOC detection dataInstructCV · VOCmAP@0.5: +23.2 in mAP@0.5Helped
Use InstructCV instead of Unified-IO for semantic segmentationInstructCV · ADE-20KmIoU: +24.401 in mIOUHelped

Surprising findings

Limitations

Experimental setup

Datasets: MS-COCO ADE20K Oxford-IIIT Pets NYUv2 SUNRGB-D VOC ImageNet FSS-1000

Benchmarks: ADE20K COCO NYUv2 Oxford-IIIT Pets SUNRGB-D VOC ImageNet-sub FSS-1000

Metrics: RMSE mIoU Acc mAP@0.5

Baselines: DepthFormer BinsFormer UviM Mask2Former SSA ResNet ViT DETR Mask R-CNN Unified-IO Pix2SeqV2 Painter

Builds on: InstructPix2Pix Stable Diffusion latent diffusion models Unified-IO Painter

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