alphaFeedback

WorksComputer Science

Continual Concept Personalization in Diffusion Models via Residual-Space Adaptation

AI contributions
Not recorded for this paper
View details

Abstract

Text-to-image personalization aims to equip pretrained diffusion models with the ability to learn user-specific concepts from only a few reference images. Existing methods, however, typically train each concept independently or assume that all concepts are available simultaneously. In practical applications, concepts often arrive sequentially, requiring models to continually learn new subjects, objects, or styles while minimizing interference with previously acquired concepts. We address this continual concept personalization setting by extending residual-space adaptation to the stochastic adaptation process of pretrained diffusion models. Our work achieves the highest concept fidelity among competing continual personalization methods while maintaining near-zero forgetting. Beyond these empirical gains, the learned residual-space dynamics reveal where adaptation capacity is concentrated within the diffusion model. Restricting adaptation to the identified layers by our method preserves nearly the same performance while reducing the number of trainable parameters by up to 72%, demonstrating that residual-space dynamics can identify the layers that contribute most to continual concept personalization and directly guide more parameter-efficient adaptation.

Introduction

Recent text-to-image models such as Stable Diffusion [15, 17] and FLUX [11] have enabled high-quality image synthesis from natural language descriptions, making generative models increasingly accessible to end users. Their flexibility has further motivated personalization methods that adapt pretrained models to user-specific concepts, subjects, objects, or artistic styles from only a few examples.

Personalization methods commonly rely on parameter-efficient fine-tuning techniques such as LoRA [9] to adapt pretrained diffusion models to new concepts. Most existing approaches [7, 10, 14, 21] perform a separate fine-tuning procedure for every concept, and subsequently merge the resulting adapters into a unified personalized model. In many practical scenarios, however, concepts are acquired sequentially over time as new subjects, objects, or styles become available. Under this setting, whenever a new concept is introduced, the expensive merging procedure must be repeated to reconstruct the final personalized model. As the number of concepts grows, this repeated merging process becomes increasingly inefficient. This motivates continual concept personalization, where a single model is updated incrementally to incorporate new concepts while preserving previously acquired ones. The main obstacle to continual concept personalization is interference: learning new concepts can degrade performance on previously learned concepts. At the same time, the fixed model capacity must be used parsimoniously to preserve sufficient adaptation capacity for future concepts. Several continual learning methods [1, 4, 12, 19] achieve low levels of interference by constraining the parameter update to a pre-computed safe subspace. The most successful among

Preprint. these rely on layer-wise input statistics to characterize the safe subspace. This procedure cannot be directly extended to stochastic backbones, thus limiting its applicability in image generation tasks. In this work, we bridge this gap, extending residual-space adaptation to stochastic diffusion models for continual concept personalization. In particular, our starting point is an adapted version of DyRA [1] into diffusion models, a recent method which we choose for its elegant formulation and explicit theoretical guarantees. We also introduce a novel systematic evaluation framework for continual concept personalization and show that the resulting residual-space dynamics reveal layerwise specialization, providing a practical tool for guiding parameter-efficient adaptation.

Contributions. Our contributions are as follows:

  • We extend residual-space adaptation from deterministic language models to stochastic diffusion training, providing a new competitive option for continual concept personalization that mitigates interference while maintaining adaptivity at low cost.

  • We introduce a systematic evaluation framework for continual personalization that captures concept fidelity, going beyond judging only image-generation quality.

  • We show that residual-space dynamics can serve as an efficient probe for identifying the layers that contribute most to continual concept adaptation, providing insights into layerwise specialization and guiding more targeted parameter-efficient fine-tuning strategies.

Related Work

Diffusion Personalization

Text-to-image personalization has been widely studied for adapting pretrained diffusion models to user-specific identities, objects, or styles from only a few examples. Early approaches such as Textual Inversion [6] learn new concepts in the text embedding space while keeping the diffusion model fixed. DreamBooth [18] improves personalization quality by fine-tuning the diffusion model using a prior-preservation objective. Subsequent methods such as Custom Diffusion [10] restrict adaptation to selected cross-attention layers, improving parameter efficiency while supporting multi-concept customization. More recent personalization methods increasingly adopt parameter-efficient fine-tuning techniques such as LoRA. Mix-of-Show [7] introduces ED-LoRA together with structured concept tokens for multi-concept personalization, while LoRACLR [21] improves composition through contrastive alignment of independently trained adapters.

Another line of work reduces interference through the design of the adaptation space itself. Orthogonal Adaptation (OA) [14] constructs approximately orthogonal LoRA subspaces, allowing independently trained adapters to be combined through simple summation with minimal interference. Although originally proposed in a non-continual setting, its orthogonal structure naturally lends itself to sequential adaptation. Overall, most existing personalization methods assume that all concepts are jointly available during training or are learned independently and combined afterward, and are therefore not explicitly designed for continual settings.

Continual Adaptation

Continual learning methods [8] operate in a sequential training fashion, aiming to preserve knowledge and plasticity over time. Classical approaches rely on regularization, replay, or gradient projection mechanisms to mitigate catastrophic forgetting [23].

Within diffusion adaptation, C-LoRA [22] extends Custom Diffusion to a sequential setting by introducing regularization to preserve earlier concepts. Orthogonal Adaptation [14] can also be naturally extended to continual settings through additive composition of orthogonal adapters.

Several methods constrain the parameter update to a safe subspace based on the layer-wise input statistics. Among these, Gradient Projection Memory (GPM) [19] estimates layer-wise feature covariance from previous tasks and constrains the new updates to its low-energy subspace. The more recent InfLoRA [12] applies the same strategy to LoRA-based fine-tuning, where each task is learned within a fixed low-rank budget. Building upon InfLoRA, Barinka et al. [1] propose a dynamic rank allocation method with simplified updates which allows for significant capacity savings. In addition to providing theoretical guarantees for mitigating forgetting, the method demonstrates strong empirical performance on challenging continual language model adaptation tasks.

Although effective, these approaches have been studied primarily in deterministic architectures. Whether residual-space methods remain effective in stochastic diffusion training, where layer statistics are influenced by sampled noise, remains unexplored.

Evaluation of Personalized Generative Models

Evaluating personalized generative models typically involves measuring text alignment, image similarity, and identity preservation. Text alignment is commonly assessed using CLIP-based [16] text-image similarity, which measures whether generated images are semantically consistent with their prompts. Image alignment metrics compare generated images with reference examples, often using CLIP image embeddings. Several works additionally evaluate identity preservation through specialized recognition models such as ArcFace [3]. However, these metrics are mainly designed for human faces and do not generalize to arbitrary objects.

Beyond specialized identity metrics, a variety of pretrained embedding models have been used to assess perceptual similarity. CLIP [16] is trained on large-scale image–text pairs and provides representations that emphasize global semantic correspondence between images and language. In contrast, DINO-based models [2, 13, 20] learn self-supervised visual representations without text supervision and preserve fine-grained visual structure and local semantic information. DreamSim [5] combines multiple pretrained visual embeddings together with human similarity annotations, leveraging both global semantic understanding and fine-grained visual details to produce perceptual representations that closely align with human judgments.

Overall, most existing evaluation protocols are designed for static personalization settings and do not measure identity preservation or forgetting over time.

Evaluating Continual Performance

Evaluating continual text-to-image personalization requires measuring not only the quality of generated images but also how well previously learned concepts are preserved throughout sequential adaptation. Unlike discriminative tasks, where forgetting can be directly quantified through accuracy, there is no canonical notion of forgetting for generative models. As discussed in § 2.3, existing evaluation protocols are primarily designed for static personalization settings and focus on aggregate measures of text or image alignment. In order to assess the degradation of the concept identity over time, it is important to have metrics that distinguish concept fidelity from general image quality. To address this problem, we first introduce a concept-level fidelity measure that quantifies how faithfully a personalized concept is preserved. We then derive continual evaluation metrics directly from this quantity.

