PyTorch model encryption using CrypTen

CrypTen is a framework for privacy-preserving Machine Learning developed by Facebook AI Research.

Requirements

• python >= 3.7
• torch >= 1.7.0
• crypten == 0.4.0

Tutorial

notebooks/crypten-pytorch.ipynb presents a brief tutorial on PyTorch model encryption using CrypTen. The tutorial assumes the existence of a pre-trained model in the models directory in the form of a pt/pth file.

References

• Knott, Brian, et al. "Crypten: Secure multi-party computation meets machine learning." Advances in Neural Information Processing Systems 34 (2021)
• facebookresearch/CrypTen/tutorials/Tutorial_4_Classification_with_Encrypted_Neural_Networks.ipynb
• Crypten: A new research tool for secure machine learning with PyTorch