Skip to content

10 AI Supply Chain and Model Security

Modern AI applications depend on a large number of external and internal components.

An enterprise AI system may rely on:

  • Foundation models
  • Open-source models
  • Embedding models
  • Tokenizers
  • Model repositories
  • Python packages
  • AI frameworks
  • Containers
  • Fine-tuning datasets
  • External APIs
  • CI/CD pipelines
  • MLOps platforms

Every dependency creates another trust boundary.

A system may appear secure at runtime while still being compromised through an upstream component.

Conceptually:

Model
+
Framework
+
Dependency
+
Dataset
+
Container
+
Pipeline
Enterprise AI Application

This is the AI supply chain.

For an AI Security Engineer, the central question is:

Can we trust the models, code, data and artifacts that make up our AI system?

By the end of this lesson, you should be able to:

  • Explain what the AI supply chain includes.

  • Understand model provenance.

  • Recognize risks associated with external and open-source models.

  • Understand model repository security.

  • Recognize malicious or tampered model artifacts.

  • Understand dependency and package risk.

  • Understand AI framework security.

  • Recognize container and runtime risks.

  • Understand dataset provenance.

  • Protect fine-tuning pipelines.

  • Understand model integrity and version control.

  • Apply vulnerability management to AI components.

  • Understand AI Bill of Materials concepts.

  • Secure model promotion and deployment.

  • Assess AI supply-chain risk systematically.

The AI supply chain includes the technologies, data and artifacts required to build and operate an AI system.

A simplified lifecycle might be:

Model Source
Dependencies
Dataset
Development
Fine-Tuning
Model Registry
Container
Deployment Pipeline
Production

Any stage may become a target.

Traditional Software Supply Chain vs AI Supply Chain

Section titled “Traditional Software Supply Chain vs AI Supply Chain”

Traditional software supply chains already include:

  • Source code

  • Packages

  • Build systems

  • Containers

  • CI/CD

  • Artifact repositories

AI adds additional components:

Models
Training Data
Fine-Tuning Data
Embedding Models
Model Registries
Model Weights
AI Frameworks

Therefore:

Software Supply Chain
+
Model Supply Chain
+
Data Supply Chain
=
AI Supply Chain

Imagine an enterprise carefully secures:

  • Authentication

  • RAG

  • AI agents

  • Cloud infrastructure

but downloads an untrusted model artifact.

Attack path:

Untrusted Model
Enterprise Development
Production Deployment

The organization may introduce the risk itself.

Supply-chain security begins before production.

A useful principle is:

Do not assume an AI component is trustworthy simply because it works.

Ask:

Where did it come from?
Who published it?
Which version is it?
Has it been reviewed?
Can integrity be verified?
What dependencies does it contain?
Who approved it?

Foundation models may come from:

  • Commercial AI providers

  • Cloud platforms

  • Open-source ecosystems

  • Internal model teams

Each source creates different security considerations.

Architecture:

Enterprise Application
External Model Provider

Security questions include:

  • Who operates the service?

  • What data is sent?

  • What is retained?

  • What model version is used?

  • Can the provider change the model?

  • How is availability handled?

Architecture:

Enterprise Infrastructure
Model Runtime
Model Files

Security responsibilities increase.

The organization now needs to protect:

  • Model files

  • Runtime

  • GPUs

  • Storage

  • Access control

  • Patching

  • Availability

Model provenance means knowing where a model came from and how it reached production.

A model record may include:

Model:
example-model
Source:
Approved Repository
Version:
v3.2
Publisher:
Approved Vendor
Checksum:
Verified
Security Review:
Completed
Production Approval:
Approved

This creates traceability.

Ask:

Who created the model?
Where was it downloaded from?
Was it modified?
Was it fine-tuned?
Which dataset was used?
Who approved deployment?

Without provenance, trust becomes difficult to establish.

Open-source models provide significant flexibility.

But organizations should not assume:

Popular
=
Secure

Potential concerns include:

  • Unknown provenance

  • Tampered artifacts

  • Unreviewed dependencies

  • Unsafe serialization

  • Licensing concerns

  • Unknown training history

Before use, consider:

