arXiv · 2504.11831
Support is All You Need for Certified VAE Training
Abstract
Variational Autoencoders (VAEs) have become increasingly popular and deployed in safety-critical applications. In such applications, we want to give certified probabilistic guarantees on performance under adversarial attacks. We propose a novel method, CIVET, for certified training of VAEs. CIVET depends on the key insight that we can bound worst-case VAE error by bounding the error on carefully chosen support sets at the latent layer. We show this point mathematically and present a novel training algorithm utilizing this insight. We show in an extensive evaluation across different datasets (in both the wireless and vision application areas), architectures, and perturbation magnitudes that our method outperforms SOTA methods achieving good standard performance with strong robustness guarantees.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Changming Xu, Debangshu Banerjee, Deepak Vasisht, Gagandeep Singh. 2025-04-16. Support is All You Need for Certified VAE Training. https://arxiv.org/abs/2504.11831
Cite the original work for its findings. Save a collection to share your selection of sources.