Roadmaps Security roadmaps AI and machine learning security

AI Infrastructure and Supply Chain Security

Secure the platforms that train and serve AI, and the models, datasets and libraries they come from.

Estimated time: 14–24 months · At about 10 hours a week.

Learning path

Click a topic to see what to learn and where. Mark topics as you go; your progress is saved in this browser only.

  • Core topic
  • Optional
  1. Beginner · 1–2 months The AI stack and its weak spots
  2. Intermediate · 3–4 months Securing MLOps and model provenance
  3. Advanced · 4–6 months Training and serving infrastructure
  4. Expert · 6–12 months Platform strategy and ecosystem security

All topics in detail

BeginnerTime: 1–2 months

The AI stack and its weak spots

Skills

  • Draw an organisation's AI stack and its trust boundaries.
  • Load and share models safely.
  • Lock down notebooks and AI API keys.

Hands-on projects and labs

  • Deploy JupyterLab in a container with authentication and no secrets, and check it from outside.
  • Scan a set of public model files with ModelScan and explain any findings.
  • Search a test repository you own for leaked AI API keys with a secret scanner such as Gitleaks.

Done when you can…

  • Explain why loading an untrusted pickle file is like running an untrusted program.
  • Present a map of an AI stack with its top three weak spots.

Beginner

Map the AI stack

List every part: data stores, notebooks, training jobs, experiment trackers, registries, inference servers, vector databases and the APIs in front of them. Each needs an owner and access control.

Resources

Beginner

Notebook security

Jupyter servers left open to the internet give attackers code execution. Require authentication, keep them off public networks and keep secrets out of notebooks.

Resources

Beginner

Malicious model files

Pickle-based model files can run code when loaded. Prefer safetensors, scan files before loading, and never load untrusted models on machines with secrets.

Resources

Beginner

Using public model hubs safely

Check who published a model, how popular and recent it is, its licence and whether it uses safe formats. Typosquatted model names exist, just like typosquatted packages.

Resources

Beginner

API keys and secrets for AI services

Model API keys are valuable and often leak in code, notebooks and logs. Store them in a secrets manager, scope them, set spending limits and rotate them.

Resources

Beginner

AI supply chain threats

Poisoned datasets, backdoored models, malicious libraries and compromised MLOps tools. ATLAS and the OWASP LLM Top 10 both list supply chain as a top risk.

Resources

IntermediateTime: 3–4 months

Securing MLOps and model provenance

Skills

  • Harden an MLflow or Kubeflow deployment.
  • Sign and verify models in a pipeline.
  • Produce an AI BOM for a system.
  • Track dataset provenance with hashes.

Hands-on projects and labs

  • Deploy MLflow with authentication behind a proxy, then add a review step before a model can reach production.
  • Sign a model with Sigstore model-transparency and make the serving step refuse unsigned models.
  • Generate a CycloneDX ML-BOM for a small project with a model, a dataset and its libraries.

Done when you can…

  • Show a pipeline where only a reviewed, signed model can reach production.
  • Answer, from records, exactly which data and code produced a deployed model.

Intermediate

Securing MLOps platforms

Experiment trackers, pipelines and registries such as MLflow and Kubeflow have had serious vulnerabilities. Patch them, require authentication and never expose them to the internet.

Resources

Intermediate

Model registry controls

Control who can push and promote models, require review before production, and keep an audit trail of which model version is deployed where.

Resources

Intermediate

Model signing

Sign models when they are produced and verify the signature before loading, so you know a file has not been swapped or changed.

Resources

Intermediate

AI bills of materials

An AI BOM records which models, datasets and libraries went into a system, and their versions and licences. CycloneDX and SPDX both support it.

Resources

Intermediate

Dataset provenance and integrity

Record where every dataset came from, hash it, and store it read-only. Then you can prove what a model was trained on and spot tampering.

Resources

Intermediate

ML library dependencies

ML projects pull in huge dependency trees. Pin versions, scan for known vulnerabilities and watch for malicious packages with names close to popular ones.

Resources

Intermediate

Build provenance for models

Apply SLSA ideas to training: build models in controlled pipelines and produce signed provenance that says what code and data produced them.

Resources

  • SLSA DocsFreeOpenSSF

AdvancedTime: 4–6 months

Training and serving infrastructure

Skills

  • Harden Kubernetes clusters that run training and inference.
  • Protect model endpoints from abuse and extraction.
  • Protect training data and model weights.
  • Assess third-party model providers.

Hands-on projects and labs

  • Run a small model server on a local Kubernetes cluster (kind or minikube) with network policies, authentication and rate limits.
  • Write detections for unusual reads of a model weights bucket in your own cloud account.
  • Compare two hosted model providers' data retention and security terms and write a recommendation.

Done when you can…

  • Lead the security design of an AI platform.
  • Show that a stolen API key cannot download model weights or training data.

Advanced

Kubernetes for AI workloads

Most training and serving runs on Kubernetes. Apply namespaces, network policies, pod security, workload identity and least-privilege service accounts.