Concept Fidelity

Our objective is to evaluate the personalized concept itself rather than the similarity between individual images. A generated image may differ substantially from its reference in pose, background, or artistic style while still faithfully depicting the same concept. We therefore introduce concept fidelity, which measures similarity at the concept level rather than the image level.

To quantify concept fidelity, we first measure similarity between individual generated and reference images in a learned embedding space. Given an embedding function F(⋅)F(\cdot), we define the similarity between two images i1i_1 and i2i_2 as

sim⁡(i1,i2)=cos⁡(F(i1),F(i2)).(1)\operatorname{sim}(i_1,i_2) = \cos\left(F(i_1), F(i_2)\right). \tag*{(1)}

While image similarity is defined between individual image pairs, the fidelity of a personalized concept should be assessed from a set of generated images rather than a single sample. We therefore generalize the image-level similarity measure by aggregating pairwise similarities across the generated and reference image sets.

Let G={g1,…,gm}G = \{g_1,\ldots,g_m\} denote the generated images corresponding to a personalized concept and let I={i1,…,in}I = \{i_1,\ldots,i_n\} denote the associated reference images. The concept fidelity score is defined as

S(G,I)=1∣G∣∣I∣∑g∈G∑i∈Isim⁡(g,i).(2)S(G,I) = \frac{1}{|G||I|} \sum_{g\in G} \sum_{i\in I} \operatorname{sim}(g,i). \tag*{(2)}

This concept fidelity score extends image-level similarity to a set of generated images, providing a single measure of how well a model preserves a personalized concept.

We consider several pretrained embedding models, including CLIP [16], DINO [20], and DreamSim [5], as candidates for computing concept fidelity. Figure 1 provides a qualitative example of nearest-neighbor retrieval using the distance (1−sim⁡)(1-\operatorname{sim}), where sim⁡\operatorname{sim} denotes the similarity score defined in Eq. (1). DreamSim retrieves images belonging to the same personalized concept despite variations in pose and background, whereas CLIP and DINO retrieve visually similar images from different concepts. Our quantitative comparison in Appendix § 8 further supports that image similarity in the DreamSim embedding space best reflects concept-level similarity. We therefore adopt DreamSim as the default embedding model throughout the paper.

Nearest-neighbor retrieval across DreamSim, DINOv3, and CLIP embedding spaces

Figure 1. Nearest-neighbor retrieval for one reference image across embedding spaces. Red frames indicate retrieved images that belong to a different concept than the query.

Final Fidelity and Forgetting Metrics

Using the concept fidelity score, we now define continual evaluation metrics that summarize model performance throughout sequential adaptation.

Let NN denote the total number of concepts. For concept kk, let Gk(T)G_k^{(T)} denote the generated images after all concepts have been learned.

Final concept fidelity. We define the final concept fidelity as:

Q=1N∑k=1NS(Gk(T),Ik),(3)Q = \frac{1}{N}\sum_{k=1}^{N} S\left(G_k^{(T)}, I_k\right), \tag*{(3)}

which characterizes the average concept fidelity retained at the end of continual training.

Forgetting dynamics and final forgetting. To investigate the dynamics of forgetting throughout training, let tkt_k denote the training stage immediately after learning concept kk. We define the average forgetting trajectory at training stage t≥tkt \ge t_k as

F(t)=1nt∑k=1nt[S(Gk(tk),Ik)−S(Gk(t),Ik)],(4)F(t) = \frac{1}{n_t}\sum_{k=1}^{n_t}\left[S\left(G_k^{(t_k)}, I_k\right)-S\left(G_k^{(t)}, I_k\right)\right], \tag*{(4)}

where ntn_t is the number of concepts that have been learned by stage tt. The trajectory measures the average loss of concept fidelity as new concepts are acquired.

The final forgetting is defined as

F=F(T),(5)F = F(T), \tag*{(5)}

which measures the average loss of concept fidelity after continual training.

Together, QQ summarizes the model’s average concept fidelity after continual training, while F(t)F(t) characterizes the forgetting dynamics throughout sequential adaptation.

Residual-Space Adaptation in Diffusion

We build upon the residual-space adaptation framework DyRA [1], originally designed for LLMs, which constructs layer-wise adaptation subspaces from the input representations. Applying this framework to diffusion models is nontrivial, as these representations depend not only on the input concept but also on the randomly sampled diffusion noise and denoising timestep. We therefore first review the residual-space adaptation procedure and then develop a stochastic interpretation that justifies applying the framework to diffusion models and motivates our implementation.

Background: DyRA

We consider a sequence of concept personalization tasks {Dt}t=1T\{\mathcal{D}_t\}_{t=1}^{T}, where each task introduces a new concept tt and training is performed sequentially. Following [1], each adapted layer ll is parameterized by a LoRA update

ΔWt(l)=At(l)Bt(l),(6)\Delta W_t^{(l)} = A_t^{(l)} B_t^{(l)}, \tag*{(6)}

where Bt(l)B_t^{(l)} is initialized before training and kept fixed, while only At(l)A_t^{(l)} is optimized. The accumulated model after task tt is

Wt(l)=W(l)+∑i=1tΔWi(l).(7)W_t^{(l)} = W^{(l)} + \sum_{i=1}^{t} \Delta W_i^{(l)}. \tag*{(7)}

The key idea of DyRA is to restrict learning to feature directions that are informative for the current task while avoiding directions already captured by previous tasks. For each adapted layer, the method applies a spectral decomposition of the input features matrix from all tasks, and identifies therein directions unique to the current task, i.e. the residual space leftover from previous tasks. Since different layers exhibit different residual-space spectra, the number of informative directions also varies, motivating adaptive rank allocation instead of a fixed LoRA rank.

Concretely, the procedure for each adapted layer consists of three steps:

  1. Estimate the input subspace most representative of previous tasks, named Mt−1(l)\mathcal{M}_{t-1}^{(l)}.

  2. Project the current task input space Ht(l)\mathcal{H}_t^{(l)} onto the orthogonal complement of Mt−1(l)\mathcal{M}_{t-1}^{(l)} to obtain the residual space.

  3. Initialize the LoRA down-projection matrix Bt(l)B_t^{(l)} in the residual space, picking only directions with singular values above a given threshold τ\tau. During training, Bt(l)B_t^{(l)} remains fixed and only the corresponding up-projection matrix At(l)A_t^{(l)} is optimized.

Importantly, it is the threshold which implicitly determines the layer-wise rank. This is the key to flexible rank allocation.

A Stochastic Formulation for Diffusion

A key step of residual-space adaptation is constructing the residual space from layer input representations collected before training each task. For deterministic backbones, this procedure is well defined since each training sample corresponds to a unique layer input. Diffusion models, however, introduce an additional source of stochasticity through the sampled diffusion noise and denoising timestep, so the layer input of a training sample is no longer unique. This raises the central question of this section: how should the layer input space Ht(l)\mathcal{H}_t^{(l)} be estimated for residual-space construction?

Start by interpreting the layer input as the random variable

xt(l)=xt(l)(s,ϵ),(8)x_t^{(l)} = x_t^{(l)}(s,\epsilon), \tag*{(8)}

where ϵ∼N(0,I)\epsilon\sim\mathcal{N}(0, I) denotes the sampled diffusion noise and ss denotes the sampled denoising timestep.

Extending the deterministic interference measure in [1], ∥Bj(l)xi(l)∥2\lVert B_j^{(l)} x_i^{(l)} \rVert^2 to its expectation, consider the following stochastic interpretation of residual-space adaptation: projection directions should maximize the expected adaptation ability of the current task while suppressing the expected interference with previously learned tasks,

