TensorX
返回文献探索

Paper · arXiv 2411.17949

ROICtrl: Boosting Instance Control for Visual Generation

Yuchao Gu, Yipin Zhou, Yunfan Ye, Yixin Nie, Licheng Yu, Pingchuan Ma, Kevin Qinghong Lin, Mike Zheng Shou

87 upvotesNovember 27, 2024arXiv 预印本
AI 摘要

ROICtrl, an adapter for diffusion models using ROI-Align and ROI-Unpool, improves regional instance control in text-based visual generation with lower computational costs.

diffusion modelsregional instance controlbounding boxfree-form captionregions of interestimplicit position encodingexplicit attention masksROI-AlignROI-UnpoolROICtrlControlNetT2I-AdapterIP-AdapterED-LoRAmulti-instance generation

Abstract

Natural language often struggles to accurately associate positional and attribute information with multiple instances, which limits current text-based visual generation models to simpler compositions featuring only a few dominant instances. To address this limitation, this work enhances diffusion models by introducing regional instance control, where each instance is governed by a bounding box paired with a free-form caption. Previous methods in this area typically rely on implicit position encoding or explicit attention masks to separate regions of interest (ROIs), resulting in either inaccurate coordinate injection or large computational overhead. Inspired by ROI-Align in object detection, we introduce a complementary operation called ROI-Unpool. Together, ROI-Align and ROI-Unpool enable explicit, efficient, and accurate ROI manipulation on high-resolution feature maps for visual generation. Building on ROI-Unpool, we propose ROICtrl, an adapter for pretrained diffusion models that enables precise regional instance control. ROICtrl is compatible with community-finetuned diffusion models, as well as with existing spatial-based add-ons (\eg, ControlNet, T2I-Adapter) and embedding-based add-ons (\eg, IP-Adapter, ED-LoRA), extending their applications to multi-instance generation. Experiments show that ROICtrl achieves superior performance in regional instance control while significantly reducing computational costs.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号