PRISM: Gauge-Invariant Tangent-Space Differentially Private LoRA
Shihao Wang ⋅ Xueru Zhang
Abstract
Applying differential privacy (DP) via DP-SGD to Low-Rank Adaptation (LoRA) is a natural approach for privacy-preserving fine-tuning. However, LoRA's low-rank parameterization poses a fundamental challenge. In LoRA, each trainable update is represented as a low-rank matrix $Z = AB^\top$, but this factorization is inherently *non-identifiable*: many factor pairs $(A, B)$ represent the same update $Z$. As a result, applying DP-SGD directly to the factors induces *gauge-dependent* perturbations on $Z$, and we show that this naive DP-LoRA can lead to unbounded noise amplification. We propose **PRISM**, an intrinsic DP mechanism for LoRA that is gauge invariant by construction, avoids bilinear noise amplification, and admits an efficient low-dimensional noise sampler. Moreover, PRISM yields a closed-form characterization of the effective intrinsic noise induced on $Z$, enabling stable privacy–utility trade-offs through bounded, gauge-invariant perturbations. We establish standard $(\varepsilon,\delta)$-DP guarantees for PRISM and introduce a DP-aware, gauge-invariant adaptive update rule that prevents adaptive optimization from amplifying injected privacy noise, improving numerical stability in practice.
Lay Summary
Fine-tuning large language models on sensitive data can expose information about the training examples, so privacy protection is important. A popular efficient fine-tuning method called LoRA updates a model through two smaller matrices, but adding privacy noise directly to those matrices can make training unstable because the same model update can be written in many equivalent ways. PRISM adds the privacy noise to the actual low-rank model update instead, so equivalent representations behave the same. This leads to more stable private LoRA fine-tuning and a better balance between privacy and model accuracy in our experiments.
Successful Page Load