HyperDreamBooth: HyperNetworks for Fast Personalization of Text-to-Image Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ruiz, Nataniel, Li, Yuanzhen, Jampani, Varun, Wei, Wei, Hou, Tingbo, Pritch, Yael, Wadhwa, Neal, Rubinstein, Michael, Aberman, Kfir
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917805384794112
author Ruiz, Nataniel
Li, Yuanzhen
Jampani, Varun
Wei, Wei
Hou, Tingbo
Pritch, Yael
Wadhwa, Neal
Rubinstein, Michael
Aberman, Kfir
author_facet Ruiz, Nataniel
Li, Yuanzhen
Jampani, Varun
Wei, Wei
Hou, Tingbo
Pritch, Yael
Wadhwa, Neal
Rubinstein, Michael
Aberman, Kfir
contents Personalization has emerged as a prominent aspect within the field of generative AI, enabling the synthesis of individuals in diverse contexts and styles, while retaining high-fidelity to their identities. However, the process of personalization presents inherent challenges in terms of time and memory requirements. Fine-tuning each personalized model needs considerable GPU time investment, and storing a personalized model per subject can be demanding in terms of storage capacity. To overcome these challenges, we propose HyperDreamBooth - a hypernetwork capable of efficiently generating a small set of personalized weights from a single image of a person. By composing these weights into the diffusion model, coupled with fast finetuning, HyperDreamBooth can generate a person's face in various contexts and styles, with high subject details while also preserving the model's crucial knowledge of diverse styles and semantic modifications. Our method achieves personalization on faces in roughly 20 seconds, 25x faster than DreamBooth and 125x faster than Textual Inversion, using as few as one reference image, with the same quality and style diversity as DreamBooth. Also our method yields a model that is 10,000x smaller than a normal DreamBooth model. Project page: https://hyperdreambooth.github.io
format Preprint
id arxiv_https___arxiv_org_abs_2307_06949
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle HyperDreamBooth: HyperNetworks for Fast Personalization of Text-to-Image Models
Ruiz, Nataniel
Li, Yuanzhen
Jampani, Varun
Wei, Wei
Hou, Tingbo
Pritch, Yael
Wadhwa, Neal
Rubinstein, Michael
Aberman, Kfir
Computer Vision and Pattern Recognition
Artificial Intelligence
Graphics
Machine Learning
Personalization has emerged as a prominent aspect within the field of generative AI, enabling the synthesis of individuals in diverse contexts and styles, while retaining high-fidelity to their identities. However, the process of personalization presents inherent challenges in terms of time and memory requirements. Fine-tuning each personalized model needs considerable GPU time investment, and storing a personalized model per subject can be demanding in terms of storage capacity. To overcome these challenges, we propose HyperDreamBooth - a hypernetwork capable of efficiently generating a small set of personalized weights from a single image of a person. By composing these weights into the diffusion model, coupled with fast finetuning, HyperDreamBooth can generate a person's face in various contexts and styles, with high subject details while also preserving the model's crucial knowledge of diverse styles and semantic modifications. Our method achieves personalization on faces in roughly 20 seconds, 25x faster than DreamBooth and 125x faster than Textual Inversion, using as few as one reference image, with the same quality and style diversity as DreamBooth. Also our method yields a model that is 10,000x smaller than a normal DreamBooth model. Project page: https://hyperdreambooth.github.io
title HyperDreamBooth: HyperNetworks for Fast Personalization of Text-to-Image Models
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Graphics
Machine Learning
url https://arxiv.org/abs/2307.06949