GuaranTEE: Towards Attestable and Private ML with CCA

Sandra Siby,Sina Abdollahi,Mohammad Maheri,Marios Kogias,Hamed Haddadi
2024-03-30
Abstract:Machine-learning (ML) models are increasingly being deployed on edge devices to provide a variety of services. However, their deployment is accompanied by challenges in model privacy and auditability. Model providers want to ensure that (i) their proprietary models are not exposed to third parties; and (ii) be able to get attestations that their genuine models are operating on edge devices in accordance with the service agreement with the user. Existing measures to address these challenges have been hindered by issues such as high overheads and limited capability (processing/secure memory) on edge devices.
Cryptography and Security
What problem does this paper attempt to address?
### What problems does this paper attempt to solve? This paper aims to solve the challenges of **model privacy** and **verifiability** encountered when deploying machine - learning (ML) models on edge devices (such as smart phones, Internet of Things gateways and home routers). Specifically, the paper focuses on the following two main issues: 1. **Model privacy protection**: - Model providers hope to ensure that their proprietary models are not exposed to third parties, including other potential competitors, end - users or malicious actors. This means that the weights of the model and other sensitive information need to be protected to prevent unauthorized access. 2. **Model verifiability and proof**: - Model providers hope to be able to obtain proof to ensure that their real models run correctly on edge devices according to the service agreement with users and have not been tampered with. This involves the verification of the model execution environment to ensure its integrity. Existing solutions face some limitations in dealing with these problems, such as high overhead and insufficient limited capabilities (processing power and secure memory) on edge devices. Therefore, this paper proposes a new framework - **GuaranTEE**, which utilizes Arm's latest architecture extension - **Confidential Computing Architecture (CCA)** to create a dynamic hardware - protected environment (called "realm"), thereby achieving verifiable private machine learning. In this way, GuaranTEE not only solves the problems of model privacy and verifiability, but also improves the security and reliability of ML models on edge devices.