Patentable/Patents/US-20260017846-A1
US-20260017846-A1

Method and System for Multi-Subject Personalization with Selective U-Net Influence

PublishedJanuary 15, 2026
Assigneenot available in USPTO data we have
Technical Abstract

Recently, an uptick in the interest in providing interpretability to foundational models has been observed. However, the significant effort is limited to the large language models. Diffusion models have proven significant for the generative AI landscape and the interpretability of these models The present disclosure presents a novel selective U-Net influence (SelUT) technique for greater control and interpretability of DreamBooth for multi-subject personalization task. Specifically, the influence of the trained U-Net block(s) is controlled in the present disclosure during model inference. It provides a greater handle on interpreting the contribution of individual U-Net blocks across quality aspects such as identity disentanglement, image aesthetic, human preference, etc. Furthermore, we present an ensemble selection strategy to incorporate the dynamicity between base DreamBooth and the models trained with selective influence of U-Net block(s) which significantly improve the capability of DreamBooth for multi subject personalization.

Patent Claims

Legal claims defining the scope of protection, as filed with the USPTO.

1

receiving, via one or more hardware processors, an input comprising a plurality of subject images, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects; finetuning, via the one or more hardware processors, a trained diffusion model based on the received input using a fine-tuning technique; generating, via the one or more hardware processors, an updated U-Net state dictionary by selecting a plurality of U-Net blocks with associated weights using a finetuned diffusion model, wherein the plurality of U-Net blocks comprises a plurality of down blocks, a plurality of mid blocks and a plurality of up blocks; generating, via the one or more hardware processors, a plurality of influenced blocks from among of a plurality of blocks associated with the trained diffusion model by updating the associated weights based on the plurality of U-Net blocks, wherein the plurality of blocks are selected from a state dictionary associated with the trained diffusion model; generating, via the one or more hardware processors, a plurality of images based on an influenced plurality of influenced blocks using the pretrained diffusion model; and identifying, via the one or more hardware processors, an optimal image from the plurality of generated images using an ensemble-based image selection technique. . A processor-implemented method, the method comprising:

2

claim 1 . The method of, wherein the ensemble based selection technique identifies an optimal image based on a personalization score, wherein the personalization score is average of an Image-Reward score and a Multi-Subject Fidelity, and wherein the image having the maximum Personalization score is identified as the optimal image.

3

claim 1 . The method of, the finetuning of the trained diffusion model is performed using DreamBooth technique.

4

claim 1 receiving a training data comprising a plurality of subject image instances comprising a plurality of subjects in a white background, a plurality of subject pairs pertaining to each of the plurality of plurality of subjects, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects, wherein the plurality of subjects is one of a) a plurality of similar subjects and b) a plurality of dissimilar subjects; and training the diffusion model based on the plurality of subject image instances, wherein the diffusion model is trained using DreamBooth training technique. . The method of, wherein the diffusion model is trained by:

5

at least one memory storing programmed instructions; one or more Input/Output (I/O) interfaces; and one or more hardware processors operatively coupled to the at least one memory, wherein the one or more hardware processors are configured by the programmed instructions to: receive an input comprising a plurality of subject images, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects; finetune a trained diffusion model based on the received input using a fine-tuning technique; generate an updated U-Net state dictionary by selecting a plurality of U-Net blocks with associated weights using a finetuned diffusion model, wherein the plurality of U-Net blocks comprises a plurality of down blocks, a plurality of mid blocks and a plurality of up blocks; generate a plurality of influenced blocks from among of a plurality of blocks associated with the trained diffusion model by updating the associated weights based on the plurality of U-Net blocks, wherein the plurality of blocks are selected from a state dictionary associated with the trained diffusion model; generate a plurality of images based on an influenced plurality of influenced blocks using the pretrained diffusion model; and identify an optimal image from the plurality of generated images using an ensemble-based image selection technique. . A system comprising:

6

claim 5 . The system of, wherein the ensemble based selection technique identifies an optimal image based on a personalization score, wherein the personalization score is average of an Image-Reward score and a Multi-Subject Fidelity, and wherein the image having the maximum Personalization score is identified as the optimal image.

7

claim 5 . The system of, the finetuning of the trained diffusion model is performed using DreamBooth technique.

8

claim 5 receiving a training data comprising a plurality of subject image instances comprising a plurality of subjects in a white background, a plurality of subject pairs pertaining to each of the plurality of plurality of subjects, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects, wherein the plurality of subjects is one of a) a plurality of similar subjects and b) a plurality of dissimilar subjects; and training the diffusion model based on the plurality of subject image instances, wherein the diffusion model is trained using DreamBooth training technique. . The system of, wherein the diffusion model is trained by:

