Tuning-Free Visual Customization via View Iterative Self-Attention Control

Xiaojie Li, Chenghao Gu, Shuzhao Xie, Yunpeng Bai, Weixiang Zhang +1 more
6/10/2024
cs.CV

No official code found. Best implementation: xjli360/VisCtrl (5 stars, Apache-2.0, last commit 2024-07-01).

Abstract

Fine-Tuning Diffusion Models enable a wide range of personalized generation and editing applications on diverse visual modalities. While Low-Rank Adaptation (LoRA) accelerates the fine-tuning process, it still requires multiple reference images and time-consuming training, which constrains its scalability for large-scale and real-time applications. In this paper, we propose \textit{View Iterative Self-Attention Control (VisCtrl)} to tackle this challenge. Specifically, VisCtrl is a training-free method that injects the appearance and structure of a user-specified subject into another subject in the target image, unlike previous approaches that require fine-tuning the model. Initially, we obtain the initial noise for both the reference and target images through DDIM inversion. Then, during the denoising phase, features from the reference image are injected into the target image via the self-attention mechanism. Notably, by iteratively performing this feature injection process, we ensure that the reference image features are gradually integrated into the target image. This approach results in consistent and harmonious editing with only one reference image in a few denoising steps. Moreover, benefiting from our plug-and-play architecture design and the proposed Feature Gradual Sampling strategy for multi-view editing, our method can be easily extended to edit in complex visual domains. Extensive experiments show the efficacy of VisCtrl across a spectrum of tasks, including personalized editing of images, videos, and 3D scenes.

View on arXivView PDF

Code Implementations(1)

xjli360/VisCtrlCommunity100%
Apache-2.0

VisCtrl:Tuning-Free Visual Customization via View Iterative Self-Attention Control

50PythonJun 27, 20242 years agoApache-2.0

Similar Papers

Cite this paper

@article{li2024tuningfree,
  title  = {Tuning-Free Visual Customization via View Iterative Self-Attention Control},
  author = {Xiaojie Li and Chenghao Gu and Shuzhao Xie and Yunpeng Bai and Weixiang Zhang and Zhi Wang},
  year   = {2024},
  eprint = {2406.06258},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2406.06258v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.06258.svg)](https://codeofpaper.com/paper/2406.06258)

Discussion