Source Repository
Publisher
Community Reputation
Release History
Artifact Integrity
File Format
Dependencies
Security Advisories
License

This is similar to evaluating open-source software.

Models may be stored in:

  • Public model hubs

  • Cloud storage

  • Internal registries

  • Artifact repositories

The repository becomes a critical trust boundary.

Model Repository
Development
Production

Potential risks include:

  • Unauthorized model upload

  • Model replacement

  • Account compromise

  • Malicious artifacts

  • Deleted or altered versions

A stronger architecture:

External Model
Security Review
Approved Internal Registry
Controlled Deployment

Production should preferably deploy from approved trusted locations rather than arbitrary internet sources.

An attacker may attempt to replace or modify a model artifact.

Attack path:

Attacker
Model Registry Access
Model Replaced
Production Deployment

This is an integrity failure.

Organizations may track integrity using:

  • Checksums

  • Signatures

  • Immutable artifacts

  • Versioned registries

  • Controlled publishing

Conceptually:

Downloaded Model
Integrity Verification
Expected?
├── No → Reject
└── Yes → Continue

Model files may represent valuable intellectual property.

Protect them from:

  • Unauthorized download

  • Modification

  • Deletion

  • Copying

Use appropriate:

  • IAM

  • Storage security

  • Encryption

  • Audit logging

  • Network controls

AI model ecosystems may use different file formats and serialization mechanisms.

Some formats may carry more risk than others depending on how they are loaded.

The practical security principle is:

Do not load arbitrary model artifacts from untrusted sources into privileged environments.

Treat model loading as a potentially security-sensitive operation.

Before introducing an external model into enterprise environments:

External Model
Isolated Evaluation Environment
Security Review
Approved Registry

This limits potential impact during assessment.

AI applications heavily depend on Python ecosystems.

Dependencies may include:

  • ML libraries

  • LLM frameworks

  • Vector clients

  • Data processing libraries

  • Agent frameworks

Traditional dependency risks still apply.

Potential threats include:

  • Vulnerable packages

  • Malicious packages

  • Dependency confusion

  • Typosquatting

  • Compromised maintainers

Use:

Approved Package Sources
Version Pinning
Dependency Scanning
Build

Avoid unnecessarily downloading arbitrary packages during runtime.

AI applications may use frameworks for:

  • Model inference

  • RAG

  • Agents

  • Tool orchestration

A vulnerability in the framework may affect many applications.

Security teams should track:

  • Framework versions

  • Security advisories

  • Dependency trees

  • Patch requirements

Framework security belongs to the broader secure development lifecycle.

Avoid uncontrolled dependency changes.

Weak:

package >= latest

may produce unexpected changes over time.

Controlled environments often prefer:

Known Tested Version

with deliberate upgrades.

The exact package-management approach depends on engineering practices.

AI applications frequently run in containers.

The container supply chain may include:

Base Image
OS Packages
Python Dependencies
AI Application
Model Runtime

Potential concerns include:

  • Vulnerable images

  • Malicious packages

  • Embedded credentials

  • Excessive privileges

  • Outdated components

Consider:

  • Approved base images

  • Image scanning

  • Minimal images

  • Non-root execution

  • Secret scanning

  • Image signing

  • Controlled registries

Detailed container security may be covered in specialist infrastructure tracks, but AI Security Engineers should understand the supply-chain dependency.

A trusted container image is only useful if the registry is secure.

Attack path:

Attacker
Container Registry
AI Image Replaced
Deployment

Controls include:

  • Strong authentication

  • Restricted publishing rights

  • Immutable tags where appropriate

  • Artifact signing

  • Audit logging

AI systems depend heavily on data.

Datasets may originate from:

  • Internal databases

  • Public datasets

  • Third-party providers

  • User submissions

  • Data marketplaces

  • Generated synthetic data

Every source has trust implications.

Track:

Dataset Name
Source
Owner
Version
Classification
Usage Rights
Approval Status
Integrity

This helps understand what influenced the model.

Attackers may manipulate training or fine-tuning data.

Conceptually:

Trusted Dataset
+
Malicious Samples
Training
Modified Model Behavior

The attack targets model integrity.