9

receiving, an input comprising a plurality of subject images, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects; finetuning, via the one or more hardware processors, a trained diffusion model based on the received input using a fine-tuning technique; generating, via the one or more hardware processors, an updated U-Net state dictionary by selecting a plurality of U-Net blocks with associated weights using a finetuned diffusion model, wherein the plurality of U-Net blocks comprises a plurality of down blocks, a plurality of mid blocks and a plurality of up blocks; generating, via the one or more hardware processors, a plurality of influenced blocks from among of a plurality of blocks associated with the trained diffusion model by updating the associated weights based on the plurality of U-Net blocks, wherein the plurality of blocks are selected from a state dictionary associated with the trained diffusion model; generating, via the one or more hardware processors, a plurality of images based on an influenced plurality of influenced blocks using the pretrained diffusion model; and identifying, via the one or more hardware processors, an optimal image from the plurality of generated images using an ensemble-based image selection technique. . One or more non-transitory machine-readable information storage mediums comprising one or more instructions which when executed by one or more hardware processors cause:

10

claim 9 . The one or more non-transitory machine-readable information storage mediums of, wherein the ensemble based selection technique identifies an optimal image based on a personalization score, wherein the personalization score is average of an Image-Reward score and a Multi-Subject Fidelity, and wherein the image having the maximum Personalization score is identified as the optimal image.

11

claim 9 . The one or more non-transitory machine-readable information storage mediums of, the finetuning of the trained diffusion model is performed using DreamBooth technique.

12

claim 9 receiving a training data comprising a plurality of subject image instances comprising a plurality of subjects in a white background, a plurality of subject pairs pertaining to each of the plurality of plurality of subjects, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects, wherein the plurality of subjects is one of a) a plurality of similar subjects and b) a plurality of dissimilar subjects; and training the diffusion model based on the plurality of subject image instances, wherein the diffusion model is trained using DreamBooth training technique. . The one or more non-transitory machine-readable information storage mediums of, wherein the diffusion model is trained by:

Detailed Description

Complete technical specification and implementation details from the patent document.

This U.S. patent application claims priority under 35 U.S.C. § 119 to: Indian Patent Application number 202421053533, filed on 12 Jul. 2024. The entire contents of the aforementioned application are incorporated herein by reference.

The disclosure herein generally relates to the field of image diffusion and, more particularly, to a method and system for multi-subject personalization with selective U-Net influence.

Recently, an uptick in the interest in providing interpretability to foundational models has been observed. However, significant effort is limited to the large language models (LLMs). Diffusion models have proven significant for the generative Artificial Intelligence (AI) landscape and the interpretability of these models would provide stronger foundation to build powerful and scalable applications. Further, diffusion models form the backbone of many generative AI architectures such as text-to-image generation, speech synthesis, etc. However, the interpretability of these models is limited. Recent advancement in text-to-image generation has inspired many downstream and fundamental applications such as image personalization and image editing.

DreamBooth is one of the pioneer works in text-to-image personalization facilitating recontextualization of a given subject by training with just 3-5 images of that subject. Though it provides an efficient mechanism for single-subject personalization, it fails to achieve similar performance for the multi-subject personalization task. It struggles to generate high-quality output with proper subject disentanglement while showing high fidelity to the text prompt. Though there are multiple follow-up works to extend the capability of DreamBooth, there is very little knowledge about the interpretability of DreamBooth. Further, it fails to preserve the identity of all the subjects in the generated image. Distortion, inappropriate mixing of identity, and repetition of subjects is a common phenomenon observed in the generated images.

Embodiments of the present disclosure present technological improvements as solutions to one or more of the above-mentioned technical problems recognized by the inventors in conventional systems. For example, in one embodiment, a method for multi-subject personalization with selective U-Net influence is provided. The method includes receiving, via one or more hardware processors, an input comprising a plurality of subject images, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects. Further, the method includes finetuning via the one or more hardware processors, a trained diffusion model based on the received input using a fine-tuning technique. Furthermore, the method includes generate via the one or more hardware processors, an updated U-Net state dictionary by selecting a plurality of U-Net blocks with associated weights using a finetuned diffusion model, wherein the plurality of U-Net blocks comprises a plurality of down blocks, a plurality of mid blocks and a plurality of up blocks. Furthermore, the method includes generating via the one or more hardware processors, a plurality of influenced blocks from among of a plurality of blocks associated with the trained diffusion model by updating the associated weights based on the plurality of U-Net blocks, wherein the plurality of blocks are selected from a state dictionary associated with the trained diffusion model. Furthermore, the method includes generating via the one or more hardware processors, a plurality of images based on an influenced plurality of influenced blocks using the pretrained diffusion model. Finally, the method includes identifying via the one or more hardware processors, an optimal image from the plurality of generated images using an ensemble-based image selection technique.