Resources

Advanced

GPU and accelerator security

GPU drivers and container toolkits have had serious vulnerabilities, and shared GPUs can leak data between tenants. Patch quickly and isolate sensitive workloads.

Resources

Advanced

Securing inference endpoints

Put authentication, rate limits and input size limits in front of every model endpoint. Monitor for abuse, extraction attempts and runaway costs.

Resources

AdvancedOptional

Confidential computing for AI

Trusted execution environments, including on recent GPUs, protect data and models while in use, even from the cloud operator.

Resources

Advanced

Protecting training data at scale

Training data is often the most sensitive asset. Encrypt it, restrict who can read it, log access and keep production personal data out unless it is truly needed.

Resources

Advanced

Protecting model weights

Valuable model weights are a target for theft. Limit who can read them, monitor downloads and treat the storage like your most sensitive data.

Resources

Advanced

Third-party model and API risk

When you call a hosted model, your data leaves. Check retention, training use, location and the provider's security, and plan for outages and model changes.

Resources

ExpertTime: 6–12 months

Platform strategy and ecosystem security

Skills

  • Design an AI platform that makes secure use the default.
  • Contribute fixes and standards to the open AI ecosystem.

Hands-on projects and labs

  • Write a reference architecture for a secure AI platform, with registry, signing, serving and monitoring.
  • Find and responsibly report a vulnerability in an open-source ML tool.

Done when you can…

  • Own the security of an AI platform used by many teams.
  • Have your fixes, tools or guidance adopted in the open AI ecosystem.

Expert

AI platform security strategy

Give every team a paved road: approved models, a scanned and signed registry, hardened serving, central secrets and logging, so the secure way is also the easy way.

Resources

Expert

Securing the open AI ecosystem

Find and report vulnerabilities in ML libraries and platforms, help projects adopt signing and safe formats, and share threat intelligence on malicious models.

Resources

ExpertOptional

Security for frontier model developers

Labs that train the largest models need protection against well-resourced attackers, including insider threats and nation states.

Resources

Overview

Behind every AI feature is a stack of infrastructure: notebooks, training clusters, GPUs, data lakes, experiment trackers, model registries and inference servers. Much of it was built for speed by research teams, and attackers have found exposed dashboards, leaked keys and unauthenticated APIs in it.

AI also has a new supply chain. Teams download pre-trained models, datasets and libraries from public hubs, often with little checking. A model file can carry malicious code or a hidden backdoor, and a dataset can be poisoned. This roadmap covers both: the platform and what flows through it.

Why it matters

Organisations are spending heavily on AI platforms and depend on models they did not build. Securing this layer uses familiar cloud, container and supply chain skills, applied to new components, and few security teams cover it yet.

Typical job roles

  • EntryMLOps or platform engineer Builds and runs the AI platform with secure defaults.
  • Mid-levelAI platform security engineer Secures training and serving infrastructure, secrets and access.
  • Mid-levelAI supply chain security engineer Scans, signs and tracks models, datasets and dependencies.
  • SeniorAI infrastructure security architect Designs isolation, identity and data protection across the AI platform.
  • LeadHead of AI platform security Owns the security of all AI infrastructure and third-party AI components.

Prerequisites

  • Cloud security basics, ideally the Cloud Security roadmap's beginner and intermediate stages.
  • Containers and Kubernetes basics.
  • Software supply chain basics: dependencies, SBOMs and signing.
  • A working idea of how models are trained and served.

Cloud and DevOps AI and machine learning Cybersecurity

Tools and technologies

  • ModelScan Open source Scans model files for unsafe code.
  • picklescan Open source Detects dangerous imports in pickle files.
  • Safetensors Open source A safe file format for model weights.
  • Sigstore model-transparency Open source Sign and verify ML models.
  • CycloneDX Open source BOM standard with ML-BOM support.
  • MLflow Open source Experiment tracking and model registry.
  • Trivy Open source Scans containers, Kubernetes and dependencies.
  • OpenSSF Scorecard Open source Rates the security practices of open-source projects.

Certifications

Take them in roughly this order. Costs and details for the ones marked with a link are on the Certifications page.

  1. Advanced Certified Kubernetes Administrator (CKA) Linux Foundation (CNCF)Kubernetes skills for running AI workloads.Cost and exam details
  2. Advanced AWS Certified Security – Specialty Amazon Web ServicesCloud security for AI platforms on AWS.Cost and exam details
  3. Intermediate AWS Certified Machine Learning Engineer – Associate Amazon Web ServicesShows you understand how ML runs in production.Cost and exam details

Resources

Frameworks and standards

All frameworks and standards

Career path

A common progression. Titles vary between companies.

  1. Cloud, DevOps or MLOps engineer0–3 years
  2. Platform security engineer2–5 years
  3. AI platform or AI supply chain security engineer4–7 years
  4. AI infrastructure security architect7+ years

Related domains to move into

All security roadmaps Back to top

Last reviewed: . Found a mistake, or want a roadmap written sooner? Contact