Potential controls include:

  • Approved data sources

  • Access control

  • Versioning

  • Integrity checks

  • Change review

  • Anomaly detection

The exact controls depend on the training environment.

Organizations may fine-tune models using enterprise data.

Architecture:

Base Model
+
Fine-Tuning Dataset
Training
Fine-Tuned Model

Every component should be traceable.

Ask:

Which base model?
Which version?
Which dataset?
Who approved the dataset?
Who initiated training?
Where was training performed?
Where is the resulting model stored?

A mature process may look like:

Approved Base Model
Approved Dataset
Controlled Training
Evaluation
Security Review
Approved Model Registry

Training environments may contain:

  • GPUs

  • Sensitive datasets

  • Model files

  • Credentials

  • Development notebooks

Potential threats include:

  • Unauthorized access

  • Data theft

  • Model theft

  • Credential compromise

Training environments deserve enterprise security controls.

AI engineers often use notebooks for experimentation.

Potential risks include:

  • Hard-coded credentials

  • Sensitive output

  • Excessive cloud permissions

  • Untrusted code execution

Development convenience should not bypass security controls.

AI applications may rely on external:

  • Model APIs

  • Embedding APIs

  • Search APIs

  • Data APIs

Each becomes a supply-chain dependency.

Questions include:

What if the API is compromised?
What if the response is manipulated?
What if the service changes behavior?
What if availability fails?

Third-party service risk should be understood.

AI platforms may support plugins or integrations.

Potential risks include:

  • Excessive permissions

  • Malicious plugin behavior

  • Compromised updates

  • Third-party data access

Only approved integrations should normally be enabled in enterprise environments.

AI applications and models may be promoted through CI/CD.

Example:

Code
Build
Security Tests
Artifact
Deployment

Attackers may target:

  • Source repositories

  • Pipeline credentials

  • Build agents

  • Artifact registries

  • Deployment identities

CI/CD systems often hold powerful credentials.

Avoid:

Pipeline
Permanent Administrator Key

Prefer restricted deployment identities where possible.

MLOps extends CI/CD into model lifecycle management.

A simplified MLOps pipeline:

Dataset
Training
Evaluation
Model Registry
Deployment
Monitoring

Each stage should have defined:

  • Identity

  • Authorization

  • Integrity

  • Logging

Avoid:

Developer Model
Direct Production

Prefer:

Development Model
Evaluation
Security Validation
Approval
Production Registry
Deployment

Model versioning is critical.

Track:

Model Name
Version
Source
Configuration
Fine-Tuning Dataset
Evaluation Results
Approval Status

This supports investigation and rollback.

Imagine production behavior changes.

Investigators ask:

Which model was active?
Was the model changed?
Was it fine-tuned?
Which dataset was used?

Without versioning, investigation becomes much harder.

Organizations should be able to move from:

Model v3

back to:

Model v2

if a serious issue appears.

Rollback capability should be tested.

Security-sensitive behavior may depend on configuration.

Track changes to:

  • System configuration

  • Safety settings

  • Tool configuration

  • Model routing

  • Prompt templates

The production artifact is not only the model file.

Credentials may appear inside:

  • Repositories

  • Model configuration

  • Notebooks

  • Containers

  • CI/CD

  • Logs

Secret scanning should include AI development environments.

MODEL_API_KEY="..."

committed to source control.

Application
Managed Secret / Workload Identity
Approved Service

AI applications still need vulnerability management.

Inventory:

Operating System
Containers
Libraries
AI Frameworks
Vector Database
Model Serving Runtime

Then:

Identify Vulnerability
Assess Exposure
Prioritize
Remediate
Validate

AI ecosystems may update frequently.

This makes component inventory particularly important.

Without inventory, teams may not know:

Are we using the vulnerable version?

Traditional software environments increasingly use an SBOM:

Software Bill of Materials

AI environments may require broader inventories including models and data.

A conceptual AI Bill of Materials might include:

Application
Foundation Model
Model Version
Embedding Model
AI Framework
Libraries
Container Images
Datasets
External Services

The exact terminology and implementation may vary.

The important goal is visibility.