In another aspect, a system for Multi-subject personalization with selective U-Net influence is provided. The system includes at least one memory storing programmed instructions, one or more Input/Output (I/O) interfaces, and one or more hardware processors operatively coupled to the at least one memory, wherein the one or more hardware processors are configured by the programmed instructions to receive an input comprising a plurality of subject images, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects. Further, the one or more hardware processors are configured by the programmed instructions to finetune a trained diffusion model based on the received input using a fine-tuning technique. Furthermore, the one or more hardware processors are configured by the programmed instructions to generate an updated U-Net state dictionary by selecting a plurality of U-Net blocks with associated weights using a finetuned diffusion model, wherein the plurality of U-Net blocks comprises a plurality of down blocks, a plurality of mid blocks and a plurality of up blocks. Furthermore, the one or more hardware processors are configured by the programmed instructions to generate a plurality of influenced blocks from among of a plurality of blocks associated with the trained diffusion model by updating the associated weights based on the plurality of U-Net blocks, wherein the plurality of blocks are selected from a state dictionary associated with the trained diffusion model. Furthermore, the one or more hardware processors are configured by the programmed instructions to generate a plurality of images based on an influenced plurality of influenced blocks using the pretrained diffusion model. Finally, the one or more hardware processors are configured by the programmed instructions to identify an optimal image from the plurality of generated images using an ensemble-based image selection technique.

In yet another aspect, a computer program product including a non-transitory computer-readable medium embodied therein a computer program for Multi-subject personalization with selective U-Net influence is provided. The computer readable program, when executed on a computing device, causes the computing device to receive an input comprising a plurality of subject images, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects. Further, the computer readable program, when executed on a computing device, causes the computing device to finetune a trained diffusion model based on the received input using a fine-tuning technique. Furthermore, the computer readable program, when executed on a computing device, causes the computing device to generate an updated U-Net state dictionary by selecting a plurality of U-Net blocks with associated weights using a finetuned diffusion model, wherein the plurality of U-Net blocks comprises a plurality of down blocks, a plurality of mid blocks and a plurality of up blocks. Furthermore, the computer readable program, when executed on a computing device, causes the computing device to generate a plurality of influenced blocks from among of a plurality of blocks associated with the trained diffusion model by updating the associated weights based on the plurality of U-Net blocks, wherein the plurality of blocks are selected from a state dictionary associated with the trained diffusion model. Furthermore, the computer readable program, when executed on a computing device, causes the computing device to generate a plurality of images based on an influenced plurality of influenced blocks using the pretrained diffusion model. Finally, the computer readable program, when executed on a computing device, causes the computing device to identify an optimal image from the plurality of generated images using an ensemble-based image selection technique.

It is to be understood that both the foregoing general description and the following detailed description are exemplary and explanatory only and are not restrictive of the invention, as claimed.

Exemplary embodiments are described with reference to the accompanying drawings. In the figures, the left-most digit(s) of a reference number identifies the figure in which the reference number first appears. Wherever convenient, the same reference numbers are used throughout the drawings to refer to the same or like parts. While examples and features of disclosed principles are described herein, modifications, adaptations, and other implementations are possible without departing from the spirit and scope of the disclosed embodiments.

DreamBooth is an extremely powerful technique for single subject text-to-image personalization using diffusion models. However, various recent works have shown the limitations of DreamBooth on the multi-subject personalization (MSP) task. Specifically, it struggles to disentangle the multiple subjects by correctly associating these subjects to the corresponding unique identifier in the text prompt. It results in poor quality image generation with low multi-subject fidelity, image aesthetic and human preference. But it remains unclear as to why DreamBooth, which is a widely popular technique for single-subject personalization fails to scale for the multiple subjects. In such a scenario, the interpretability of Dream-Booth becomes crucial.

