Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

OSOR: One-Step Diffusion Inpainting for Effect-Aware Object Removal

About

Real-world object removal is challenging due to two key difficulties: the target object's non-local effects, such as shadows and reflections, which are difficult to model, and the fact that user-provided masks are often inaccurate or incomplete. With billions of parameters and tens of denoising steps, diffusion-based models achieve strong removal performance at the expense of substantial computational cost, limiting their use in interactive applications and on edge devices. To address these challenges, we present OSOR (One-Step Object Removal), which simultaneously achieves efficient, effect-aware, and mask-robust object removal. Concretely, OSOR introduces: (1) an occupancy-guided discriminator for precise boundary supervision, enabling stable single-step diffusion training; (2) an alpha head that leverages knowledge from pretrained diffusion models to predict appropriate removal regions with minimal overhead, thereby handling imperfect masks; and (3) a semantic-anchored verification pipeline (SAVP) that filters noisy instruction-based triplets to produce effect-aware supervision at scale. Using SAVP, we curate CORNE, which contains 280K verified removal pairs, and further annotate AnimeEraseBench and TextEraseBench to evaluate performance on more complex removal tasks. Experiments show that OSOR surpasses strong multi-step diffusion baselines in perceptual quality while achieving $4\times$ to $30\times$ faster inference.

Qinming Zhou, Chenxi Sun, Deyang Kong, Junhao He, Xiangheng Tang, Peike Yu, Haotian Wu, Leilei Cao, Linfeng Zhang• 2026

Related benchmarks

TaskDatasetResultRank
Object RemovalRORD (val)
PSNR25.613
36
Object RemovalCORNE (val)
FID12.3241
18
Object RemovalAnimeEraseBench (val)
FID26.2539
18
Object RemovalTextEraseBench
FID19.6169
9
Object RemovalOmniPaint-Bench
FID49.1927
9
Object RemovalRemovalBench
FID43.8899
9
Showing 6 of 6 rows

Other info

Follow for update