We introduce LLA, an effective intellectual property (IP) protection scheme for generative AI models. LLA leverages the synergy between hardware and software to defend against various supply chain threats, including model theft, model corruption, and information leakage. On the software side, it embeds key bits into neurons that can trigger outliers to degrade performance and applies invariance transformations to obscure the key values. On the hardware side, it integrates a lightweight locking module into the AI accelerator while maintaining compatibility with various dataflow patterns and toolchains. An accelerator with a pre-stored secret key acts as a license to access the model services provided by the IP owner. The evaluation results show that LLA can withstand a broad range of oracle-guided key optimization attacks, while incurring a minimal computational overhead of less than 0.1% for 7,168 key bits.
@article{arxiv.2512.22307,
title = {LLA: Enhancing Security and Privacy for Generative Models with Logic-Locked Accelerators},
author = {You Li and Guannan Zhao and Yuhao Ju and Yunqi He and Jie Gu and Hai Zhou},
journal= {arXiv preprint arXiv:2512.22307},
year = {2025}
}
Comments
Accepted by AAAI'26 as a conference paper and selected for oral presentation