To overcome the challenges of the conventional approaches, embodiments herein provide a method and system for multi-subject personalization with selective U-Net influence. The present disclosure presents a selective U-Net influence (SelUT) technique for greater control and interpretability of Dreambooth for multi-subject personalization task. Specifically, the influence of the trained U-Net block(s) are controlled in the present disclosure during model inference. It provides a greater handle on interpreting the contribution of individual U-Net blocks across quality aspects such as identity disentanglement, image aesthetic, human preference, etc. Furthermore, an ensemble selection strategy is provided, which incorporates the dynamicity between base DreamBooth and the models trained with selective influence of U-Net block(s) which significantly improves the capability of DreamBooth for MSP.

1 FIG.A 3 FIG.C Referring now to the drawings, more particularly tothrough, where similar reference characters denote corresponding features consistently throughout the figures, there are shown preferred embodiments, and these embodiments are described in the context of the following exemplary system and/or method.

1 FIG.A 100 100 102 104 112 102 104 112 108 102 is a functional block diagram of systemfor Multi-subject personalization with selective U-Net influence, in accordance with some embodiments of the present disclosure. The systemincludes or is otherwise in communication with hardware processors, at least one memory such as a memory, an Input/Output (I/O) interface. The hardware processors, memory, and the I/O interfacemay be coupled by a system bus such as a system busor a similar mechanism. In an embodiment, the hardware processorscan be one or more hardware processors.

112 112 112 100 The I/O interfacemay include a variety of software and hardware interfaces, for example, a web interface, a graphical user interface, and the like. The I/O interfacemay include a variety of software and hardware interfaces, for example, interfaces for peripheral device(s), such as a keyboard, a mouse, an external memory, a printer and the like. Further, the I/O interfacemay enable systemto communicate with other devices, such as web servers, and external databases.

112 112 112 The I/O interfacecan facilitate multiple communications within a wide variety of networks and protocol types, including wired networks, for example, local area network (LAN), cable, etc., and wireless networks, such as Wireless LAN (WLAN), cellular, or satellite. For the purpose, the I/O interfacemay include one or more ports for connecting several computing systems with one another or to another server computer. The I/O interfacemay include one or more ports for connecting several devices to one another or to another server.

102 102 104 The one or more hardware processorsmay be implemented as one or more microprocessors, microcomputers, microcontrollers, digital signal processors, central processing units, node machines, logic circuitries, and/or any devices that manipulate signals based on operational instructions. Among other capabilities, the one or more hardware processorsis configured to fetch and execute computer-readable instructions stored in memory.

104 104 106 104 110 106 The memorymay include any computer-readable medium known in the art including, for example, volatile memory, such as static random access memory (SRAM) and dynamic random access memory (DRAM), and/or non-volatile memory, such as read only memory (ROM), erasable programmable ROM, flash memories, hard disks, optical disks, and magnetic tapes. In an embodiment, memoryincludes a plurality of modules. Memoryalso includes a data repository (or repository)for storing data processed, received, and generated by the plurality of modules.

106 100 106 106 106 102 106 106 100 The plurality of modulesincludes programs or coded instructions that supplement applications or functions performed by the systemfor Multi-subject personalization with selective U-Net influence. The plurality of modules, amongst other things, can include routines, programs, objects, components, and data structures, which perform particular tasks or implement particular abstract data types. The plurality of modulesmay also be used as, signal processor(s), node machine(s), logic circuitries, and/or any other device or component that manipulates signals based on operational instructions. Further, the plurality of modulescan be used by hardware, by computer-readable instructions executed by the one or more hardware processors, or by a combination thereof. The plurality of modulescan include various sub-modules (not shown). The plurality of modulesmay include computer-readable instructions that supplement applications or functions performed by the systemfor Multi-subject personalization with selective U-Net influence.

110 106 The data repository (or repository)may include a plurality of abstracted pieces of code for refinement and data that is processed, received, or generated as a result of the execution of the plurality of modules in the module(s).

110 100 110 100 110 110 100 1 FIG.A Although the data repositoryis shown internal to the system, it will be noted that, in alternate embodiments, the data repositorycan also be implemented external to the system, where the data repositorymay be stored within a database (repository) communicatively coupled to the system. The data contained within such an external database may be periodically updated. For example, new data may be added into the database (not shown in) and/or existing data may be modified and/or non-useful data may be deleted from the database. In one example, the data may be stored in an external system, such as a Lightweight Directory Access Protocol (LDAP) directory, a Relational Database Management System (RDBMS) and a Timeseries Database.

