Uni-AdaVD: Universal Concept Erasure for Visual Generation via Orthogonal Value Decomposition

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, quick

Summary

Uni-AdaVD is a novel universal inference-time concept erasure framework designed for visual generative models. It addresses the critical issue of undesirable concepts absorbed from uncurated pretraining data, a challenge existing architecture-specific methods often fail to resolve while preserving non-target content and generative priors. Uni-AdaVD operates by treating the value space of multimodal attention as a unified intervention space. It employs encoder-aware target representation construction to localize target semantics across diverse text encoders and integrates orthogonal value decomposition with an adaptive erasing shift to suppress target semantic directions without altering original model weights. Extensive experiments confirm its strong performance on single- and multi-concept erasure across U-Net-, DiT-, and autoregressive image generators, as well as text-to-video models, establishing it as an efficient and adaptable safety mechanism.

Key takeaway

For Machine Learning Engineers deploying visual generative models, Uni-AdaVD provides a crucial safety mechanism. If you are struggling with architecture-specific concept erasure or preserving non-target content, consider integrating Uni-AdaVD. It allows you to universally remove undesirable concepts from models like U-Net, DiT, and text-to-video generators at inference time, without costly model retraining. This approach ensures safer model deployment while maintaining generative quality.

Key insights

Uni-AdaVD offers a universal, inference-time method to erase undesirable concepts from visual generative models without retraining.

Principles

Method

Uni-AdaVD constructs encoder-aware target representations, then applies orthogonal value decomposition with an adaptive erasing shift in the multimodal attention's value space to suppress target semantics.

In practice

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.