max⁡Bj(l)Es,ϵ[∥Bj(l)xj(l)∥2]s.t.Es,ϵ[∥Bj(l)xi(l)∥2]≤δ,i<j.(9)\begin{aligned} \max_{B_j^{(l)}}\quad&\mathbb{E}_{s,\epsilon}\left[\left\|B_j^{(l)}x_j^{(l)}\right\|^2\right] \\ \text{s.t.}\quad&\mathbb{E}_{s,\epsilon}\left[\left\|B_j^{(l)}x_i^{(l)}\right\|^2\right] \leq\delta,\quad i<j. \tag*{(9)} \end{aligned}

From this interpretation the answer to our question directly follows. Since the optimization objective is defined over the training distribution of diffusion noise and denoising timesteps, the collected layer inputs should provide an empirical estimate of the same distribution. Therefore, layer inputs are collected using exactly the same sampling strategy as diffusion training, namely Gaussian noise with uniformly sampled denoising timesteps. This simple step extends residual-space adaptation from deterministic feature representations to stochastic diffusion training by aligning the feature collection procedure with the stochastic optimization objective. Detailed implementation and derivation are provided in Appendix § 9.

Stable Continual Personalization

This section evaluates whether residual-space adaptation can be successfully extended to continual diffusion personalization. We first assess its ability to preserve concept fidelity while mitigating forgetting throughout continual learning, and then examine scalability and capacity preservation.

Experimental Setup

We conduct all experiments on Concept101 [10], a benchmark containing 101 concepts spanning 17 semantic classes, including visually similar concepts, making it particularly suitable for studying interference and identity preservation in continual learning. All concepts are learned sequentially in a rehearsal-free setting, where only the current concept data are available during training. Unless otherwise stated, all methods follow the same randomly shuffled task ordering. Throughout the experiments, we denote DyRA configured with the singular value threshold τ\tau as DyRA-τ\tau. We compare DyRA with τ∈{0.5,1.0}\tau\in\{0.5, 1.0\} against three continual LoRA baselines: Trainable-AB, which optimizes both LoRA projection matrices AA, BB and corresponds to a continual variant of Mix-of-Show [7]; Rank-Fixed-B, which optimizes only the LoRA up-projection AA while keeping the randomly initialized down-projection BB fixed; and OA, the continual version of Orthogonal Adaptation [14]. Unless otherwise specified, all methods adapt the linear projections in the transformer blocks of both the text encoder and the UNet, and all fixed-rank methods use rank r=4r=4. All experiments are conducted on Stable Diffusion v1.5 [17]. Models are periodically evaluated throughout continual training. For each concept, images are generated using only the learned concept token as the prompt, without additional textual descriptions, thereby isolating the learned concept representation and enabling a direct comparison of concept fidelity across methods. At the end of training, we further validate the model using diverse prompt templates. Additional implementation details and training hyperparameters are provided in Appendix § 10.

Concept Fidelity and Forgetting

Quantitative Results. Table 1 summarizes the final concept fidelity (QQ) and forgetting (FF). Trainable-AB achieves competitive concept fidelity but exhibits the largest forgetting, highlighting the trade-off between adapting to new concepts and preserving previously learned ones. In contrast, Rank-Fixed-B and OA effectively suppress forgetting but achieve slightly lower concept fidelity. DyRA maintains negligible forgetting while achieving the highest concept fidelity. These results demonstrate that residual-space adaptation extends effectively to the stochastic training process of diffusion models.

MethodQQ ↑FF ↓
Trainable-AB0.6392 ± 0.00150.0368 ± 0.0014
Rank-Fixed-B0.6442 ± 0.0029−0.0033 ± 0.0020
OA0.6435 ± 0.0031−0.0023 ± 0.0003
DyRA-0.50.6454 ± 0.00220.0002 ± 0.0001
DyRA-1.00.6497 ± 0.0008−0.0001 ± 0.0007

Table 1. Concept fidelity (QQ) and forgetting (FF) scores. Values are mean ± SD over three seeds.

Qualitative Results. The quantitative improvements are also reflected in the generated images. Figure 2a tracks concept pet dog2_2, introduced at task 14, from the moment it is learned to the final model. As additional concepts are introduced, Trainable-AB exhibits the largest appearance drift, Rank-Fixed-B and OA preserve the concept with only minor degradation, and DyRA remains nearly unchanged. Figure 2b shows that DyRA maintains high concept fidelity under diverse prompt templates, demonstrating generalization beyond the token-only prompts used during continual evaluation.

Forgetting dynamics. While Table 1 summarizes the final forgetting score, Figure 3 reveals how forgetting evolves throughout continual learning. Trainable-AB oscillates around substantially higher forgetting values, indicating persistent forgetting throughout the training process. In contrast, Rank-Fixed-B, OA, and DyRA all fluctuate closely around zero, demonstrating that their low forgetting is maintained consistently over time rather than occurring only at the final evaluation. Figure 4 provides a zoomed-in view, enabling a clearer comparison of the forgetting dynamics among Rank-Fixed-B, OA, and DyRA.

Forgetting trajectories across continual training for Trainable-AB, Rank-Fixed-B, DyRA-0.5, and DyRA-1.0

Figure 3. Forgetting trajectory across continual training. Higher values indicate more forgetting.

Forgetting trajectories over four task intervals

Figure 4. Zoomed views of the forgetting trajectory over four task intervals.

Overall, the quantitative metrics, qualitative generations, and forgetting trajectories consistently demonstrate that DyRA preserves high concept fidelity while effectively mitigating catastrophic forgetting.

Dynamic Rank Improves Parameter Efficiency

The performance gains of DyRA stem not only from restricting adaptation to the residual space, but also from allocating adaptation capacity according to the residual-space structure of each layer. We therefore compare DyRA with the fixed-rank OA under different trainable parameter budgets to evaluate the effectiveness of dynamic rank allocation.

Figure 5 compares concept fidelity against the number of trainable parameters for OA with different fixed ranks and DyRA with different singular-value thresholds. Each point corresponds to a different parameter budget. Across the entire range of budgets, DyRA consistently achieves higher concept fidelity than OA with comparable numbers of trainable parameters, demonstrating a superior performance–parameter trade-off.

Parameter-efficiency frontier for OA and DyRA variants

Figure 5. Parameter-efficiency frontier comparing concept fidelity and trainable parameter count for OA and DyRA variants. Error bars indicate one standard deviation over three random seeds.

Notably, even under high singular-value thresholds, where DyRA has substantially fewer trainable parameters, it outperforms OA with considerably larger parameter budgets. These results indicate that allocating rank adaptively according to the residual-space structure is substantially more parameter-efficient than assigning a uniform rank to every layer, allowing DyRA to maintain strong concept fidelity under a reduced parameter budget.

Layer-wise Specialization

Section 5 demonstrated that DyRA achieves strong continual personalization performance while improving parameter efficiency through adaptive rank allocation. Beyond these empirical gains, the adaptive ranks themselves provide a signal of where adaptation capacity is consumed during continual learning, since additional rank is allocated only when new concepts introduce directions outside the accumulated input subspace. This section analyzes these rank dynamics to understand how adaptation capacity is distributed across the layers and how this information can inform future parameter-efficient adaptation.

Continual generation across the task sequence, showing reference images and generations from Trainable-AB, Rank-Fixed-B, OA, and DyRA-1.0 across Tasks 14, 36, 58, 80, and 101

(a) (a) Continual generation across the task sequence. Row 1 shows reference images; rows 2–5 show Trainable-AB, Rank-Fixed-B, OA, and DyRA-1.0.