1 FIG.A 1 FIG.B 1 FIG.B The overall architecture of the system ofis explained in conjunction with. Now referring to, initially, a diffusion model like Stable Diffusion Extra Large (SDXL) model is trained using a plurality of objects and corresponding prompts and a U-Net state dictionary is obtained. The U-Net state dictionary includes a plurality of down blocks, a plurality of mid blocks and a plurality of up blocks. During inference stage, an iterative U-Net block selection is performed and the weights associated with the pre-trained Stable Diffusion Extra Large (SDXL) are updated with the weights associated with the plurality of U-Net blocks of the obtained U-Net state dictionary. Further, images are generated for the received input prompt and an optimal image is selected using an ensemble based selection approach.

100 2 FIG. The working of the components of systemare explained with reference to the method steps depicted in.

2 FIG. 1 1 FIGS.A andB 1 1 FIGS.A andB 2 FIG. 200 100 104 102 200 102 200 100 200 is an exemplary flow diagram illustrating a methodfor multi-subject personalization with selective U-Net influence implemented by the system of, according to some embodiments of the present disclosure. In an embodiment, the systemincludes one or more data storage devices or the memoryoperatively coupled to the one or more hardware processor(s)and is configured to store instructions for execution of steps of the methodby the one or more hardware processors. The steps of methodof the present disclosure will now be explained with reference to the components or blocks of systemas depicted inand the steps of flow diagram as depicted in. The methodmay be described in the general context of computer executable instructions. Generally, computer executable instructions can include routines, programs, objects, components, data structures, procedures, modules, functions, etc., that perform particular functions or implement particular abstract data types.

200 200 200 200 The methodmay also be practiced in a distributed computing environment where functions are performed by remote processing devices that are linked through a communication network. The order in which the methodis described is not intended to be construed as a limitation, and any number of the described method blocks can be combined in any order to implement the method, or an alternative method. Furthermore, the methodcan be implemented in any suitable hardware, software, firmware, or combination thereof.

2 FIG. 202 200 102 Now referring to, at stepof method, the one or more hardware processorsare configured by the programmed instructions to receive an input comprising a plurality of subject images, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects.

The text prompt is in the format “[V1*]<subject1> and [V2*]<subject2>”. For example, “[V1*] bald man and [V2*] girl dancing together”, “[V1*] cat watching television and [V2*] Ethan talking on phone”, “[V1*] old man and [V2*] cat dancing together” and the like.

204 200 102 At stepof the method, the one or more hardware processorsis configured by the programmed instructions to finetune a trained diffusion model, for example SDXL model, based on the received data using DreamBooth fine-tuning technique.

For example, the diffusion model is trained using training data. The training data includes a plurality of subject image instances, a plurality of subject pairs pertaining to each of the plurality of subjects, a text description pertaining to each of the plurality of subject images and an associated text prompt explaining action to be performed by each of the plurality of subjects. The plurality of image instances includes a plurality of subjects in a white background. The plurality of subjects is one of a) a plurality of similar subjects and b) a plurality of dissimilar subjects. Further, the diffusion model is trained based on the plurality of subject image instances, wherein the diffusion model is trained using DreamBooth training technique.

206 200 102 1 FIG.B At stepof the method, the one or more hardware processorsis configured by the programmed instructions to generate an updated U-Net state dictionary by selecting a plurality of U-Net blocks with a corresponding updated weights using a finetuned diffusion model, wherein the plurality of U-Net blocks comprises a plurality of down blocks, a plurality of mid blocks and a plurality of up blocks as shown in.

208 200 102 At stepof the methodthe one or more hardware processorsis configured by the programmed instructions to generate a plurality of influenced blocks from among of a plurality of blocks associated with the trained diffusion model by updating the associated weights based on the plurality of U-Net blocks, wherein the plurality of blocks are selected from a state dictionary associated with the trained diffusion model.

For example, it has been observed that different blocks of the U-Net contribute to different aspects of image generation such as layout, color, and style. As discussed earlier, the interpretability of Dream-Booth remains unclear. To this extent, the present disclosure iteratively influence the selected blocks of a pre-trained SDXL model M with the DreamBooth-based trained U-Net. Formally,

where, UM (b) and USD (b) represent the weights of the block b in the U-Net of the pre-trained SDXL model M and the state dictionary of the DreamBooth-based trained U-Net respectively. The operation ⊕signifies updating the b block weights of UM while freezing the other blocks. The block set B={D1,D2,D,M,U0,U1,U}. The blocks D1 and D2 (collectively denoted as D) are the down-blocks and U0 and U1 (collectively denoted as U) are the U-Net up-blocks. Block M is the middle block of the U-Net.