Component Version Source Owner Approved
AI Application 4.2 Internal App Team Yes
Foundation Model v3 Approved Provider AI Team Yes
Embedding Model v2 Internal Registry AI Team Yes
RAG Framework 1.x Approved Package Repo Platform Yes
Container Image 2026.08 Internal Registry DevOps Yes

This makes supply-chain risk manageable.

External AI providers should be assessed like other critical vendors.

Consider:

  • Security controls

  • Data handling

  • Availability

  • Subprocessors

  • Incident notification

  • Access

  • Compliance

AI architecture should document vendor dependencies.

Hosted model providers may update models.

This may change:

  • Model behavior

  • Safety characteristics

  • Output format

  • Performance

Organizations should understand how model changes are managed.

Where necessary, regression testing may be required.

Developers may experiment with new models.

Example:

Developer
Downloads Model
Connects Enterprise Data

Potential concerns include:

  • Unknown trust

  • Data exposure

  • Licensing

  • Unreviewed behavior

Organizations may need approved model catalogs.

Example:

Approved Models
Model A
→ General Enterprise Use
Model B
→ Internal Code Use
Model C
→ Restricted Data Allowed

This helps engineering teams choose supported models.

Attackers may target proprietary models.

Attack path:

Compromised Storage
Model File
Exfiltration

Controls may include:

  • Strong access control

  • Private storage

  • Monitoring

  • Encryption

  • Restricted downloads

When models are exposed only through APIs, attackers may attempt to approximate behavior through repeated querying.

Potential controls may include:

  • Authentication

  • Rate limiting

  • Monitoring

  • Quotas

The exact threat depends on the business value of the model.

Security also includes availability.

What happens if:

External Model Provider
Unavailable

or:

Model Repository
Unavailable

Organizations may require:

  • Redundancy

  • Cached approved artifacts

  • Alternative providers

  • Recovery procedures

A useful model is:

Source
Acquire
Validate
Store
Build
Test
Approve
Deploy
Monitor

Ask security questions at every stage.

Where did it come from?
Was it downloaded through an approved channel?
Can we verify integrity and provenance?
Is it stored in a controlled registry?
Are dependencies controlled?
Was security evaluation performed?
Who approved production usage?
Is deployment controlled?
Can changes or vulnerabilities be detected?

A stronger model lifecycle might look like:

External / Internal Model Source
Provenance Check
Integrity Verification
Isolated Evaluation
Security Review
Approved Internal Model Registry
Controlled Deployment Pipeline
Production
Monitoring
Approved Base Model
+
Approved Dataset
Controlled Fine-Tuning
Evaluation
Security Testing
Versioned Model Registry
Approval
Production

When reviewing an enterprise AI application:

Document:

Foundation Model
Embedding Model
Framework
Packages
Containers
Datasets
External APIs
Model Registry
CI/CD

For each component:

Internal?
Vendor?
Open Source?
Public Repository?
Unknown?

Ask:

Who owns this component internally?
Who approves updates?
Who monitors vulnerabilities?

Verify:

Source
Version
Publisher
Integrity
Approval

Determine who can:

Upload Models
Modify Models
Delete Models
Publish Packages
Change Pipelines
Deploy Production

Check:

  • Authentication

  • Authorization

  • Versioning

  • Logging

  • Integrity controls

Identify:

  • Vulnerable libraries

  • Unpinned packages

  • Unapproved package sources

Determine:

Where does training data come from?
Who can modify it?
How is it versioned?
Is sensitive data included?

Verify approved:

Base Model
Dataset
Training Environment
Resulting Model

Ask:

Can developers deploy directly?
Is approval required?
Are artifacts immutable?
Which identity performs deployment?

Verify teams can detect:

  • Model changes

  • Registry changes

  • Pipeline changes

  • Package vulnerabilities

  • Unauthorized deployment

Determine whether the organization can:

