Skip to content
InfoResearchPeer-reviewed

Gem5-Shark: Smaug-based Heterogeneous Deep Learning Accelerator Security and Robustness Evaluation and Characterization Framework

Published
Record updated
View JSON

Summary

Gem5-Shark is a cycle-level simulation framework, built on Gem5-Aladdin's Smaug, for evaluating security threats to heterogeneous deep learning accelerators before deployment. It integrates eight threat vectors across dataset, compilation, SoC communication and accelerator execution stages, including two new compilation-stage attacks, Operator Attack (OPAtt) and Tiling Attack (TAtt). Across Minerva, LeNet-5, VGG-16 and pretrained LLMs up to 8B parameters, the attacks degrade accuracy by up to 97.7%, raise latency by up to 84× and implant backdoors with as few as 20 poisoned fine-tuning samples.

Mitigation

Two defenses are proposed: a compiler-stage tiling range check and a host-side finish-flag watchdog, each removing over 99% of the injected overhead on CNNs and LLMs.