210 200 102 At stepof the method, the one or more hardware processorsis configured by the programmed instructions to generate a plurality of images based on an influenced plurality of blocks using the pretrained diffusion model.

212 200 126 102 1 FIG.B At stepof the method, an output orchestrator module(shown in) when executed by the one or more hardware processorsis configured by the programmed instructions to identify an optimal image from the plurality of generated images using an ensemble-based image selection technique

For example, the ensemble based selection technique identifies an optimal image based on a personalization score, wherein the personalization score is average of an Image-Reward score and a Multi-Subject Fidelity and, wherein the image having the maximum Personalization score is identified as the optimal image.

The personalization score for the generated image with model Mb is defined as given in equation (2).

b where, N(IR(M′)) shows the normalized IR score and avg( ) represents the average of both the metric scores. The present disclosure normalizes the IR score between 0 and 1. For a prompt P, the ensemble selection picks the generated image (from one of the eight models) with the best personalization score.

Image-Reward score: This metric rates generated image based on the human preference across parameters such as text-image alignment, human aesthetic, toxicity, and bias and assign scores to them. A higher score signifies that the generated image is highly preferred by humans as compared to the other images generated using the same text prompt

Multi-Subject Fidelity: This metric involves detecting the subjects in the generated image and comparing them against the corresponding subjects in the input images used during training phase. The metric score ranges from 0 to 1, where a higher score signifies high multi-subject fidelity as well as no character mixing.

11 Dataset: The present disclosure was trained using Subject-Datasetwhich consists of 11 diverse characters in different pose. The dataset includes 6 comic characters and 5 real human characters. Out of these 11 characters, a multi-subject dataset of 25 character pairs was prepared by placing two characters on a white background. The character pairing was performed such that the real characters are paired with the real characters and vice-versa.

Training: The SDXL model was trained and DreamBooth Low-Rank Adaptation of Large Language models (LoRA) fine-tuning was performed. 5 instance images were selected randomly for each of the 25 subject pairs. For all the 5 training images of a subject pair, a common training prompt is used: [V1*]<subject1> and [V2*]<subject2>, simple background.

Inference: For evaluation of various aspects of multisubject personalization, 4 prompt categories were considered: (i) Same action prompts (SAP): In this category of prompt, both the subjects are performing the same action (ii) Different action prompts (DAP): In this category of prompt, both the subjects are performing the different actions (iii) Subject 1 prompts (S1P): In this category of prompt, only subject 1 is performing an action and there is no mention of subject 2 in the prompt (iv) Subject2 prompts (S2P): In this category of prompt, only subject 2 is performing an action and there is no mention of subject 1 in the prompt.

Evaluation: Three evaluation metrics were used: multi-subject fidelity, image reward, and personalization score (refer equation 2). To evaluate multi-subject fidelity, Detect and Compare technique was used, which involves detecting the subjects in the generated image and comparing them against the corresponding subjects in the instance images. The metric score ranges from 0 to 1, where a higher score signifies high multi-subject fidelity as well as no character mixing. Furthermore, we use Image-Reward function which rates generated image based on the human preference across parameters such as text-image alignment, human aesthetic, toxicity, and bias. A higher score signifies that the generated image is highly preferred by humans as compared to the other images generated using the same text prompt. We further combine both these metrics and compute the personalization score (refer equation 2) that provides a comprehensive idea about the image generation quality across a number of quality parameters.

3 3 FIGS.A throughC 3 302 302 FIG.A,A andB 3 304 304 FIG.B,A andB 3 306 306 FIG.C,A andB 302 304 306 The performance evaluation with these metrics is in Table I. Now referring to Table I, ES stands for Ensemble Selection. Also, in Table II, the percentage of prompts achieving the best metric score is presented. The best metric score is identified for a given prompt by comparing the scores across all the models. The quality results are shown in. Now referring toare the multi-subject instances (cat and dog) in white background andC is an output for the prompt “[V1*] cat painting and [V2*] dog watching television”. Now referring toare the multi-subject instances (old man and dog) in white background andC is the output for the prompt “[V1*] old man and [V2*] dog sitting on camel back”. Now referring toare the multi-subject instances (bald man and middle aged woman) in white background andC is the output for the prompt “[V1*] bald man and [V2*] middle aged woman jumping in a playground”.

