Paper 2025/1315

CostSphere: A Cost Model-Driven Privacy-Preserving Machine Learning Framework with Network Context Adaptation

Yuntian Chen
Zhanyong Tang
Tianpei Lu
Bingsheng Zhang
Zhiying Shi
Zhiyuan Ning
Abstract

Privacy-preserving machine learning (PPML) is critical for protecting sensitive data in domains like healthcare, finance, and recommendation systems. Fully Homomorphic Encryption (FHE) and Secure Multi-Party Computation (MPC) are key enablers of secure computation, yet existing hybrid approaches often suffer from fixed protocol assignments, resulting in inefficiencies across diverse network environments, such as LANs and WANs. To address this, we introduce CostSphere, a cost-model-driven framework that dynamically assigns FHE and MPC protocols to optimize computational efficiency under varying network conditions. Utilizing a predictive cost model based on MLIR’s TOSA-level dialect and an ILP-based solver, CostSphere ensures robust performance for Transformer-based models. Experimental results demonstrate that CostSphere delivers $6.68\times$ to $12.92\times$ improvements in inference runtime compared to state-of-the-art solutions like BumbleBee (NDSS ’25), enabling scalable and network-agnostic PPML across diverse computational scenarios.

Metadata
Available format(s)
PDF
Category
Applications
Publication info
Preprint.
Contact author(s)
chenyt_x @ 163 com
zytang @ nwu edu cn
lutianpei @ zju edu cn
bingsheng @ zju edu cn
shizhiying @ stumail nwu edu cn
ningzhiyuan @ stumail nwu edu cn
History
2025-07-19: approved
2025-07-18: received
See all versions
Short URL
https://ia.cr/2025/1315
License
Creative Commons Attribution
CC BY

BibTeX

@misc{cryptoeprint:2025/1315,
      author = {Yuntian Chen and Zhanyong Tang and Tianpei Lu and Bingsheng Zhang and Zhiying Shi and Zhiyuan Ning},
      title = {{CostSphere}: A Cost Model-Driven Privacy-Preserving Machine Learning Framework with Network Context Adaptation},
      howpublished = {Cryptology {ePrint} Archive, Paper 2025/1315},
      year = {2025},
      url = {https://eprint.iacr.org/2025/1315}
}
Note: In order to protect the privacy of readers, eprint.iacr.org does not use cookies or embedded third party content.