中文

认证VAE训练所需的支持:只需支持即可

机器学习 2025-04-29 v2 机器学习

摘要

变分自编码器(VAEs)在安全关键领域越来越受欢迎。在此类应用中,我们希望在对抗攻击下对性能提供认证的概率性保证。我们提出了一种新方法CIVET,用于对VAEs进行认证训练。CIVET依赖于我们可以通过对潜在层上精选选择的支持集上的错误上界来界定最坏情况VAE误差的关键见解。我们对此进行数学证明,并提出一种新训练算法。我们在广泛的评估中展示了该方法在不同数据集(涵盖无线和视觉应用领域)、不同架构和不同扰动幅度下均优于SOTA方法,实现了良好的常规性能与强大的鲁棒性保证。

关键词

引用

@article{arxiv.2504.11831,
  title  = {Support is All You Need for Certified VAE Training},
  author = {Changming Xu and Debangshu Banerjee and Deepak Vasisht and Gagandeep Singh},
  journal= {arXiv preprint arXiv:2504.11831},
  year   = {2025}
}

备注

21 pages, 3 figures, ICLR '25