TABLE I Multisubject fidelity Image Reward Personalization score SAP DAP SIP SAP DAP SIP SAP DAP SIP S2P S2P S2P Base 16.78 18.97 67.57 60.66 − − 0.444 0.176 35.38 29.88 64.44 58.98 U 15.88 17.03 11.74 11.74 − − 0.924 0.516 39.21 32.17 37.86 34.59 U0 17.11 16.45 60.58 56.75 − − 0.918 0.591 40.23 31.94 62.5 56.99 U1 11.81 12.55 55.57 54.01 0.116 0.115 0.957 0.689 36.83 32.26 61.73 57.01 M 11.34 12.46 56.36 56.19 0.141 0.14 1.04 0.801 38.1 32.66 62.37 59.21 D1 12.58 13.18 55.62 54.29 0.023 0.023 0.984 0.699 38.4 31.4 60.46 57.58 D2 12.58 14.01 57.68 55.14 −0.016 −0.016 0.95 0.701 39.36 31.38 60.31 57.84 D 13.31 13.53 11.74 11.74 −0.023 −0.022 0.882 0.69 39.28 30.95 40.28 36.73 ES 26.3 27.3 72.5 63.4 1.36 1.15 1.56 1.42 53.9 50.3 76.7 71.7 indicates data missing or illegible when filed

TABLE II Multisubject fidelity Image Reward Personalization score SAP DAP SIP SAP DAP SIP SAP DAP SIP S2P S2P S2P Base 25 20.5 58.2 46.8 0 0 0 0 14.9 13.3 27.1 29.1 U 20.26 16.53 5.14 0 0 0 0 0 15.2 13.6 2 0 U0 25.06 34.13 17.71 16.85 16.53 12.53 19.71 14.28 18.13 12.26 17.42 12.85 U1 9.06 10.1 5.14 13.7 16 17.8 17.1 16 8.53 13 14.2 13.4 M 6.13 8 6.85 9.71 21.33 20 14.57 18.57 9.86 12.53 12.57 16.85 D1 8.8 10.6 6.85 12.8 16 18.4 16 10.5 9.86 14.4 10.5 11.9 D2 0 0 0 0 17.8 14.5 17.1 21.7 11.7 12.2 11.9 15.7 D 0 0 0 0 12.26 15.73 15.42 18.85 11.73 8.53 4 0 indicates data missing or illegible when filed Several Interesting Observations were Made During Experimentation:

Block U0 tends to capture the subject identity: It was observed that the model with block U0 shows competitive performance to the base DreamBooth model for the multi-subject fidelity metric (refer Table I). It also shows similar behavior with the percentage of prompts achieving the best metric score (refer Table 3). It suggests that the subject identity is largely controlled through the U0 block in the U-Net.

Block M tends to control the image aesthetic: As shown in Table I, the model with block M achieves the second best image reward score (after the ensemble selection) on all the prompt categories. It further denotes that the middle block in the U-Net tends to control the aesthetic in the generated image producing high-quality fine-grained details.

Blocks U and D poorly aligns the unique identifier in the text prompt with the subject's visual concepts: As shown in Table I, the models with U and D blocks achieve significantly lower multi-subject fidelity scores on S1P and S2P prompt categories. They also consistently fail to generate the top-ranked images for these two prompt categories (see Table II). It signifies that these models fails to align the subject's unique identifier in the text prompt ([V1*] or [V2*]) with the visual concepts of these subjects in the image.

DreamBooth with selective U-Net influence generates images that are likely to be more preferred by humans: It was observed that the image reward score for the base DreamBooth model is extremely low as compared to the other models. Also, as observed in Table II, the base DreamBooth model consistently fails to generate the top-ranked images on the image reward metric for all the prompt categories signifying the lower aesthetic quality of the model.

DreamBooth is inherently a single-subject personalization technique. It was observed that all the three metric scores for the base DreamBooth are significantly higher for the prompt categories with only the single subject mentioned as compared to the prompt categories with the mention of both the subjects. It further reinforces our observation that DreamBooth inherently is a single-subject personalization technique that requires a systematic investigation to extend its capability for the multi-subject personalization task.

Selective U-Net influence and ensemble selection significantly boost the DreamBooth's personalization capability: In Table I, we consistently observe that the ensemble selection outperforms the individual models by a large margin for all the prompt categories and on all the three metrics. Ensemble selection of models has been largely studied and successfully applied across numerous machine learning tasks such as classification, regression, time-series forecasting, etc. It also help here in the multisubject personalization task to boost the performance of DreamBooth with greater control and interpretability

The written description describes the subject matter herein to enable any person skilled in the art to make and use the embodiments. The scope of the subject matter embodiments is defined by the claims and may include other modifications that occur to those skilled in the art. Such other modifications are intended to be within the scope of the claims if they have similar elements that do not differ from the literal language of the claims or if they include equivalent elements with insubstantial differences from the literal language of the claims.

