让大模型在边缘安全运行,兼顾速度与隐私。
TEE-X: TEE-aware Acceleration Framework for Large Vision Models at the Edge

- 通过感知敏感度的模块化设计,优化模型在可信环境中的执行
- 在Jetson AGX Xavier上实现接近GPU的推理延迟
- 适合对安全和实时性要求高的边缘视觉应用
尽管机器学习模型在视觉应用中表现卓越,但面临严重安全威胁。白盒与黑盒攻击模型的差异表明,受限于模型信息访问时,黑盒攻击效果受限。使用可信执行环境(TEEs)可保护模型机密性和执行完整性,将执行环境从白盒转向黑盒,显著提升安全性。然而,在边缘设备上部署大型视觉模型(如视觉变换器ViTs)时,内存限制和计算延迟仍是挑战,尤其在对安全与实时性要求高的场景下。本文提出TEE-X,一种面向TEEs的加速框架,引入感知敏感度的模块化技术,并支持向量化推理。该框架在Arm TrustZone的OP-TEE环境下实现,针对NVIDIA Jetson AGX Xavier进行优化,用于高效边缘视觉应用。实验结果表明,TEE-X在保证性能的同时,实现了最小的精度-延迟权衡,为视觉模型提供快速且安全的边缘推理能力。
原文摘要 · Abstract (English)
Despite their remarkable success, machine learning models, particularly in vision applications, are alarmingly vulnerable to a range of security threats. One key factor in the attack landscape is the distinction between white-box and black-box threat models, as the latter poses challenges that limit attack effectiveness when access to model information is limited. As a result, using Trusted Execution Environments (TEEs) enhances security for machine learning applications by protecting model confidentiality and execution integrity, effectively shifting the execution environment from the white-box to the black-box side of the threat model spectrum. While adopting TEEs for large vision models, e.g., Vision Transformers (ViTs), is crucial for enhancing security and privacy, significant challenges related to memory constraints and increased computational latency must be addressed, especially in time-sensitive edge applications where safety and privacy are paramount. The objective of this work is to enable large vision models to be fully hosted within TEEs, achieving GPU-level inference latency for time-sensitive edge vision applications while maintaining performance. To this end, we propose TEE-X, a TEE-aware acceleration framework that introduces a sensitivity-aware modularization technique and enables vectorization in TEE inference. This design is validated on OP-TEE for Arm TrustZone, configured to optimize performance on the NVIDIA Jetson AGX Xavier for efficient edge vision applications using ViT models. The findings reveal that TEE-X delivers an effective TEE-aware acceleration framework that achieves minimal accuracy-latency trade-offs while ensuring fast and secure edge inference for vision models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。