Final generations under diverse prompt templates from Trainable-AB, Rank-Fixed-B, OA, and DyRA-1.0

(b) (b) Final generations under diverse prompt templates. Rows show Trainable-AB, Rank-Fixed-B, OA, and DyRA-1.0.

Figure 2. Qualitative results for concept 14 (pet_dog2).

Adaptation Capacity Concentration

We first analyze how DyRA distributes adaptation capacity across the adapted layers by averaging the allocated ranks over all 101 concepts. Figure 6 illustrates the average rank allocated to each layer. The x-axis indexes the adapted layers in architectural order, and the corresponding module names are provided in Appendix § 11.

Average allocated rank per adapted layer over 101 concepts

Figure 6. Average allocated rank per adapted layer over 101 concepts.

Rather than allocating the same rank to every layer, DyRA exhibits clear layer-dependent allocation patterns that closely follow the repeated transformer architecture. In the text encoder, the allocated ranks exhibit a four-layer periodicity that exactly mirrors the repeated key(K), value(V), query(Q), and output projection layers within each transformer block. A similar correspondence is observed in the UNet, where every group of eight adapted layers corresponds to the self-attention and cross-attention projections within a transformer block. Within every block, the two dominant rank peaks consistently coincide with the cross-attention key(K) and value(V) projection layers, which repeatedly receive substantially larger ranks than all other projections.

The repeated concentration of residual rank in the UNet cross-attention K,V projections indicates that these layers consistently consume more adaptation capacity than other adapted modules during continual concept personalization. This observation motivates a closer examination of these layers in the following analysis.

Early Capacity Saturation

The UNet cross-attention K,V projections receive the largest residual ranks. We next examine whether exhausting the residual capacity of these layers is associated with degraded adaptation to subsequently learned concepts.

Figure 7 summarizes the remaining rank of every adapted layer after learning all 101 concepts. Under DyRA-0.5, all 32 UNet cross-attention K,V projections become fully saturated, whereas no UNet layer saturates under DyRA-1.0. Figure 8 further illustrates representative remaining-rank trajectories. While non-saturated layers retain sufficient residual capacity throughout training, representative cross-attention K,V layers under DyRA-0.5 steadily exhaust their available rank and become saturated around tasks 77–78. In contrast, the corresponding layers under DyRA-1.0 retain residual capacity throughout the entire task sequence.

Remaining rank capacity after 101 tasks

Figure 7. Remaining rank capacity after 101 tasks.

Representative remaining-rank trajectories of UNet layers

Figure 8. Representative remaining-rank trajectories of UNet layers: a non-saturated layer (left) and a cross-attention K,V layer that saturates under DyRA-0.5 (right).

To assess the effect of this capacity exhaustion, Table 2 compares the final concept fidelity with the average fidelity of the final 20 learned concepts. DyRA-1.0 maintains essentially unchanged performance on later concepts, whereas DyRA-0.5 exhibits a clear late-task fidelity degradation after the cross-attention K,VK,V layers become saturated. Although this evidence is correlational, the consistent agreement between the rank allocation, capacity saturation, and late-task performance suggests that preserving residual capacity in the UNet cross-attention K,VK,V projections is important for sustaining continual concept adaptation over long task sequences.

MethodFinal QQ ↑Last-20 QQ ↑Gap ↓
DyRA-0.50.64540.63400.0114
DyRA-1.00.64970.6501-0.0004

Table 2. Final concept fidelity compared with the average fidelity on the final 20 learned concepts. The reported score are averaged over three seeds. Standard deviations are below 0.007 in all cases.

Layer-Guided Parameter-Efficient Adaptation

The analyses in § 6.1 and § 6.2 consistently identify the UNet cross-attention K,VK,V projections as the layers that require the largest residual adaptation subspaces throughout continual learning. This suggests that these layers differentiate newly learned concepts through distinct adaptation directions. If this interpretation is correct, restricting adaptation to the identified layers would be expected to preserve most of the continual personalization performance while substantially reducing the number of trainable parameters.

Table 3 supports this hypothesis. Restricting adaptation to only the UNet K,VK,V projections reduces the trainable parameter count by approximately 60% (67.56M → 27.12M) while achieving virtually identical concept fidelity (QQ = 0.6496 vs. 0.6497) and comparable forgetting. Restricting adaptation further to only the UNet cross-attention K,VK,V projections reduces the trainable parameter count by approximately 72% (67.56M → 18.87M), yet still preserves nearly all continual personalization performance in practice.

Beyond validating the proposed interpretation, these results also provide a structural explanation for previous personalization methods such as Custom Diffusion [10] and C-LoRA [22], which restrict adaptation to cross-attention K,VK,V projections based primarily on empirical observations of large weight updates during training. In contrast, residual-space dynamics explain the same design choice from the perspective of feature-space utilization, providing a principled interpretation of why these layers are particularly effective adaptation targets.

More broadly, residual-space dynamics provide a principled way to identify where adaptation capacity is most effectively utilized. Rather than relying on empirical heuristics or exhaustive layerwise search, the proposed analysis directly guides targeted adaptation, enabling substantially more parameter-efficient continual personalization while preserving nearly the same overall performance.

MethodQQ ↑FF ↓Params (M)
Full DyRA0.6497 ± 0.0008−0.0001 ± 0.000767.56 ± 0.06
UNet K,VK,V0.6496 ± 0.0013−0.0011 ± 0.000327.12 ± 0.06
UNet cross-attention K,VK,V0.6479 ± 0.0004−0.0008 ± 0.000218.87 ± 0.00

Table 3. Performance of targeted adaptation strategies guided by residual-space dynamics. Values are mean ± SD over three seeds.

Conclusion

Limitations and Future Directions. Although DyRA achieves strong continual personalization performance, several limitations remain. First, the adaptation capacity of residual-space methods is fundamentally bounded by the width of individual layers. As continual learning progresses, layers with consistently high adaptation demand may eventually exhaust their available residual subspaces, limiting scalability to arbitrarily long task sequences. Second, the method is implemented only on Stable Diffusion v1.5 and Concept101. Nevertheless, the residual-space analysis developed in this work is considerably more general than the specific adaptation algorithm. Since it relies only on layer input statistics and residual-space dynamics, it can potentially be applied to other diffusion backbones, transformer-based generative models, or broader neural architectures to identify the layers that contribute most to learning new tasks. Exploring these adaptation patterns across architectures may provide further insights into representation specialization and guide the design of future parameter-efficient adaptation methods.

Key Takeaways. Our work shows that residual-space adaptation extends naturally to stochastic diffusion models, enabling continual concept personalization with stable performance, negligible interference, and strong parameter efficiency. Beyond its empirical performance, the residual-space dynamics observed during continual adaptation reveal how adaptation demand is distributed across the adapted layers. By identifying the layers that consistently require new concept-specific adaptation directions, it provides a structural interpretation of existing adaptation strategies and directly guides more parameter-efficient adaptation. We hope this perspective motivates future work that uses adaptation dynamics not only to improve continual learning algorithms, but also to better understand layer specialization in modern generative models.

Embedding Metric Comparison

We quantitatively compare the embedding spaces used to compute concept fidelity in the main text. The goal of this comparison is to measure whether an embedding model captures concept-level identity rather than only low-level visual similarity. A suitable embedding should map images of the same personalized concept close to one another while separating images from different concepts.

We evaluate this property using a clustering protocol on the ground-truth reference images. Given image embeddings, we define the pairwise clustering distance as

d(i1,i2)=1−sim⁡(i1,i2),(10)d(i_1,i_2) = 1 - \operatorname{sim}(i_1,i_2), \tag*{(10)}

where sim⁡(⋅,⋅)\operatorname{sim}(\cdot,\cdot) denotes the cosine similarity in the corresponding embedding space. We cluster all ground-truth images into 101 clusters, matching the number of personalized concepts. To make the comparison deterministic and reduce sensitivity to random initialization, the cluster centroids are initialized using one image from each ground-truth concept.

We report clustering accuracy and Adjusted Rand Index (ARI) with respect to the ground-truth concept labels. Clustering accuracy is computed as the fraction of images assigned to their correct concept cluster. ARI measures the agreement between the predicted clustering and the ground-truth partition based on pairwise co-clustering decisions, adjusted for chance. Higher values indicate that the embedding space better recovers the underlying concept structure.

As shown in Table 4, DreamSim achieves the highest performance on both metrics. This quantitative result complements the nearest-neighbor visualization in the main text and supports the use of DreamSim as the default embedding for concept fidelity.

Embedding modelAcc.ARI
DINOv3 ViT-S/160.90850.8304
DINO ViT-B/160.91960.8044
OpenCLIP ViT-B/320.95430.8935
CLIP ViT-B/320.93850.8873
DreamSim ensemble0.95580.9050

Table 4. Clustering-based comparison of embedding spaces on ground-truth reference images.

Stochastic Residual-Space Derivation

This section derives the residual-space construction used in the main text as the solution of a stochastic adaptation objective. We focus on one adapted layer ll and omit the layer superscript when the notation is unambiguous.

Stochastic Objective

For concept ctc_t, the layer input in a diffusion model depends on the sampled training image, the diffusion timestep ss, and the Gaussian noise ϵ\epsilon. We therefore treat the layer input as a random variable

xt(l)=x(l)(ct,s,ϵ),ϵ∼N(0,I),(11)x_t^{(l)} = x^{(l)}(c_t,s,\epsilon), \qquad\epsilon\sim\mathcal{N}(0,I), \tag*{(11)}

which induces a concept-level feature distribution pt(l)(x)p_t^{(l)}(x). Let

Mt(l)=Ext∼pt(l)[xtxt⊤](12)M_t^{(l)} = \mathbb{E}_{x_t\sim p_t^{(l)}}[x_t x_t^{\top}] \tag*{(12)}

denote its second-moment matrix.

The new projection Bt(l)B_t^{(l)} should satisfy two competing objectives. For the current concept, it should preserve as much projected feature energy as possible:

max⁡Bt(l)Ext∼pt(l)[∥Bt(l)xt∥2],Bt(l)(Bt(l))⊤=I.(13)\max_{B_t^{(l)}} \mathbb{E}_{x_t\sim p_t^{(l)}}\left[\left\|B_t^{(l)}x_t\right\|^2\right], \qquad B_t^{(l)}\left(B_t^{(l)}\right)^{\top}=I. \tag*{(13)}

For a previous concept cic_i, i<ti < t, it should instead have small expected response:

min⁡Bt(l)Exi∼pi(l)[∥Bt(l)xi∥2],Bt(l)(Bt(l))⊤=I.(14)\min_{B_t^{(l)}} \mathbb{E}_{x_i\sim p_i^{(l)}}\left[\left\|B_t^{(l)}x_i\right\|^2\right], \qquad B_t^{(l)}\left(B_t^{(l)}\right)^{\top}=I. \tag*{(14)}

For any concept cic_i, this expected response can be written as

Exi∼pi[∥Bt(l)xi∥2]=E[xiT(Bt(l))TBt(l)xi]=Tr⁡(Bt(l)Mi(l)(Bt(l))T).(15)\mathbb{E}_{x_i\sim p_i}\left[\left\|B_t^{(l)}x_i\right\|^2\right] = \mathbb{E}\left[x_i^{\mathsf{T}}\left(B_t^{(l)}\right)^{\mathsf{T}}B_t^{(l)}x_i\right] = \operatorname{Tr}\left(B_t^{(l)}M_i^{(l)}\left(B_t^{(l)}\right)^{\mathsf{T}}\right). \tag*{(15)}

Let the eigenvalues of Mi(l)M_i^{(l)} be sorted in descending order. By the Rayleigh–Ritz variational principle, the exact rank-rr minimizer of Eq. (14) is the bottom-rr eigenspace of Mi(l)M_i^{(l)}, while the exact rank-rr maximizer of Eq. (13) is the top-rr eigenspace of Mt(l)M_t^{(l)}. Enforcing the bottom eigenspace solution simultaneously for all previous concepts is impractical. DyRA instead uses a tractable relaxation: after a task selects its dominant adaptation directions, later tasks are constrained to be orthogonal to those directions. This suppresses the directions with largest expected response on previous concepts while leaving a residual subspace for new concepts.

Residual-Space Construction

For the first task, there is no previous occupied subspace. The method therefore solves only the adaptability objective in Eq. (13). Its rank-r1(l)r_1^{(l)} solution is

R1(l)=TopEig⁡r1(l)(M1(l)),Row⁡(B1(l))=R1(l).(16)\mathcal{R}_1^{(l)} = \operatorname{TopEig}_{r_1^{(l)}}\left(M_1^{(l)}\right), \qquad\operatorname{Row}\left(B_1^{(l)}\right)=\mathcal{R}_1^{(l)}. \tag*{(16)}

We set M1(l)=R1(l)M_1^{(l)}=\mathcal{R}_1^{(l)}.

For task t>1t>1, let Mt−1(l)\mathcal{M}_{t-1}^{(l)} denote the accumulated occupied subspace from previous tasks and let Pt−1(l)P_{t-1}^{(l)} be its orthogonal projector. The relaxed interference objective constrains the new projection to avoid previously selected dominant directions:

Row⁡(Bt(l))⊆(Mt−1(l))⊥.(17)\operatorname{Row}\left(B_t^{(l)}\right)\subseteq\left(\mathcal{M}_{t-1}^{(l)}\right)^\perp. \tag*{(17)}

This constraint implies

Bt(l)Pt−1(l)=0.(18)B_t^{(l)}P_{t-1}^{(l)}=0. \tag*{(18)}

Therefore the current feature can be replaced by its residual component in the adaptability objective:

Bt(l)xt=Bt(l)(I−Pt−1(l))xt.(19)B_t^{(l)}x_t=B_t^{(l)}\left(I-P_{t-1}^{(l)}\right)x_t. \tag*{(19)}

The constrained current-task objective becomes

max⁡Bt(l)E[∥Bt(l)(I−Pt−1(l))xt∥2]s.t.Row⁡(Bt(l))⊆(Mt−1(l))⊥,Bt(l)(Bt(l))T=I.(20)\begin{aligned} \max_{B_t^{(l)}}\quad&\mathbb{E}\left[\left\|B_t^{(l)}\left(I-P_{t-1}^{(l)}\right)x_t\right\|^2\right]\\ \text{s.t.}\quad&\operatorname{Row}\left(B_t^{(l)}\right)\subseteq\left(\mathcal{M}_{t-1}^{(l)}\right)^\perp,\\ & B_t^{(l)}\left(B_t^{(l)}\right)^{\mathsf{T}}=I. \tag*{(20)} \end{aligned}

The corresponding residual second-moment matrix is

Mt,⊥(l)=(I−Pt−1(l))Mt(l)(I−Pt−1(l)).(21)M_{t,\perp}^{(l)}=\left(I-P_{t-1}^{(l)}\right)M_t^{(l)}\left(I-P_{t-1}^{(l)}\right). \tag*{(21)}

Using Eq. (15), the objective in Eq. (20) becomes

