Performance Acceleration of Secure Machine Learning Computations for Edge Applications

Zi Jie Lin, Chuan Chi Wang, Chia Heng Tu, Shih Hao Hung

研究成果: Conference contribution

1 引文 斯高帕斯(Scopus)

摘要

Edge appliances built with machine learning applications have been gradually adopted in a wide variety of application fields, such as intelligent transportation, the banking industry, and medical diagnosis. Privacy-preserving computation approaches can be used on smart appliances in order to secure the privacy of sensitive data, including application data and the parameters of machine learning models. Nevertheless, the data privacy is achieved at the cost of execution time. That is, the execution speed of a secure machine learning application is several orders of magnitude slower than that of the application in plaintext. Especially, the performance gap is enlarged for edge appliances. In this work, in order to improve the execution efficiency of secure applications, an open-source software framework CrypTen is targeted, which is widely used for building secure machine learning applications using the Secure Multi-Party Computation (SMPC) based privacy-preserving computation approach. We analyze the performance characteristics of the secure machine learning applications built with CrypTen, and the analysis reveals that the communication overhead hinders the execution of the secure applications. To tackle the issue, a communication library, OpenMPI, is added to the CrypTen framework as a new communication backend to boost the application performance by up to 50%. We further develop a hybrid communication scheme by combining the OpenMPI backend with the original communication backend with the CrypTen framework. The experimental results show that the enhanced CrypTen framework is able to provide better performance for the small-size data (LeNet5 on MNIST dataset by up to 50% of speedup) and maintain similar performance for large-size data (AlexNet on CIFAR-10), compared to the original CrypTen framework.

原文English
主出版物標題Proceedings - 2022 IEEE 28th International Conference on Embedded and Real-Time Computing Systems and Applications, RTCSA 2022
發行者Institute of Electrical and Electronics Engineers Inc.
頁面138-147
頁數10
ISBN(電子)9781665453448
DOIs
出版狀態Published - 2022
事件28th IEEE International Conference on Embedded and Real-Time Computing Systems and Applications, RTCSA 2022 - Taipei, Taiwan
持續時間: 2022 8月 232022 8月 25

出版系列

名字Proceedings - 2022 IEEE 28th International Conference on Embedded and Real-Time Computing Systems and Applications, RTCSA 2022

Conference

Conference28th IEEE International Conference on Embedded and Real-Time Computing Systems and Applications, RTCSA 2022
國家/地區Taiwan
城市Taipei
期間22-08-2322-08-25

All Science Journal Classification (ASJC) codes

  • 電腦網路與通信
  • 電腦科學應用
  • 資訊系統與管理
  • 控制和優化

指紋

深入研究「Performance Acceleration of Secure Machine Learning Computations for Edge Applications」主題。共同形成了獨特的指紋。

引用此