Graph Reinforcement Learning for the MPC Protocol Mixing Problem
Master Thesis
Motivation
From enabling Danish farmers to participate in auctions without giving away their bids [1], to permitting the Estonian government to investigate systemic interdependencies while conforming to the law [2], multi party computation has enjoyed a fast-paced development in the last decades. Its core mechanism, the secure multiparty cryptographic protocol, allows parties to jointly compute a function without needing to trust each other [3]. As there exist different types of MPC protocols and each protocol has its own trade-offs, mixed-protocol MPC has been proposed, a concept where different protocols encrypt granular parts of an MPC program [4, 5]. Although mixed-protocol MPC drastically improves the computational efficiency of MPC [4, 5, 6], finding an optimal combination of more than two protocols is NP-hard [7]. To tackle this, hybrid MPCs compilers have been developed; however, they suffer from non-ideal cost models and simplifications in optimum search [6, 7, 8].
Goal
The goal of this thesis is to investigate protocol mixing can be formulated as a reinforcement learning problem [9, 10]. We explore a model-based, AlphaZero-like approach trained on runtimes, cost models from existing tools, and convex combinations of communication costs and runtimes on small graphs [6, 10, 11, 12, 13, 14]. We evaluate the resulting reinforcement learning approaches against modern MPC compilers using a unified MPC benchmark [6, 7].
Requirements
- Programming skills in Python, C++ or Rust
- At least basic knowledge of cryptography
- Familiarity with machine learning and reinforcement learning in any common Python framework
- High motivation + ability to work independently
- Knowledge of the English language, Git, LaTeX, etc.
References
- [1] P. Bogetoft, D. L. Christensen, I. Damgård, M. Geisler, T. Jakobsen, M. Krøigaard, J. D. Nielsen, J. B. Nielsen, K. Nielsen, J. Pagter, et al., “Secure multiparty computation goes live.” (PDF file) (opens in new tab) In Financial Cryptography and Data Security, 2009.
- [2] B. Kubo, “Students and taxes: a privacy-preserving study using secure computation.” (PDF file) (opens in new tab) In Proceedings on Privacy Enhancing Technologies, 2016.
- [3] D. Evans, V. Kolesnikov, and M. Rosulek, “A pragmatic introduction to secure multi-party computation.” In Foundations and Trends in Privacy and Security, 2018.
- [4] D. Demmler, T. Schneider, and M. Zohner, “ABY: A framework for efficient mixed-protocol secure two-party computation.” (PDF file) (opens in new tab) In NDSS, 2015.
- [5] P. Mohassel and P. Rindal, “ABY3: A mixed protocol framework for machine learning.” (PDF file) (opens in new tab) In CCS, 2018.
- [6] E. Chen, J. Zhu, A. Ozdemir, R. S. Wahby, F. Brown, and W. Zheng, “Silph: A framework for scalable and accurate generation of hybrid MPC protocols.” In IEEE Symposium on Security and Privacy, 2023.
- [7] M. Ishaq, A. L. Milanova, and V. Zikas, “Efficient MPC via program analysis: A framework for efficient optimal mixing.” In CCS, 2019.
- [8] F. Kerschbaum, T. Schneider, and A. Schröpfer, “Automatic protocol selection in secure two-party computations.” In ACNS, 2014.
- [9] R. S. Sutton and A. G. Barto, “Reinforcement Learning: An Introduction.” (PDF file) (opens in new tab) MIT Press, 1998.
- [10] Y. Bengio, A. Lodi, and A. Prouvost, “Machine learning for combinatorial optimization: a methodological tour d’horizon.” In European Journal of Operational Research, 2021.
- [11] V.-A. Darvariu, S. Hailes, and M. Musolesi, “Graph reinforcement learning for combinatorial optimization: A survey and unifying perspective.” arXiv preprint arXiv:2404.06492, 2024.
- [12] D. Silver, A. Huang, C. J. Maddison, A. Guez, L. Sifre, G. van den Driessche, J. Schrittwieser, I. Antonoglou, V. Panneershelvam, M. Lanctot, et al., “Mastering the game of Go with deep neural networks and tree search.” In Nature, 2016.
- [13] D. Silver, T. Hubert, J. Schrittwieser, I. Antonoglou, M. Lai, A. Guez, M. Lanctot, L. Sifre, D. Kumaran, T. Graepel, et al., “A general reinforcement learning algorithm that masters chess, shogi, and Go through self-play.” In Science, 2018.
- [14] J. Schrittwieser, I. Antonoglou, T. Hubert, K. Simonyan, L. Sifre, S. Schmitt, A. Guez, E. Lockhart, D. Hassabis, T. Graepel, et al., “Mastering Atari, Go, chess and shogi by planning with a learned model.” In Nature, 2020.
Supervisors
- Nora Khayata, M.Sc. ( khayata@encrypto.cs.tu-…)
- Moritz Huppert, M.Sc. ( moritz.huppert@tu-…)
- Prof. Dr.-Ing. Thomas Schneider
Core data