Critical Bugs Put Hugging Face AI Platform in a ‘Pickle’

Critical Bugs Put Hugging Face AI Platform in a 'Pickle'

April 5, 2024 at 04:51PM

Two critical security vulnerabilities in the Hugging Face AI platform allowed attackers to access customer data and overwrite images in a shared container registry. Researchers at Wiz found weaknesses in Hugging Face’s Inference API, Endpoints, and Spaces. The vulnerabilities were exploited by uploading a Pickle-based model. Hugging Face has since mitigated the risks. Wiz recommended organizations to assess risks and governance in AI-as-a-service environments. Schwake highlighted the risks of harmful AI models and data poisoning. He suggested exploring Explainable AI to make AI models more comprehensible and mitigate bias or risk.

From the provided meeting notes, the following key takeaways are:

1. Two critical security vulnerabilities were discovered in the Hugging Face AI platform, allowing attackers to access and alter customer data and models.
2. The vulnerabilities included weaknesses in the Inference API, Inference Endpoints, and Hugging Face Spaces, leading to potential unauthorized access and tampering with customer data and models.
3. The use of Pickle files was identified as a significant security risk, enabling malicious code execution and vulnerabilities.
4. Wiz emphasized the need for organizations to be aware of the risks associated with shared infrastructure for hosting AI, recommending similar mitigations as those used for public cloud environments.
5. Eric Schwake highlighted the potential threats related to AI-as-a-service, including the upload of harmful AI models, exploitation of vulnerabilities, and data poisoning.
6. It was suggested that understanding the interactions of AI apps and models with API and exploring Explainable AI (XAI) could help mitigate risks associated with AI models.

These takeaways underline the critical need for robust security measures and risk analysis in AI-as-a-service environments to protect against potential threats and vulnerabilities.

Full Article