DP-DocLDM: Differentially Private Document Image Generation using Latent Diffusion Models

TL;DR

This paper introduces DP-DocLDM, a method that uses differentially private latent diffusion models to generate synthetic document images, enhancing privacy and improving downstream classification performance.

Contribution

It proposes a novel combination of conditional latent diffusion models with differential privacy for synthetic document image generation, addressing privacy and performance issues.

Findings

Generates realistic, class-specific document images under strict privacy constraints.

Improves downstream classification accuracy on small datasets.

Effective across various document types and privacy levels.

Abstract

As deep learning-based, data-driven information extraction systems become increasingly integrated into modern document processing workflows, one primary concern is the risk of malicious leakage of sensitive private data from these systems. While some recent works have explored Differential Privacy (DP) to mitigate these privacy risks, DP-based training is known to cause significant performance degradation and impose several limitations on standard training procedures, making its direct application to downstream tasks both difficult and costly. In this work, we aim to address the above challenges within the context of document image classification by substituting real private data with a synthetic counterpart. In particular, we propose to use conditional latent diffusion models (LDMs) in combination with differential privacy (DP) to generate class-specific synthetic document images under strict privacy constraints, which can then be utilized to train a downstream classifier following standard training procedures. We investigate our approach under various pretraining setups, including unconditional, class-conditional, and layout-conditional pretraining, in combination with multiple private training strategies such as class-conditional and per-label private fine-tuning with DPDM and DP-Promise algorithms. Additionally, we evaluate it on two well-known document benchmark datasets, RVL-CDIP and Tobacco3482, and show that it can generate useful and realistic document samples across various document types and privacy levels ($\varepsilon \in \{1, 5, 10\}$). Lastly, we show that our approach achieves substantial performance improvements in downstream evaluations on small-scale datasets, compared to the direct application of DP-Adam.