max⁡Bt(l)Tr⁡(Bt(l)Mt,⊥(l)(Bt(l))T)s.t.Row⁡(Bt(l))⊆(Mt−1(l))⊥,Bt(l)(Bt(l))T=I.(22)\begin{aligned} \max_{B_t^{(l)}}\quad&\operatorname{Tr}\left(B_t^{(l)}M_{t,\perp}^{(l)}\left(B_t^{(l)}\right)^{\mathsf{T}}\right)\\ \text{s.t.}\quad&\operatorname{Row}\left(B_t^{(l)}\right)\subseteq\left(\mathcal{M}_{t-1}^{(l)}\right)^\perp,\\ & B_t^{(l)}\left(B_t^{(l)}\right)^{\mathsf{T}}=I. \tag*{(22)} \end{aligned}

By the Rayleigh–Ritz variational principle, the optimal rank-rt(l)r_t^{(l)} row space is the dominant eigenspace of the residual second-moment matrix:

Rt(l)=TopEig⁡rt(l)(Mt,⊥(l)),Row⁡(Bt(l))=Rt(l).(23)\mathcal{R}_t^{(l)}=\operatorname{TopEig}_{r_t^{(l)}}\left(M_{t,\perp}^{(l)}\right), \qquad\operatorname{Row}\left(B_t^{(l)}\right)=\mathcal{R}_t^{(l)}. \tag*{(23)}

The occupied subspace is then updated recursively as

Mt(l)=Mt−1(l)⊕Rt(l).(24)\mathcal{M}^{(l)}_t = \mathcal{M}^{(l)}_{t-1} \oplus\mathcal{R}^{(l)}_t . \tag*{(24)}

Consequently, the DyRA construction is the natural solution of the stochastic max/min adaptation problem under the practical interference relaxation: task 1 selects the dominant eigenspace of its stochastic features, and each later task selects the dominant eigenspace of the current stochastic features after projecting out directions occupied by previous tasks.

Empirical Estimation and Initialization

In implementation, Mt(l)\mathcal{M}^{(l)}_t is estimated from collected stochastic layer features. If Ht(l)=[xt,1(l),…,xt,n(l)]H^{(l)}_t = [x^{(l)}_{t,1},\ldots,x^{(l)}_{t,n}] is the feature matrix collected before training task tt, we use

M^t(l)=1nHt(l)(Ht(l))T.(25)\widehat{\mathcal{M}}^{(l)}_t = \frac{1}{n}H^{(l)}_t\left(H^{(l)}_t\right)^{\mathsf{T}}. \tag*{(25)}

The samples used to construct Ht(l)H^{(l)}_t follow the same stochastic process as diffusion training: each forward pass uses sampled Gaussian noise and a sampled denoising timestep. Therefore M^t(l)\widehat{\mathcal{M}}^{(l)}_t is an empirical estimate of the second moment appearing in the stochastic objective.

Algorithm 1 summarizes the initialization. The rank is selected independently for each layer by thresholding the singular values of the residual second-moment estimate, so a single threshold τ\tau induces layer-dependent ranks rt(l)r^{(l)}_t.

We maintain Q(l)Q^{(l)} as an orthonormal basis of the occupied subspace and form the projector as P(l)=Q(l)(Q(l))TP^{(l)} = Q^{(l)}\left(Q^{(l)}\right)^{\mathsf{T}}. This avoids accumulating numerical error from previously initialized LoRA matrices and keeps the residual covariance in Eq. (21) symmetric positive semidefinite up to numerical precision.

Implementation Details

We provide additional implementation details for the continual personalization experiments in the main text.

Dataset and task order. We randomly shuffle the 101 concepts once and use this fixed task order for all methods and random seeds in the main experiments. This avoids presenting concepts in semantic clusters, such as all visually similar instances from the same class arriving consecutively. During each task, the dataloader contains only the reference images for the current concept. To check robustness to task ordering, we additionally repeat all experiments with two other independently shuffled concept orders and observe results consistent with those reported in the main text.

Backbone and adapted layers. We use Stable Diffusion v1.5 [17] as the pretrained text-to-image backbone. The model uses a CLIP ViT-L/14 text encoder with 12 transformer blocks and a UNet with 16 cross-attention blocks. LoRA modules are inserted into the query, key, value, and output projections of the attention layers in both the text encoder and the UNet. This yields 48 adapted projection layers in the text encoder and 128 adapted projection layers in the UNet.

Training configuration. Each concept is trained for 400 gradient-update steps. Since each concept has only a small number of reference images, the concept dataset is cycled during training. We train with fp16 mixed precision on two NVIDIA RTX 4090 GPUs. All methods are run with three global random seeds, {0,42,619}\{0, 42, 619\}. The training prompt for concept cc follows the dual-token format

“a photo of ⟨c1⟩⟨c2⟩”.(26)\text{“a photo of }\langle c_1\rangle\langle c_2\rangle\text{”.} \tag*{(26)}

Residual-space construction. Layer inputs for subspace estimation are collected before training each new task using the same stochastic inputs as diffusion training: Gaussian noise is sampled for each image and the denoising timestep is sampled uniformly.

Algorithm 1: DyRA initialization for stochastic diffusion features.

Input: adapted layers Ladapt\mathcal{L}_{\mathrm{adapt}}; current data Dt\mathcal{D}_t; threshold τ\tau; task index tt; layer bases Q(l)Q^{(l)} and allocated ranks cumRank⁡(l)\operatorname{cumRank}^{(l)}, initialized as empty and zero before task 1

Output: fixed LoRA projections Bt(l)B_t^{(l)}; updated bases Q(l)Q^{(l)} and ranks cumRank⁡(l)\operatorname{cumRank}^{(l)}

  1. Collect stochastic layer features by forwarding Dt\mathcal{D}_t with sampled diffusion timesteps and Gaussian noise;

  2. Compute M^t(l)=1nHt(l)(Ht(l))T\widehat{M}_t^{(l)} = \frac{1}{n} H_t^{(l)}(H_t^{(l)})^{\mathsf{T}} for each l∈Ladaptl \in\mathcal{L}_{\mathrm{adapt}};

  3. foreach l∈Ladaptl \in\mathcal{L}_{\mathrm{adapt}} do

  4. &nbsp;&nbsp;if t=1t = 1 then

  5. &nbsp;&nbsp;&nbsp;&nbsp;C(l)←M^t(l)C^{(l)} \leftarrow\widehat{M}_t^{(l)};

  6. &nbsp;&nbsp;else

  7. &nbsp;&nbsp;&nbsp;&nbsp;if cumRank⁡(l)≥din(l)\operatorname{cumRank}^{(l)} \ge d_{\mathrm{in}}^{(l)} then continue;

  8. &nbsp;&nbsp;&nbsp;&nbsp;P(l)←Q(l)(Q(l))TP^{(l)} \leftarrow Q^{(l)}(Q^{(l)})^{\mathsf{T}};

  9. &nbsp;&nbsp;&nbsp;&nbsp;C(l)←(I−P(l))M^t(l)(I−P(l))C^{(l)} \leftarrow(I-P^{(l)})\widehat{M}_t^{(l)}(I-P^{(l)});

  10. &nbsp;&nbsp;end

  11. &nbsp;&nbsp;U,S←SVD⁡(C(l))U,S \leftarrow\operatorname{SVD}(C^{(l)});

  12. &nbsp;&nbsp;r←∣{k:Sk≥τ}∣r \leftarrow|\{k:S_k \ge\tau\}|;

  13. &nbsp;&nbsp;if t=1t = 1 then

  14. &nbsp;&nbsp;&nbsp;&nbsp;if r=0r = 0 then r←1r \leftarrow1;

  15. &nbsp;&nbsp;else

  16. &nbsp;&nbsp;&nbsp;&nbsp;if r=0r = 0 then continue;

  17. &nbsp;&nbsp;&nbsp;&nbsp;r←min⁡(r,din(l)−cumRank⁡(l))r \leftarrow\min(r,d_{\mathrm{in}}^{(l)}-\operatorname{cumRank}^{(l)});

  18. &nbsp;&nbsp;end

  19. &nbsp;&nbsp;Ur←U[:,1:r]U_r \leftarrow U[:,1:r];

  20. &nbsp;&nbsp;Bt(l)←UrB_t^{(l)} \leftarrow U_r;

  21. &nbsp;&nbsp;Initialize At(l)←0A_t^{(l)} \leftarrow0;

  22. &nbsp;&nbsp;Q(l)←QR⁡([Q(l)∣Ur])Q^{(l)} \leftarrow\operatorname{QR}([Q^{(l)}\mid U_r]);

  23. &nbsp;&nbsp;cumRank⁡(l)←cumRank⁡(l)+r\operatorname{cumRank}^{(l)} \leftarrow\operatorname{cumRank}^{(l)}+r;

  24. end

SettingValue
Training steps per task400
Evaluation intervalEvery 50 steps
Per-device batch size2
OptimizerAdamW
AdamW parametersβ1=0.9\beta_1=0.9, β2=0.999\beta_2=0.999, wd 0.01
Learning rate
LR scheduleLinear decay, no warmup
Token-embedding LR1×10−31 \times10^{-3}
Text-encoder LoRA LR1×10−51 \times10^{-5}
UNet LoRA LR1×10−41 \times10^{-4}
LoRA configuration
Rank (fixed-rank baselines)r=4r=4
Scaling coefficientα=1.0\alpha=1.0
Image preprocessing
Input resolution512×512512 \times512
Random cropp=0.5p=0.5
Random scale0.80.8–1.01.0

Table 5. Training hyperparameters shared across methods.

Evaluation protocol. We evaluate the model every 50 training steps. At each evaluation checkpoint, we generate images for all concepts that have been learned so far. For each concept, we sample 10 images using different random seeds. The generated images are compared with the ground-truth reference images using the concept-fidelity metric described in the main text. During this quantitative evaluation, we prompt the model only with the two concept-specific tokens, ⟨c1⟩⟨c2⟩\langle c_1\rangle\langle c_2\rangle, to isolate concept identity from prompt-induced scene variation.

Final qualitative validation. For final qualitative validation after all 101 concepts have been learned, we use the class-level prompt templates provided by Concept 101. Each semantic class has 10 prompts, and we replace the placeholder token in each template with the two concept-specific tokens. We sample 8 images per prompt with different random seeds. For the pet_dog2 validation example shown in the main text, the five selected prompts are <TOK> at a beach with a view of the seashore., <TOK> in Times Square., <TOK> is in a construction outfit., <TOK> is playing with a ball., and <TOK> is wearing a sombrero. In all templates, <TOK> is replaced by the two concept-specific tokens. All methods use the same prompts and seeds.

SettingValue
SamplerDPM-Solver++ multistep
Inference steps50
CFG scale7.5
Output resolution512×512512 \times512
Evaluation samples per concept10
Validation prompts per class10
Validation samples per prompt8

Table 6. Generation settings used for evaluation and final validation.

Layer-Index Mapping

This section provides the mapping between the adapted layer indices used in the layer-wise rank analysis and the corresponding attention projections. We use QQ, KK, VV, and OO to denote the query, key, value, and output projections, respectively. For the UNet, we explicitly distinguish self-attention from cross-attention projections.

Text encoder. The CLIP text encoder contains 12 transformer blocks. In each block, we adapt the self-attention projections in the order KK, VV, QQ, OO. Thus, for text-encoder block b∈{0,…,11}b \in\{0,\ldots,11\}, the adapted layer indices are

4b+{0,1,2,3}⟷{K,V,Q,O}.(27)4b + \{0,1,2,3\} \longleftrightarrow\{K,V,Q,O\}. \tag*{(27)}

All text-encoder adapted layers have input dimension din=768d_{\mathrm{in}} = 768.

Block bbKKVVQQOOdind_{\mathrm{in}}
00123768
14567768
2891011768
312131415768
416171819768
520212223768
624252627768
728293031768
832333435768
936373839768
1040414243768
1144454647768

Table 7. Text-encoder layer-index mapping. The text encoder uses self-attention only.

UNet. The UNet contains 16 adapted transformer modules across the down, up, and middle blocks. Each module contributes eight adapted projection layers: four self-attention projections followed by four cross-attention projections. For UNet transformer module mm, the self-attention indices are

8m+{0,1,2,3}⟷{self-Q,self-K,self-V,self-O},(28)8m + \{0,1,2,3\} \longleftrightarrow\{\mathrm{self}\text{-}Q,\mathrm{self}\text{-}K,\mathrm{self}\text{-}V,\mathrm{self}\text{-}O\}, \tag*{(28)}

and the cross-attention indices are

8m+{4,5,6,7}⟷{cross-Q,cross-K,cross-V,cross-O}.(29)8m + \{4, 5, 6, 7\} \longleftrightarrow\{\text{cross-}Q, \text{cross-}K, \text{cross-}V, \text{cross-}O\}. \tag*{(29)}

The cross-attention KK and VV projections take text-encoder features as input, so their input dimension is always 768. The remaining projections use the UNet hidden dimension of the corresponding transformer module.

mmRangeBlockUNet dim.
00–7down 0320
18–15down 0320
216–23down 1640
324–31down 1640
432–39down 21280
540–47down 21280
648–55up 11280
756–63up 11280
864–71up 11280
972–79up 21280
1080–87up 21280
1188–95up 21280
1296–103up 3320
13104–111up 3320
14112–119up 3320
15120–127middle1280

Table 8. UNet transformer-module mapping. Within each layer range, offsets 0–3 correspond to self-attention QQ, KK, VV, OO, and offsets 4–7 correspond to cross-attention QQ, KK, VV, OO.

This mapping explains the periodic structure in the rank-allocation plots. In particular, the UNet cross-attention KK, VV projections correspond to indices 8m+58m + 5 and 8m+68m + 6 for each transformer module mm, which are the layers that consistently receive the largest residual ranks in the analysis.

Additional Qualitative Examples

We provide additional final-validation and continual-generation examples for three other concepts spanning different categories. For each concept, the continual-generation figure is shown first, followed by the final validation generations. In the validation figures, columns P1P_1–P5P_5 correspond to the prompt set listed in the same subsection, and <TOK> is replaced by the two learned concept tokens. For each continual-generation figure, the first shown task is the task at which the concept is introduced.

Jewelry Ring

Validation prompts.

P1P_1: A close shot of <TOK> worn by a model.

P2P_2: A <TOK> lying in a red jewelry box.

P3P_3: A <TOK> on a blue velvet rug.

P4P_4: A <TOK> on a mossy forest floor.

P5P_5: A <TOK> on a sunlit beach, a seashell beside it.

Additional qualitative results for Jewelry Ring, showing continual generation beginning at task 23 and final validation across prompts $P_1$–$P_5$

Figure 9. (a) Continual generation, beginning at task 23. (b) Final validation; columns correspond to prompts P1P_1–P5P_5. Additional qualitative results for Jewelry Ring.

Plushie Panda

Validation prompts.

P1P_1: Photo of a <TOK>.

P2P_2: <TOK> in front of a medieval castle.

P3P_3: <TOK> in Grand Canyon.

P4P_4: <TOK> in Times Square.

P5P_5: <TOK> on a boat in the sea.

Additional qualitative results for Plushie Panda, showing continual generation across tasks 52, 64, 76, 88, and 101, and final validation for prompts P1–P5

Figure 10. (a) Continual generation, beginning at task 52. (b) Final validation; columns correspond to prompts P1P_1–P5P_5. Additional qualitative results for Plushie Panda.

Wearable Jacket

Validation prompts.

  • P1P_{1}: A humanoid robot wearing <TOK> with a futuristic city in the background.

  • P2P_{2}: A mannequin showcasing <TOK>.

  • P3P_{3}: A <TOK> on a hanger in the window of a shop.

  • P4P_{4}: A <TOK> on a hook in a study room with bookshelves.

  • P5P_{5}: Photo of a <TOK>.

Additional qualitative results for Wearable Jacket

Figure 11. (a) Continual generation, beginning at task 32. (b) Final validation; columns correspond to prompts P1P_{1}–P5P_{5}. Additional qualitative results for Wearable Jacket.

References

  1. [1]Louis Barinka, Giulia Lanzillotta, Enis Simsar, Tanguy Dieudonné, and Thomas Hofmann. Provable forgetting bounds drive capacity savings: Spectral thresholding in continual lora. In Continual Adaptation at Scale: Towards Sustainable AI Workshop, 2026.
  2. [2]Mathilde Caron, Hugo Touvron, Ishan Misra, Hervé Jégou, Julien Mairal, Piotr Bojanowski, and Armand Joulin. Emerging properties in self-supervised vision transformers. In Proceedings of the IEEE/CVF international conference on computer vision, pages 9650–9660, 2021.arxiv.org/abs/2104.14294
  3. [3]Jiankang Deng, Jia Guo, Niannan Xue, and Stefanos Zafeiriou. Arcface: Additive angular margin loss for deep face recognition. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pages 4690–4699, 2019.DOI
  4. [4]Mehrdad Farajtabar, Navid Azizan, Alex Mott, and Ang Li. Orthogonal Gradient Descent for Continual Learning. In Proceedings of the Twenty Third International Conference on Artificial Intelligence and Statistics, pages 3762–3773. PMLR, 2020.arxiv.org/abs/1910.07104
  5. [5]Stephanie Fu, Netanel Tamir, Shobhitha Sundaram, Lucy Chai, Richard Zhang, Tali Dekel, and Phillip Isola. Dreamsim: Learning new dimensions of human visual similarity using synthetic data. arXiv preprint arXiv:2306.09444, 2023.
  6. [6]Rinon Gal, Yuval Alaluf, Yuval Atzmon, Or Patashnik, Amit H Bermano, Gal Chechik, and Daniel Cohen-Or. An image is worth one word: Personalizing text-to-image generation using textual inversion. arXiv preprint arXiv:2208.01618, 2022.arxiv.org/abs/2208.01618
  7. [7]Yuchao Gu, Xintao Wang, Jay Zhangjie Wu, Yujun Shi, Yunpeng Chen, Zihan Fan, Wuyou Xiao, Rui Zhao, Shuning Chang, Weijia Wu, Yixiao Ge, Ying Shan, and Mike Zheng Shou. Mix-of-show: Decentralized low-rank adaptation for multi-concept customization of diffusion models, 2023.arxiv.org/abs/2305.18292
  8. [8]Raia Hadsell, Dushyant Rao, Andrej A Rus u, and Razvan Pascanu. Embracing change: Continual learning in deep neural networks. Trends in cognitive sciences, 24(12):1028–1040, 2020.DOI
  9. [9]Edward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yuanzhi Li, Shean Wang, Lu Wang, and Weizhu Chen. Lora: Low-rank adaptation of large language models, 2021.arxiv.org/abs/2106.09685
  10. [10]Nupur Kumari, Bingliang Zhang, Richard Zhang, Eli Shechtman, and Jun-Yan Zhu. Multi-concept customization of text-to-image diffusion. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pages 1931–1941, 2023.arxiv.org/abs/2212.04488
  11. [11]Black Forest Labs, Stephen Batifol, Andreas Blattmann, Frederic Boesel, Saksham Consul, Cyril Diagne, Tim Dockhorn, Jack English, Zion English, Patrick Esser, Sumith Kulal, Kyle Lacey, Yam Levi, Chen Li, Dominik Lorenz, Jonas Müller, Dustin Podell, Robin Rombach, Harry Saini, Axel Sauer, and Luke Smith. Flux.1 kontext: Flow matching for in-context image generation and editing in latent space, 2025.arxiv.org/abs/2506.15742
  12. [12]Yan-Shuo Liang and Wu-Jun Li. Inflora: Interference-free low-rank adaptation for continual learning. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 23638–23647, 2024.arxiv.org/abs/2404.00228
  13. [13]Maxime Oquab, Timothée Darcet, Théo Moutakanni, Huy Vo, Marc Szafraniec, Vasil Khalidov, Pierre Fernandez, Daniel Haziza, Francisco Massa, Alaaeldin El-Nouby, et al. Dinov2: Learning robust visual features without supervision. arXiv preprint arXiv:2304.07193, 2023.
  14. [14]Ryan Po, Guandao Yang, Kfir Aberman, and Gordon Wetzstein. Orthogonal adaptation for modular customization of diffusion models. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pages 7964–7973, 2024.arxiv.org/abs/2312.02432
  15. [15]Dustin Podell, Zion English, Kyle Lacey, Andreas Blattmann, Tim Dockhorn, Jonas Müller, Joe Penna, and Robin Rombach. Sdxl: Improving latent diffusion models for high-resolution image synthesis, 2023.arxiv.org/abs/2307.01952
  16. [16]Alec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh, Gabriel Goh, Sandhini Agarwal, Girish Sastry, Amanda Askell, Pamela Mishkin, Jack Clark, et al. Learning transferable visual models from natural language supervision. In International conference on machine learning, pages 8748–8763. PMLR, 2021.arxiv.org/abs/2103.00020
  17. [17]Robin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser, and Björn Ommer. High-resolution image synthesis with latent diffusion models. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pages 10684–10695, 2022.
  18. [18]Nathaniel Ruiz, Yuanzhen Li, Varun Jampani, Yael Pritch, Michael Rubinstein, and Kfir Aberman. Dreambooth: Fine tuning text-to-image diffusion models for subject-driven generation. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pages 22500–22510, 2023.arxiv.org/abs/2208.12242
  19. [19]Gobinda Saha, Isha Garg, and Kaushik Roy. Gradient projection memory for continual learning. arXiv preprint arXiv:2103.09762, 2021.arxiv.org/abs/2103.09762
  20. [20]Oriane Siméoni, Huy V Vo, Maximilian Seitzer, Federico Baldassarre, Maxime Oquab, Cijo Jose, Vasil Khalidov, Marc Szafraniec, Seungeun Yi, Michaël Ramamonjisoa, et al. Dinov3. arXiv preprint arXiv:2508.10104, 2025.arxiv.org/abs/2508.10104
  21. [21]Enis Simsar, Thomas Hofmann, Federico Tombari, and Pinar Yanardag. Loracl: Contrastive adaptation for customization of diffusion models, 2025.arxiv.org/abs/2412.09622
  22. [22]James Seale Smith, Yen-Chang Hsu, Lingyu Zhang, Ting Hua, Zsolt Kira, Yilin Shen, and Hongxia Jin. Continual diffusion: Continual customization of text-to-image diffusion with c-lora. arXiv preprint arXiv:2304.06027, 2023.arxiv.org/abs/2304.06027
  23. [23]Liyuan Wang, Xingxing Zhang, Hang Su, and Jun Zhu. A comprehensive survey of continual learning: Theory, method and application. IEEE transactions on pattern analysis and machine intelligence, 46(8): 5362–5383, 2024.DOI

Paper details

vlnc:W_phdsmpgb
v1 · latestHistory (1)

published · Fresh — added to the Hub within the last 30 days

added ·

Usage

views · 0

downloads · 0

A view counts once per browser per day, a download each time the PDF is requested.

CC BY 4.0

Original source

  • Computer Science

Not recorded for this paper · View statement →

Looking for related papers…
Contents