LLMscope: Extracting LLM Assets from Edge AI Chips via Optical Probing
Organizations: Worcester Polytechnic Institute
Abstract
The move of LLM inference to edge AI accelerators introduces new physical vulnerabilities. During execution, model parameters and intermediate inference states are repeatedly loaded into and processed on the chip, making them suscep- tible to physical side-channel attacks. In this work, by deploying laser voltage imaging, we show that one can extract LLM assets during inference, namely embeddings, attention, and quantized MLP weights, activations, and other inference states, from localized memories and compute subcircuits. To validate our claims, we perform an attack on an FPGA-based LLM accelerator. Since such accelerators reuse the same buffers and compute subcircuits across addresses, tiles, modules, and layers, reading asset values comes down to probing different memories during inference. We demonstrate full recovery of the targeted values; however, we also establish a methodology to recover asset values even if some weights or bits remain unread. We further derive lower bounds that relate imaging effort to asset dimensions and show that even direct recovery scales linearly with the size of the targeted asset
Figures & tables
| Design | Flip-flops | BRAM | URAM | Security-relevant implementation characteristic |
|---|---|---|---|---|
| LlamaF [ 43 , 25 ] | K | 223 | – | TinyLlama-scale embedded inference with quantized layer weights, scale factors, BRAM-cached activations, and pipelined matrix-vector computation. |
| Pushing up to the Limit [ 26 , 25 ] | K | 10 | LLaMA2-7B W4 inference on KV260 with a 4 GB external-memory budget, streamed weights, on-chip hidden state, and partial KV-cache storage. | |
| Hummingbird [ 25 ] | 25,422 | 59 | 18 | LLaMA3-8B embedded inference with a 512-bit weight bus, 4-bit model weights, 8-bit KV cache, and time-multiplexed on-chip KV-cache buffering. |
| FlightLLM [ 44 ] | 943K | 1,252 | 792 | U280 implementation with a large HBM-backed model, on-chip decode activations, and a buffer subsystem using 816 BRAMs and all 792 URAMs. |
LLM Security
Large language model (LLM) security studies attacks, privacy risks, and defenses across model training, inference, and deployment.
See all 322 papers