The embodiments of the present disclosure herein address the unresolved problem of multisubject personalization. The present disclosure provides a selective U-Net influence and ensemble selection based approach which provides greater interpretability and boost the capability of DreamBooth for multi-subject personalization.

It is to be understood that the scope of the protection is extended to such a program and in addition to a computer-readable means having a message therein such computer-readable storage means contain program-code means for implementation of one or more steps of the method when the program runs on a server or mobile device or any suitable programmable device. The hardware device can be any kind of device which can be programmed including e.g., any kind of computer like a server or a personal computer, or the like, or any combination thereof. The device may also include means which could be e.g., hardware means like e.g., an application-specific integrated circuit (ASIC), a field-programmable gate array (FPGA), or a combination of hardware and software means, e.g. an ASIC and an FPGA, or at least one microprocessor and at least one memory with software modules located therein. Thus, the means can include both hardware means, and software means. The method embodiments described herein could be implemented in hardware and software. The device may also include software means. Alternatively, the embodiments may be implemented on different hardware devices, e.g., using a plurality of CPUs, GPUs and edge computing devices.

The embodiments herein can comprise hardware and software elements. The embodiments that are implemented in software include but are not limited to, firmware, resident software, microcode, etc. The functions performed by various modules described herein may be implemented in other modules or combinations of other modules. For the purposes of this description, a computer-usable or computer readable medium can be any apparatus that can comprise, store, communicate, propagate, or transport the program for use by or in connection with the instruction execution system, apparatus, or device. The illustrated steps are set out to explain the exemplary embodiments shown, and it should be anticipated that ongoing technological development will change the manner in which particular functions are performed. These examples are presented herein for purposes of illustration, and not limitation. Further, the boundaries of the functional building blocks have been arbitrarily defined herein for the convenience of the description. Alternative boundaries can be defined so long as the specified functions and relationships thereof are appropriately performed. Alternatives (including equivalents, extensions, variations, deviations, etc., of those described herein) will be apparent to persons skilled in the relevant art(s) based on the teachings contained herein. Such alternatives fall within the scope and spirit of the disclosed embodiments. Also, the words “comprising,” “having,” “containing,” and “including,” and other similar forms are intended to be equivalent in meaning and be open ended in that an item or items following any one of these words is not meant to be an exhaustive listing of such item or items or meant to be limited to only the listed item or items. It must also be noted that as used herein and in the appended claims, the singular forms “a,” “an,” and “the” include plural references unless the context clearly dictates otherwise. Furthermore, one or more computer-readable storage media may be utilized in implementing embodiments consistent with the present disclosure. A computer-readable storage medium refers to any type of physical memory on which information or data readable by a processor may be stored. Thus, a computer-readable storage medium may store instructions for execution by one or more processors, including instructions for causing the processor(s) to perform steps or stages consistent with the embodiments described herein. The term “computer-readable medium” should be understood to include tangible items and exclude carrier waves and transient signals, i.e. non-transitory. Examples include random access memory (RAM), read-only memory (ROM), volatile memory, nonvolatile memory, hard drives, CD ROMs, DVDs, flash drives, disks, and any other known physical storage media.

It is intended that the disclosure and examples be considered as exemplary only, with a true scope of disclosed embodiments being indicated by the following claims.

Classification Codes (CPC)

Cooperative Patent Classification codes for this invention. Click any code to explore related patents in that topic.

Patent Metadata

Filing Date

July 10, 2025

Publication Date

January 15, 2026

Inventors

Ashutosh RANJAN
Vivek SRIVASTAVA
Shirish Subhash KARANDE

Want to explore more patents?

Browse 5M+ US patents with plain-English claim translations and AI-generated analysis.

Citation & reuse

Analysis on this page is generated by Patentable — an AI-powered patent intelligence platform. AI-generated summaries, explanations, and analysis may be reused with attribution and a visible link back to the canonical URL below. Patent abstracts and claims are USPTO public domain.

Cite as: Patentable. “METHOD AND SYSTEM FOR MULTI-SUBJECT PERSONALIZATION WITH SELECTIVE U-NET INFLUENCE” (US-20260017846-A1). https://patentable.app/patents/US-20260017846-A1

© 2026 Patentable. All rights reserved.

Patentable is a research and drafting-assistant tool, not a law firm, and does not provide legal advice. Documents we generate are drafts for review by a licensed patent attorney.