OSTAF: A One-Shot Tuning Method for Improved Attribute-Focused T2I Personalization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Ye, Yi, Zili, Ma, Rui
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911800343134208
author Wang, Ye
Yi, Zili
Ma, Rui
author_facet Wang, Ye
Yi, Zili
Ma, Rui
contents Personalized text-to-image (T2I) models not only produce lifelike and varied visuals but also allow users to tailor the images to fit their personal taste. These personalization techniques can grasp the essence of a concept through a collection of images, or adjust a pre-trained text-to-image model with a specific image input for subject-driven or attribute-aware guidance. Yet, accurately capturing the distinct visual attributes of an individual image poses a challenge for these methods. To address this issue, we introduce OSTAF, a novel parameter-efficient one-shot fine-tuning method which only utilizes one reference image for T2I personalization. A novel hypernetwork-powered attribute-focused fine-tuning mechanism is employed to achieve the precise learning of various attribute features (e.g., appearance, shape or drawing style) from the reference image. Comparing to existing image customization methods, our method shows significant superiority in attribute identification and application, as well as achieves a good balance between efficiency and output quality.
format Preprint
id arxiv_https___arxiv_org_abs_2403_11053
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle OSTAF: A One-Shot Tuning Method for Improved Attribute-Focused T2I Personalization
Wang, Ye
Yi, Zili
Ma, Rui
Computer Vision and Pattern Recognition
Personalized text-to-image (T2I) models not only produce lifelike and varied visuals but also allow users to tailor the images to fit their personal taste. These personalization techniques can grasp the essence of a concept through a collection of images, or adjust a pre-trained text-to-image model with a specific image input for subject-driven or attribute-aware guidance. Yet, accurately capturing the distinct visual attributes of an individual image poses a challenge for these methods. To address this issue, we introduce OSTAF, a novel parameter-efficient one-shot fine-tuning method which only utilizes one reference image for T2I personalization. A novel hypernetwork-powered attribute-focused fine-tuning mechanism is employed to achieve the precise learning of various attribute features (e.g., appearance, shape or drawing style) from the reference image. Comparing to existing image customization methods, our method shows significant superiority in attribute identification and application, as well as achieves a good balance between efficiency and output quality.
title OSTAF: A One-Shot Tuning Method for Improved Attribute-Focused T2I Personalization
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.11053