Skip to content

Loading Models, Launching Shells: Abusing AI File Formats for Code Execution

C Parzian

C Parzian analyzes weaponized machine learning file formats, demonstrating arbitrary code execution vectors in widely deployed AI frameworks.

Summary

The Blind Trust in AI Model Weights

C Parzian investigates the rampant assumption of trust when data scientists and developers download pretrained neural network model weights from open-source repositories and public hubs.

Dissecting Dangerous Model Formats

The presentation uncovers unsafe deserialization vulnerabilities, embedded script execution, and parser flaws across popular serialization formats including Pickle, PyTorch checkpoints, and custom tensor structures.

Exploitation Scenarios and Defensive Hardening

Parzian demonstrates realistic supply chain attack chains executing arbitrary shells upon model load and presents best practices for safe loading using formats like SafeTensors and isolated sandboxing.

Related talks