Skip to main content

Package for the training, pruning and verification of neural networks.

Project description

pyNeVer

Neural networks Verifier (NeVer 2) is a tool for the training, pruning and verification of neural networks. At present it supports sequential fully connected neural networks with ReLU and Sigmoid activation functions. pyNeVer is the corresponding python package providing all the main capabilities of the NeVer 2 tool and can be easily installed using pip. The PyPI project page can be found at https://pypi.org/project/pyNeVer/ whereas the github repository can be found at https://github.com/NeVerTools/pyNeVer.

REQUIREMENTS AND INSTALLATION

pyNeVer depends on several packages, which should be installed automatically. The packages required for the correct execution are the following:

  • numpy
  • ortools
  • onnx
  • torch
  • torchvision
  • pysmt

All the above packages are available via pip.

To install pyNeVer, run the command:

pip install pynever

To run some examples, further packages may be required. If an example requires a specific package, it will be detailed in the example directory.

DOCUMENTATION

The documentation related to the pyNeVer package can be found in the directory docs/pynever/ as html files.

SUPPORTED INPUTS

At present the pyNeVer package supports only the abstraction and verification of fully connected neural networks with ReLU and Sigmoid activation functions. The training, pruning and conversion supports also batch normalization layers. A network with batchnorm layers following fully connected layers can be converted to a "pure" fully connected neural networks using the capabilities provided in the utilities.py module.
The conversion.py provides the capabilities for the conversion of PyTorch and ONNX networks: therefore this kind of networks can be loaded using the respective frameworks and then converted to the internal representation used by pyNeVer. The properties for the verification and abstraction of the networks must be defined either in python code following the specification which can be found in the documentation, or via an SMT-LIB file compliant to the VNN-LIB standard. Examples of the python specification of the properties can be found in all the scripts in the directory examples/submissions/ATVA2021/.

EXAMPLES

NB: All the scripts should be executed INSIDE the related directory!

  • The directory examples/ contains some examples of application of the pyNeVer package. In particular the jupyter notebook shows a graphical example of the application of the abstraction module for the reachability of a small network with bi-dimensional input and outputs.

  • The pruning_example.py script show how to train and prune some small fully connected neural networks with relu activation function. It also show how it is possible to combine batch norm layer and fully connected layers to make the networks compliant with the requirements of the verification and abstraction modules.

  • The directory examples/submissions/ATVA2021 contains the experimental setup used for the experimental evaluation in our ATVA2021 paper. The experiments can be easily replicated by executing the python scripts acas_experiment.py from within the ATVA2021/ directory. The log files will be generated and will be saved in the logs/ directory.

CONTRIBUTORS

The main contributors of pyNeVer are Dario Guidotti and Armando Tacchella, further contributions are provided by Stefano Demarchi.

Students contributions:

  • Alessandro Drago - TensorFlow conversion

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pyNeVer-0.0.2a8.tar.gz (52.9 kB view hashes)

Uploaded Source

Built Distribution

pyNeVer-0.0.2a8-py3-none-any.whl (58.3 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page