atlas

Research map / Image generation / Text-to-image diffusion

Recovering the Pre-Fine-Tuning Weights of Generative Models

ICML 2024 · Image generation › Text-to-image diffusion · approach: Multi-subject personalization · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows that the pre-fine-tuning weights of LoRA-fine-tuned generative and vision models can be recovered from several public LoRA variants via iterative low-rank factorization (Spectral DeTuning) with a rank scheduler.

Problem

Pre-fine-tuning weights of models are assumed unrecoverable after alignment/personalization fine-tuning; the paper asks whether multiple LoRA fine-tuned variants of the same source model leak the original weights.

Approach

Model the fine-tuned weights as a shared matrix plus per-model low-rank residuals and solve alternating truncated-SVD (M-step) and averaging (W-step) updates, with a rank scheduler that increases the allowed residual rank on plateau; ranks are estimated via pairwise difference ranks and an LP.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Spectral DeTuning instead of Mean LoRAViT weight recovery · ViTW-Error: -5.214 -> -15.942Helped
Use Spectral DeTuning instead of Mean LoRAViT weight recovery · ViTAct.-Dist.: 5e{-}2 -> 1e{-}6Helped
Use Spectral DeTuning instead of Mean LoRAStable Diffusion 1.5 personalization recovery · Stable Diffusion 1.5 personalizationW-Error: -7.540 -> -17.816Helped
Use Spectral DeTuning instead of Mean LoRAStable Diffusion 1.5 personalization recovery · Stable Diffusion 1.5 personalizationLPIPS: 0.482 -> 0.009Helped
Use Spectral DeTuning instead of Mean LoRAMistral-7B SFT recovery · Mistral-7B SFTW-Error: -9.299 -> -16.502Helped
Use Spectral DeTuning instead of Mean LoRAMistral-7B SFT recovery · Mistral-7B SFTSBERT: -1.007 -> -9.324Helped
Use Spectral DeTuning instead of Mean LoRAMistral-7B DPO recovery · Mistral-7B DPOW-Error: -10.757 -> -22.062Helped
Use Spectral DeTuning instead of Mean LoRAMistral-7B DPO recovery · Mistral-7B DPOSBERT: -3.455 -> -14.708Helped
Add the rank scheduler (ablation)Spectral DeTuning on Stable Diffusion · Stable Diffusion 1.5 personalizationW-Error convergenceHelped
Use varying LoRA ranks with a dedicated rank scheduler for each LoRA (ablation)Spectral DeTuning · Stable Diffusion 1.5, varying ranks [8,32,32,32,64,100]LPIPS: 0.009 -> 0.073Hurt
Estimate LoRA ranks using the rank-estimation heuristicSpectral DeTuning · Hundreds of LoRA combinations with different ranksrank estimation accuracy: 100%Helped

Surprising findings

Limitations

Experimental setup

Datasets: VTAB-1K ImageNet-1K UltraChat UltraFeedback COCO Captions AlpacaFarm evaluation

Benchmarks: LoWRA Bench

Metrics: W-Error (log10 MSE) Activation cosine distance LPIPS SBERT log cosine distance

Baselines: LoRA FT Mean LoRA

Builds on: LoRA VTAB-1K DPO

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy