为手机端模型提供自动知识产权保护,防盗版且无需开发者懂机器学习。
THEMIS: Towards Practical Intellectual Property Protection for Post-Deployment On-Device Deep Learning Models
- 通过重构可写模型,突破手机端只读限制,实现水印嵌入。
- 在403个真实应用中成功率达81.14%,跨模型结构验证有效。
- 无需训练、不依赖开发者技术,适合应用商店等第三方部署。
手机端深度学习(DL)在移动应用中广泛应用,具备离线推理和保护用户隐私的优势。然而,模型存储于用户设备中,面临模型窃取和知识产权侵犯风险。尽管可信执行环境(TEEs)提供强防护,但其在支持第三方模型和实际移动端部署中仍存在扩展性难题。当前,水印是常见防御手段,但多数开发者缺乏机器学习知识,且手机端模型的只读与仅推理特性使第三方(如应用商店)难以实施现有水印方案。为此,本文提出THEMIS,一种自动化工具,通过重建可写模型突破只读限制,并利用模型不可训练特性求解水印参数,从而保护模型所有者知识产权。在多种数据集与模型结构上的实验表明,THEMIS在多项指标上表现优异。进一步对谷歌应用商店中403个真实移动应用的实证研究显示,成功率高达81.14%,验证了其实际可行性。
原文摘要 · Abstract (English)
On-device deep learning (DL) has rapidly gained adoption in mobile apps, offering the benefits of offline model inference and user privacy preservation over cloud-based approaches. However, it inevitably stores models on user devices, introducing new vulnerabilities, particularly model-stealing attacks and intellectual property infringement. While system-level protections like Trusted Execution Environments (TEEs) provide a robust solution, practical challenges remain in achieving scalable on-device DL model protection, including complexities in supporting third-party models and limited adoption in current mobile solutions. Advancements in TEE-enabled hardware, such as NVIDIA's GPU-based TEEs, may address these obstacles in the future. Currently, watermarking serves as a common defense against model theft but also faces challenges here as many mobile app developers lack corresponding machine learning expertise and the inherent read-only and inference-only nature of on-device DL models prevents third parties like app stores from implementing existing watermarking techniques in post-deployment models. To protect the intellectual property of on-device DL models, in this paper, we propose THEMIS, an automatic tool that lifts the read-only restriction of on-device DL models by reconstructing their writable counterparts and leverages the untrainable nature of on-device DL models to solve watermark parameters and protect the model owner's intellectual property. Extensive experimental results across various datasets and model structures show the superiority of THEMIS in terms of different metrics. Further, an empirical investigation of 403 real-world DL mobile apps from Google Play is performed with a success rate of 81.14%, showing the practicality of THEMIS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。