Generative models trained on sensitive image datasets risk memorizing and reproducing individual training examples, making strong privacy guarantees essential. While differential privacy (DP) provides a principled framework for such guarantees, standard DP finetuning (e.g., with DP-SGD) often results in severe degradation of image quality, particularly in high-frequency textures, due to the indiscriminate addition of noise across all model parameters. In this work, we propose a spectral DP framework based on the hypothesis that the most privacy-sensitive portions of an image are often low-frequency components in the wavelet space (e.g., facial features and object shapes) while high-frequency components are largely generic and public. Based on this hypothesis, we propose the following two-stage framework for DP image generation with coarse image intermediaries: (1) DP finetune an autoregressive spectral image tokenizer model on the low-resolution wavelet coefficients of the sensitive images, and (2) perform high-resolution upsampling using a publicly pretrained super-resolution model. By restricting the privacy budget to the global structures of the image in the first stage, and leveraging the post-processing property of DP for detail refinement, we achieve promising trade-offs between privacy and utility. Experiments on the MS-COCO and MM-CelebA-HQ datasets show that our method generates images with improved quality and style capture relative to other leading DP image frameworks.
Figures & tables
Figure 1: Illustration of the DP-Wavelet method.
Figure 2: Visualization of the 2D Discrete Wavelet Transform. Left . Original image, Center . A one-level decomposition where the image is split into the approximation LL0 and details {LH0,HL0,HH0} , Right , A two-level decomposition where the image is first split into scale 1 details in blue and a scale 1 approximation, which is then recursively split into scale 0 details in green and the final approximation LL0 in red. Note that indices are relative to the coarsest scale LL0 .
Figure 3: Left . Pixel-based patchification, Right . Wavelet-based patchification where separate codebooks are used for the red, green, and blue patches [ 17 ] .
Figure 4: An example of AR-SIT’s partial generation of images using 25%, 50%, 75%, and 100% (left-to-right) of the full AR transformer token sequence [ 17 ] .
FID ↓ / LPIPS ↓
DP-LDM
DP-LlamaGen
DP-Wavelet
Pretrained
13.2 / 0.759
17.7 / 0.728
23.1 / 0.751
ε=∞
12.2 / 0.755
11.6 / 0.715
15.5 / 0.666
ε=10
12.3 / 0.757
13.8 / 0.723
19.4 / 0.751
ε=1
12.8 / 0.758
16.1 / 0.728
20.1 / 0.749
Table 1: Performance on MS-COCO.
Figure 5: Sample images generated by the benchmark algorithms for MS-COCO and MM-CelebA-HQ.
FID ↓ / LPIPS ↓
DP-LDM
DP-LlamaGen
DP-Wavelet
Pretrained
64.7 / 0.734
86.5 / 0.672
54.5 / 0.711
ε=∞
25.7 / 0.726
32.5 / 0.586
18.2 / 0.581
ε=10
22.9 / 0.719
63.7 / 0.649
22.2 / 0.659
ε=1
25.9 / 0.726
74.2 / 0.659
39.4 / 0.678
Table 2: Performance on MM-CelebA-HQ.
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
Setting
DP-LDM [ 28 ]
DP-LlamaGen [ 40 ]
DP-Wavelet (Ours)
Base Architecture
Stable Diffusion v1.4 (UNet)
LlamaGen-XL
AR-SIT (SIT-SC-5)
Trainable Parameters
343M
775M
350M
Loss Function
Mean-Squared Error
LlamaGen Training Loss 5 5 5 See the training losses paragraph in Section 2.2 of [ 40 ] for details.
Softmax Cross-Entropy
Compute Resources
64-128 v5p TPUs
1 H100 GPU
32 v5p TPUs
Finetuning Speed ↑
7.14 steps/s
0.29 steps/s
7.78 steps/s
DP-Finetuning Speed ↑
0.20 steps/s
0.04 steps/s
8.46 steps/s
Appendix
Table 3: Summary of the experimental setup.
DP-LDM
DP-LlamaGen
DP-Wavelet
Max LR for ε=∞
1e\mbox−5
2e\mbox−5
2e\mbox−5
Max LR for ε<∞
1e\mbox−4
2e\mbox−5
2e\mbox−5
# steps for ε=∞
3,230
4,500
100,000
# steps for ε<∞
3,230
1,125
100,000
# LR warmup steps
100
500
500
Batch size 6 6 6 DP-LlamaGen used a batch size of 256 for non-private finetuning.
256
1024
1024
Appendix
Table 4: MS-COCO hyperparameters.
DP-LDM
DP-LlamaGen
DP-Wavelet
Max LR for ε=∞
5e\mbox−5
2e\mbox−5
2e\mbox−5
Max LR for ε<∞
2e\mbox−5
2e\mbox−5
2e\mbox−5
# steps for ε=∞
4,680
1,125
100,000
# steps for ε<∞
4,680
1,125
100,000
# LR warmup steps
100
500
500
Batch size
1024
1024
1024
Appendix
Table 5: MS-COCO hyperparameters
Figure 6: MS-COCO and MM-CelebA-HQ sample images generated by T^ in stage 2 of DP-Wavelet. These are the decoded images formed by {t^i} when passing them through the decoder D .
Figure 7: MS-COCO and MM-CelebA-HQ sample images generated by DP-Wavelet. Left . Original image, Center . Final generated image, Right . Low resolution intermediary.