mnist

MNIST With FHE

Running the application on your machine

In this directory, ie mnist, you can do the following steps.

Do once

• First, create a virtual env and activate it:

python3.9 -m venv .venv
source .venv/bin/activate

• Then, install required packages:

pip3 install -U pip wheel setuptools --ignore-installed
pip3 install -r requirements.txt --ignore-installed

• Then, create a directory for check_points

mkdir -p .checkpoints

Do each time you relaunch the application

source .venv/bin/activate
python mnist_in_fhe.py

Remark that you may want to change some options in mnist_in_fhe.py, and notably:

• epochs: the number of epoch, default is 1
• sparsity: the sparsity of the model, default is 4
• quantization_bits: the number of bits for quantization, default is 2
• do_test_in_fhe: whether we perform tests in FHE, default is True
• do_training: whether we do the training, default is True