Rollback Model
Replace Dependency
Rebuild Container
Revoke Credential
Disable Model
Finding:
Production Application Uses Unapproved External Model
Affected Component:
Enterprise AI Assistant
Observed Behavior:
The production application loads a model directly
from an external public repository.
Security Concern:
Model provenance and artifact integrity are not
validated before production use.
Potential Impact:
A tampered or malicious model artifact could enter
the enterprise AI environment.
Recommendation:
Establish an approved model intake process, perform
provenance and integrity validation, evaluate models
in an isolated environment and promote approved
artifacts into an internal model registry.
Finding:
Developers Can Replace Production Model Artifacts
Affected Component:
Enterprise Model Registry
Observed Permission:
Development users have write access to the production
model repository.
Potential Impact:
Compromised developer credentials could modify or replace
the model used by production AI applications.
Recommendation:
Separate development and production model registries,
restrict publishing privileges, require controlled
promotion and monitor production model changes.
Finding:
AI Application Uses Uncontrolled Runtime Dependencies
Affected Component:
AI Application Build Pipeline
Observed Behavior:
The production build downloads package versions dynamically
without a controlled dependency lock or approved package source.
Potential Impact:
Unexpected or malicious dependency changes may enter the
production build.
Recommendation:
Use controlled package sources, version pinning or lock files,
dependency scanning and reproducible build processes.
Finding:
Fine-Tuning Dataset Has No Provenance or Change History
Affected Component:
Enterprise Fine-Tuning Pipeline
Observed Behavior:
The training dataset can be modified without documented
ownership, versioning or approval.
Potential Impact:
Unauthorized or poisoned data may influence production
model behavior without reliable traceability.
Recommendation:
Establish dataset ownership, trusted data sources,
version control, approval workflows and integrity
monitoring for fine-tuning datasets.

Example — Direct Production Deployment Finding

Section titled “Example — Direct Production Deployment Finding”
Finding:
AI Models Can Be Promoted Directly From Development to Production
Affected Component:
MLOps Deployment Pipeline
Observed Behavior:
Developers can deploy new model versions without
independent validation or production approval.
Potential Impact:
Untested, compromised or incorrectly configured models
may reach production.
Recommendation:
Introduce evaluation, security validation, controlled
model promotion, approval gates and rollback capability.
  • Model sources documented.

  • Model provenance known.

  • Model versions tracked.

  • Integrity validated.

  • Approved model catalog maintained.

  • Model registries authenticated.

  • Publishing rights restricted.

  • Production artifacts protected.

  • Changes logged.

  • Public models reviewed.

  • External artifacts evaluated safely.

  • License requirements reviewed.

  • Untrusted artifacts isolated.

  • Package sources controlled.

  • Versions managed.

  • Vulnerability scanning implemented.

  • Unnecessary dependencies removed.

  • AI framework versions inventoried.

  • Security advisories monitored.

  • Updates tested before production.

  • Approved base images used.

  • Images scanned.

  • Registries protected.

  • Secrets excluded from images.

  • Dataset sources documented.

  • Provenance maintained.

  • Dataset access restricted.

  • Changes versioned.

  • Poisoning risk considered.

  • Base model approved.

  • Dataset approved.

  • Training environment controlled.

  • Output model versioned.

  • Evaluation completed.

  • Pipelines authenticated.

  • Pipeline credentials restricted.

  • Deployment identities least privileged.

  • Production approval exists.

  • Rollback supported.

  • Credentials excluded from source.

  • Credentials excluded from model context.

  • Secrets managed centrally.

  • Secret scanning enabled where appropriate.

  • Models inventoried.

  • Libraries inventoried.

  • Containers inventoried.

  • External services inventoried.

  • Dataset dependencies understood.

  • Model changes monitored.

  • Registry changes logged.

  • Unauthorized publishing detectable.

  • Vulnerability exposure tracked.

  • Models can be rolled back.

  • Compromised artifacts can be removed.

  • Dependencies can be rebuilt.

  • Credentials can be revoked.

  • Production models can be disabled.

Section titled “Mistake 1 — Trusting Popular Models Automatically”

Popularity does not establish provenance or integrity.

Mistake 2 — Downloading Models Directly Into Production

Section titled “Mistake 2 — Downloading Models Directly Into Production”

External models should pass through a controlled intake process.

Mistake 3 — Protecting Code but Ignoring Models

Section titled “Mistake 3 — Protecting Code but Ignoring Models”

Models are also production artifacts.

Training data can influence production model behavior.

Mistake 5 — Letting Developers Replace Production Models

Section titled “Mistake 5 — Letting Developers Replace Production Models”

