English

Chordal Sparsity for Lipschitz Constant Estimation of Deep Neural Networks

Machine Learning 2024-01-09 v2

Abstract

Lipschitz constants of neural networks allow for guarantees of robustness in image classification, safety in controller design, and generalizability beyond the training data. As calculating Lipschitz constants is NP-hard, techniques for estimating Lipschitz constants must navigate the trade-off between scalability and accuracy. In this work, we significantly push the scalability frontier of a semidefinite programming technique known as LipSDP while achieving zero accuracy loss. We first show that LipSDP has chordal sparsity, which allows us to derive a chordally sparse formulation that we call Chordal-LipSDP. The key benefit is that the main computational bottleneck of LipSDP, a large semidefinite constraint, is now decomposed into an equivalent collection of smaller ones: allowing Chordal-LipSDP to outperform LipSDP particularly as the network depth grows. Moreover, our formulation uses a tunable sparsity parameter that enables one to gain tighter estimates without incurring a significant computational cost. We illustrate the scalability of our approach through extensive numerical experiments.

Keywords

Cite

@article{arxiv.2204.00846,
  title  = {Chordal Sparsity for Lipschitz Constant Estimation of Deep Neural Networks},
  author = {Anton Xue and Lars Lindemann and Alexander Robey and Hamed Hassani and George J. Pappas and Rajeev Alur},
  journal= {arXiv preprint arXiv:2204.00846},
  year   = {2024}
}