How to control the privacy budget

[TeDiou]

As we set the private = True, in your source code it only calculates the privacy budget. How can we control the privacy budget? By adding a if statement?

[zhao-zilong]

Hi @TeDiou

If you set private = True, then you enable the training with DP. And for calculate privacy budget, the code block is starting from here:

CTAB-GAN-Plus-DP/model/synthesizer/ctabgan_synthesizer.py

Line 581 in 6507b8a

# if self.private:

And from this line of code:

rdp = compute_rdp(self.micro_batch_size / train_data.shape[0], self.sigma, steps, lmbds)

You can see that to calculate RDP, the batch_size, dataset size, sigma and training steps are four features influencing the privacy budget.

then in the following line:

epsilon, _, _ = get_privacy_spent(lmbds, rdp, target_delta=1e-5)

Epsilon is the privacy budget, can you add an if in the beginning of the loop to control the training only if the epsilon is less than a certain value.

Hope that solves your question.

[TeDiou]

Thanks for your answer!

[TeDiou]

Sorry to bother u, why this dp-synthesizer.sample method is different from the ctabganplus.sample 。The two models differ only in a privacy module. However, in ctabganplusdp, the generation part requires multiple loops for generation.

[zhao-zilong]

Hi @TeDiou
Yeah, we need a loop to generate enough synthetic data, the reason is because we implemented a filter to filter out the invalid generation, so it takes more sampling than the required data number. Check this issue answer:
Team-TUD/CTAB-GAN-Plus#7 (comment)

[TeDiou]

I got that. Thanks a lot!_