Production publishing should be controlled.

Mistake 6 — Ignoring Python Dependencies

Section titled “Mistake 6 — Ignoring Python Dependencies”

Traditional software supply-chain risks still apply.

Mistake 7 — Assuming Model Files Are Harmless Data

Section titled “Mistake 7 — Assuming Model Files Are Harmless Data”

Model-loading processes may themselves have security implications.

Mistake 8 — Using Production Data for Uncontrolled Experiments

Section titled “Mistake 8 — Using Production Data for Uncontrolled Experiments”

Development environments should use appropriate datasets and access controls.

Organizations should be able to restore a previously approved version.

You cannot effectively secure what you do not know you are using.

When reviewing the AI supply chain, ask:

Where did the model come from?
Can we verify the artifact?
Who approved it?
Where is it stored?
Who can replace it?
Which dependencies does the application use?
Which container runs it?
Where did the training data come from?
Who can modify the dataset?
How was the model fine-tuned?
Who can deploy to production?
Can we identify every production version?
Can we detect unauthorized changes?
Can we roll back quickly?

This is the mindset of supply-chain security for AI.

You may be asked:

What is the AI supply chain?

A strong answer is:

The AI supply chain includes the models, datasets, libraries, frameworks, containers, repositories, APIs and development or MLOps pipelines required to build and operate an AI application. Each component creates a trust dependency that must be inventoried, validated and controlled.

Another question may be:

What is model provenance?

A strong answer is:

Model provenance is the ability to understand where a model came from, who published or modified it, which version is being used, whether it was fine-tuned, which data influenced it and how it reached production. Provenance helps establish trust and supports incident investigation.

Another question may be:

How would you securely adopt an open-source model?

A strong answer is:

I would obtain it from a trusted source, verify provenance and artifact integrity, review its format and dependencies, evaluate it in an isolated environment, assess relevant security and licensing concerns and promote the approved artifact into a controlled internal model registry before production deployment.

Another question may be:

How would you protect a model registry?

A strong answer is:

I would require strong authentication, restrict publishing and deletion rights, separate development and production repositories, maintain immutable or versioned artifacts where appropriate, log changes and use controlled model promotion rather than allowing arbitrary production replacement.

Another question may be:

Why is dataset provenance important?

A strong answer is:

Because training and fine-tuning data influence model behavior. If the source, ownership, version and modification history of a dataset are unknown, it becomes difficult to detect poisoning, reproduce model behavior or determine what influenced a compromised model.

The AI supply chain includes:

Models
+
Datasets
+
Embedding Models
+
Libraries
+
Frameworks
+
Containers
+
Repositories
+
External APIs
+
CI/CD
+
MLOps

A secure lifecycle should look like:

Trusted Source
Provenance
Integrity Verification
Evaluation
Security Review
Approved Registry
Controlled Deployment
Monitoring

Important principles include:

  • Know where models come from.

  • Verify model integrity.

  • Use controlled model registries.

  • Restrict who can publish production models.

  • Track model versions.

  • Protect training and fine-tuning data.

  • Control software dependencies.

  • Secure container images and registries.

  • Protect CI/CD and MLOps pipelines.

  • Maintain component inventories.

  • Monitor vulnerabilities and changes.

  • Support rollback and recovery.

Most importantly:

A secure AI application cannot be built from components whose origin, integrity and ownership are unknown.

➡️ 11 — Model and Data Poisoning

You now understand the AI supply chain and the importance of trusted models, datasets and dependencies.

The next lesson focuses more deeply on what happens when an attacker deliberately manipulates the information or artifacts used by AI systems.

You will learn:

  • Training data poisoning

  • Fine-tuning poisoning

  • RAG poisoning

  • Model poisoning

  • Backdoored models

  • Trigger-based behavior

  • Dataset integrity

  • Model integrity

  • Poisoning attack paths

  • Detection challenges

  • Data provenance

  • Model provenance

  • Security testing

  • Incident response

  • Enterprise poisoning defenses

You will move from:

Can We Trust Our AI Supply Chain?

to:

What Happens If an Attacker
Deliberately Manipulates
Models or Data?

➡️ Next: 11 — Model and Data Poisoning