Robust Machine Learning
Python Mathematica
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
mnist
optimize
verify
README.md
readings.md

README.md

Verdict: Safe and Robust Machine Learning

Mission Statement

Machine learning has brought lots of excitement and potential for AI technologies to be beneficial for humanity. However, a series of challenges have to be addressed before AI/ML systems can be practically deployed and integrated into our eveyday life. Cutting-edge research challenges include performance, accuracy, scalability, energy-efficiency, privacy, security, etc. While we strongly support research in these areas, this project specifically focuses on the safty and robustness aspect of machine learning. That is, how do we make sure a machine learning-based system is robust against uncertainty and failure, and will not perform unintended and harmful behaviors? Think about the autopilot system in a self driving car, or the controller of a satellite, or more controversially predicting how likely someone is going to commit a crime.

Our mission is to build systems that enable safe and robust machine learning, thereby pushing machine learning systems one step closer toward reality. By systems, we mean both software systems (e.g., programming language, runtime/middleware, formal verifier) and hardware systems (e.g., customized accelerators, FPGAs).

State of the Project

As of now, this project focuses on the robustness aspect of machine learning models. We have created formal verfication tools that can verify the robustness of a given machine learning model against certain level of perturbation, as well as optimization tools that find the maximal perturbation a particular model can tolerate while still being robust. These tools will provide the necessary theoretical foundation for building a robust machine learning system in practice.

This is a living document of the project. It might be messy and is constantly being updated. More goals will be defined and more systems will be developed as we get a deeper understanding of the problem space.

What is a Robust Model

Informally, a robust machine learning model should produce same or very close predictions for inputs that are very similar to each other. For example, an image classification model should produce the same label for an image even if the image is slightly perturbated. Formally, we want to make sure:

\forany x, |x - x'| < \delta --> |f(x) - f(x')| < \theta

where \delta and \theta are parameters to control the perturbation and robustness, respectively. --> denotes the logic implication operator.

Note that this is just one (perhaps most intuitive) definition of robustness. There could be others. For example, one could argue that a robust model should also guarantee \forany x, |f(x) - f(x')| < \theta --> |x - x'| < \delta. That is, if two predictions are close enough, their corresponding inputs should not be too different. However, one could further argue that if a model can correctly label two seemingly completely different inputs as the same thing, then that model is robust because it is not easily fooled! See this paper for a concrete example. The exact definition of robustness can vary, and the verifier should be general enough to handle different cases.

Dependencies

  1. Z3 theorem prover. Follow its installation instructions. You will need it to run formal verification tools. Remember to build its Python binding.
  2. or-tools. For running optimization tools, which are based on linearing programming and constraints solving. Remember to build its Python binding.
  3. Tensorflow. For training machine learning models (running applications under mnist/ for example). Technically this is not a required dependency if you know how to train a machine learning model by yourself or in other languages.
  4. Python 2.7. We have not tested it on Python 3.x.

Directory Structure

  • mnist/ contains various tensorflow-based applications for constructing machine learning models for MNIST. The para/ subdirectory contains ML model parameters, which are inputs to verifiers and optimizers.
  • verify/ contains various formal verification tools that verify different aspects of robustness on different types of models. Read it's README for details.
  • optimize/ contains optimizers that find the maximal input perturbation that ML models can tolerate. Read it's README for details.

Readings

See the reading list page. There are a wide range of papers discussing the safty and robustness of machine learning in specific applications domains as well as techniques to address them. The list is undoutedly incomplete, and is getting constantly updated. Let me know if you know of a relavant paper!