Proceedings

Truth Table Net: Scalable, Compact & Verifiable Neural Networks with a Dual Convolutional Small Boolean Circuit Networks Form

Download BibTeX Cite

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence. Main Track. Pages 13-21.

https://doi.org/10.24963/ijcai.2024/2

Abstract

We introduce "Truth Table net"' (TTnet), a novel Deep Neural Network (DNN) architecture designed to provide excellent scalability/compactness trade-offs among DNNs, allowing in turn to tackle the DNN challenge of fast formal verification. TTnet is constructed using Learning Truth Table (LTT) filters, analogous to how a Deep Convolutional Neural Network (DCNN) is built upon convolutional filters. The differentiable LTT filters are unique by their dual form: they are both a neural network-based function and a small-sized truth table that can be computed within a practical time frame. This characteristic guarantees, by design and independently of the overall architecture, the ability to practically extract an efficient (in terms of the number of logical gates) and functionally equivalent Conjunctive Normal Form (CNF) Boolean logic gate implementation. This CNF circuit is even optimal when the LTT truth table's input bit size n < 12. In particular, TTnet architecture is the first differentiable DNN with as dual form a compact logic gate representation that can scale to datasets larger than CIFAR-10: we achieve an accuracy of 41% on the ImageNet dataset while ensuring that each LTT filter truth table is fully computable within 2^{16} operations. We further compare the compactness and scalability performances of TTnet Boolean logic circuit representation to state-of-the-art differentiable logic DNNs across tabular, MNIST, and CIFAR-10 datasets. We emphasize that TTnet is the first solution to the open problem of designing differentiable convolutional neural networks with an exact dual logic gate circuit representation, bridging the gap between symbolic AI and trainable DCNNs. Finally, as improving DNNs compactness in Boolean logic circuit form reduces the complexity of their formal verification, we demonstrate TTnet effectiveness in exact sound and complete formal verification. Notably, our model achieves robustness verification in 10ms vs 100s for traditional state-of-the-art DNNs solvers.

  1. [Agarwal et al., 2021] Rishabh Agarwal, Levi Melnick, Nicholas Frosst, Xuezhou Zhang, Ben Lengerich, Rich Caruana, and Geoffrey E Hinton. Neural additive models: Interpretable machine learning with neural nets. Advances in Neural Information Processing Systems (NeurIPS), 2021.
  2. [AI, 2023] NIST AI. Artificial Intelligence Risk Management Framework (AI RMF 1.0). 2023.
  3. [Arora and Barak, 2009] Sanjeev Arora and Boaz Barak. Computational complexity: a modern approach. Cam- bridge University Press, 2009.
  4. [Bello et al., 2021] Irwan Bello, William Fedus, Xianzhi Du, Ekin Dogus Cubuk, Aravind Srinivas, Tsung-Yi Lin, Jonathon Shlens, and Barret Zoph. Revisiting resnets: Im- proved training and scaling strategies. Advances in Neural Information Processing Systems (NeurIPS), 2021.
  5. [Blake, 1938] Archie Blake. Corrections to Canonical ex- pressions in Boolean algebra. Journal of Symbolic Logic, 1938.
  6. [Brendel and Bethge, 2018] Wieland Brendel and Matthias Bethge. Approximating CNNs with Bag-of-local-Features models works surprisingly well on ImageNet. In Inter- national Conference on Learning Representations (ICLR), 2018.
  7. [Brix et al., 2023] Christopher Brix, Mark Niklas Müller, Stanley Bak, Taylor T Johnson, and Changliu Liu. First three years of the international verification of neural net- works competition (VNN-COMP). International Journal on Software Tools for Technology Transfer, 2023.
  8. [Chatterjee, 2018] Satrajit Chatterjee. Learning and memo- rization. In International conference on machine learning (ICML), 2018.
  9. [Daemen and Rijmen, 2002] Joan Daemen and Vincent Rij- men. The Design of Rijndael: AES -The Advanced Encryp- tion Standard. Springer, 2002.
  10. [Driscoll, 2020] Michael Driscoll. System and method for adapting a neural network model on a hardware platform, 2020. US Patent App. 16/728,884.
  11. [Goodfellow et al., 2016] Ian Goodfellow, Yoshua Bengio, and Aaron Courville. Deep learning. MIT Press, 2016.
  12. [Han et al., 2016] Song Han, Huizi Mao, and William J Dally. Deep compression: Compressing deep neural networks with pruning, trained quantization and huffman coding. In International Conference on Learning Representations (ICLR), 2016.
  13. [Hirtzlin et al., 2019] Tifenn Hirtzlin, Bogdan Penkovsky, Marc Bocquet, Jacques-Olivier Klein, Jean-Michel Portal, and Damien Querlioz. Stochastic computing for hardware implementation of binarized neural networks. IEEE Access, 2019.
  14. [Hubara et al., 2016] Itay Hubara, Matthieu Courbariaux, Daniel Soudry, Ran El-Yaniv, and Yoshua Bengio. Bi- narized neural networks. Advances in Neural Information Processing Systems (NeurIPS), 2016.
  15. [Ignatiev et al., 2018] Alexey Ignatiev, Antonio Morgado, and Joao Marques-Silva. PySAT: A Python Toolkit for Prototyping with SAT Oracles. In International Conference on Theory and Applications of Satisfiability Testing (SAT), 2018.
  16. [Jia and Rinard, 2020] Kai Jia and Martin Rinard. Efficient exact verification of binarized neural networks. Advances in Neural Information Processing Systems (NeurIPS), 2020.
  17. [Jia and Rinard, 2021] Kai Jia and Martin Rinard. Exploiting verified neural networks via floating point numerical error. In International Static Analysis Symposium, 2021.
  18. [Klir et al., 1997] George J Klir, Ute St. Clair, and Bo Yuan. Fuzzy set theory: foundations and applications. Prentice- Hall, Inc, 1997.
  19. [Liffiton and Maglalang, 2012] Mark H Liffiton and Jor- dyn C Maglalang. A cardinality solver: More expressive constraints for free. In International Conference on Theory and Applications of Satisfiability Testing (SAT), 2012.
  20. [Liu et al., 2018] Zechun Liu, Baoyuan Wu, Wenhan Luo, Xin Yang, Wei Liu, and Kwang-Ting Cheng. Bi-real net: Enhancing the performance of 1-bit cnns with improved representational capability and advanced training algorithm. Proceedings of the European Conference on Computer Vi- sion (ECCV), 2018.
  21. [Mocanu et al., 2018] Decebal Constantin Mocanu, Elena Mocanu, Peter Stone, Phuong H Nguyen, Madeleine Gibescu, and Antonio Liotta. Scalable training of artificial neural networks with adaptive sparse connectivity inspired by network science. Nature communications, 2018.
  22. [Molchanov et al., 2017] Dmitry Molchanov, Arsenii Ashukha, and Dmitry Vetrov. Variational dropout sparsifies deep neural networks. In International conference on machine learning (ICML), 2017.
  23. [Müller et al., 2022] Mark Niklas Müller, Gleb Makarchuk, Gagandeep Singh, Markus Püschel, and Martin Vechev. PRIMA: general and precise neural network certification via scalable convex hull approximations. Proceedings of the ACM on Programming Languages, 2022.
  24. [Narodytska et al., 2019] Nina Narodytska, Hongce Zhang, Aarti Gupta, and Toby Walsh. In search for a SAT-friendly binarized neural network architecture. In International Conference on Learning Representations (ICLR), 2019.
  25. [Paszke et al., 2019] Adam Paszke, Sam Gross, Francisco Massa, Adam Lerer, James Bradbury, Gregory Chanan, Trevor Killeen, Zeming Lin, Natalia Gimelshein, Luca Antiga, et al. Pytorch: An imperative style, high- performance deep learning library. Advances in Neural Information Processing Systems (NeurIPS), 2019.
  26. [Petersen et al., 2022] Felix Petersen, Christian Borgelt, Hilde Kuehne, and Oliver Deussen. Deep differentiable logic gate networks. Advances in Neural Information Pro- cessing Systems (NeurIPS), 2022.
  27. [Rastegari et al., 2016] Mohammad Rastegari, Vicente Or- donez, Joseph Redmon, and Ali Farhadi. Xnor-net: Ima- genet classification using binary convolutional neural net- works. Proceedings of the European Conference on Com- puter Vision (ECCV), 2016.
  28. [Roussel and Manquinho, 2009] Olivier Roussel and Vasco Manquinho. Pseudo-Boolean and cardinality constraints. In Handbook of satisfiability. IOS Press, 2009.
  29. [Rumelhart et al., 1986] David E Rumelhart, Geoffrey E Hin- ton, and Ronald J Williams. Learning representations by back-propagating errors. nature, 1986. [Udovenko, 2023] Aleksei Udovenko. DenseQMC: an ef- ficient bit-slice implementation of the Quine-McCluskey algorithm. arXiv, 2023.
  30. [Umuroglu et al., 2017] Yaman Umuroglu, Nicholas J Fraser, Giulio Gambardella, Michaela Blott, Philip Leong, Magnus Jahre, and Kees Vissers. Finn: A framework for fast, scal- able binarized neural network inference. In Proceedings of the ACM/SIGDA International Symposium On Field- Programmable Gate Arrays, 2017.
  31. [Van Der Walt et al., 2011] Stefan Van Der Walt, S Chris Col- bert, and Gael Varoquaux. The NumPy array: a structure for efficient numerical computation. Computing in Science & Engineering, 2011.
  32. [Wang et al., 2019] Erwei Wang, James J Davis, Peter YK Cheung, and George A Constantinides. LUTNet: Rethink- ing inference in FPGA soft logic. In Annual International Symposium on Field-Programmable Custom Computing Machines (FCCM), 2019.
  33. [Wang et al., 2021] Shiqi Wang, Huan Zhang, Kaidi Xu, Xue Lin, Suman Jana, Cho-Jui Hsieh, and J Zico Kolter. Beta-crown: Efficient bound propagation with per-neuron split constraints for neural network robustness verifica- tion. Advances in Neural Information Processing Systems (NeurIPS), 2021.
  34. [Xu et al., 2020] Kaidi Xu, Huan Zhang, Shiqi Wang, Yihan Wang, Suman Jana, Xue Lin, and Cho-Jui Hsieh. Fast and Complete: Enabling Complete Neural Network Verification with Rapid and Massively Parallel Incomplete Verifiers. In International Conference on Learning Representations (ICLR), 2020.
  35. [Zeng and Urtasun, 2018] Wenyuan Zeng and Raquel Urta- sun. Mlprune: Multi-layer pruning for automated neural network compression. openreview, 2018.
  36. [Zhou et al., 2021] Xiao Zhou, Weizhong Zhang, Hang Xu, and Tong Zhang. Effective sparsification of neural networks with global sparsity constraint. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2021.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards