atlas

Research map / Image generation / Autoregressive and latent generation

DRIP: Unleashing Diffusion Priors for Joint Foreground and Alpha Prediction in Image Matting

NeurIPS 2024 · Image generation › Autoregressive and latent generation · approach: Classical generative image synthesis · 7 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.

Drip repurposes a pre-trained Stable Diffusion v2 latent diffusion model to jointly predict foreground color and alpha matte from an image and trimap, adding a switcher, cross-domain attention, and a latent transparency decoder.

Problem

Image matting requires recovering both foreground color and alpha from a single image, but two-stage methods (alpha prediction then post-processing) accumulate errors and yield poor foregrounds, and limited matting data hurts generalization to semi-transparent and rare objects.

Approach

Use the generative data priors of a pre-trained latent diffusion model to jointly denoise foreground and alpha latents in one U-Net, with a domain switcher embedding and cross-domain self-attention for mutual guidance, plus a trainable latent transparency decoder that corrects VAE reconstruction errors.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Drip instead of ViTMatteAlpha prediction · Composition-1kSAD: 21.5 -> 20.8Helped
Use Drip instead of ViTMatteAlpha prediction · Composition-1kMSE: 3.3 -> 3.1Helped
Use Drip instead of AIMAlpha prediction · AIM-500SAD: 43.9 -> 17.3Helped
Use Drip instead of AIMAlpha prediction · AIM-500MSE: 16.1 -> 1.5Helped
Use Drip instead of FBAForeground prediction · Composition-1kSAD: 38.8 -> 34.1Helped
Use Drip instead of FBAForeground prediction · Composition-1kMSE: 6.0 -> 4.3Helped
Add cross-domain attention (ablation)Drip · AIM-500SAD: 17.8 -> 17.3Helped
Add cross-domain attention (ablation)Drip · AIM-500Foreground SAD: 21.3 -> 20.6Helped
Add cross-domain attention (ablation)Drip · AIM-500RGBA SAD: 26.7 -> 23.3Helped
Add the foreground and alpha switcher (ablation)Drip · AIM-500SAD: 18.1 -> 17.3Helped
Add the transparent latent decoder (ablation)Drip · AIM-500SAD: 21.3 -> 17.3Helped
Add the transparent latent decoder (ablation)Drip · AIM-500Foreground SAD: 25.3 -> 20.6Helped

Limitations

Experimental setup

Datasets: Composition-1k COCO VOC2012 LAION-5B

Benchmarks: Composition-1k AIM-500

Metrics: SAD MSE Grad Conn foreground SAD foreground MSE

Baselines: DIM IndexNet FBA HATT AIM GFM MFormer ViTMatte DiffMat Global-Matting KNN-Matting ContextAware-Matting

Builds on: Stable Diffusion ControlNet LayerDiffusion Marigold DiffMatte

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