BlogData Architecture

Data Governance for AI: What Changes When Models Consume Your Data

Austin Duncan
Austin Duncan
Senior Data Architect
·July 28, 202610 min read

AI systems consuming enterprise data create governance requirements that traditional data governance frameworks did not anticipate — training data quality, model lineage, feature store governance, and the audit requirements for AI-driven decisions. This guide covers what changes.

Traditional data governance was designed for human consumers: analysts querying tables, BI tools rendering dashboards, reports reviewed in board meetings. The primary concerns were accuracy, access control, and compliance. AI systems change the governance requirements significantly — they consume data at scale, learn from it, encode it into weights, and make decisions from it. This guide covers what changes in data governance when AI models are part of your data infrastructure.

Why AI creates new governance requirements

When a human analyst uses a report, they bring judgment, context, and the ability to recognise when something looks wrong. When an AI model uses data, it has no such judgment. It treats training data as ground truth. Biased training data produces biased models. Stale training data produces models with stale beliefs about the world. Missing data produces models with blind spots. The model does not know what it does not know.

This creates governance requirements that were not important in analytics-only environments:

**Training data quality**: For a dashboard, incorrect data produces incorrect numbers that a viewer might notice and question. For a model, incorrect data is encoded into weights and applied silently to every subsequent prediction.

**Training data lineage**: When a model produces an unexpected output, the investigation often starts with "what data did this model learn from?" If the training dataset's provenance is undocumented, debugging the model's behavior is extremely difficult.

**Feature governance**: ML features — the derived inputs to a model — encode business logic. A customer churn model with a "days_since_last_purchase" feature encodes a specific definition of engagement. That definition needs to be governed, versioned, and documented like any other business metric.

**Model-level audit requirements**: Regulatory frameworks (GDPR Article 22, the EU AI Act, US state-level AI regulations) require that AI-driven decisions be explainable and auditable. You cannot produce an audit trail for a decision made by a model trained on ungoverned, undocumented data.

Training data governance

The first governance requirement for AI is treating training datasets as first-class data assets — documented, versioned, and quality-tested like any other production dataset.

**Dataset versioning**: Each training run should reference a specific, immutable version of the training dataset. If the model is retrained six months later, the original training dataset must still be accessible to reproduce the original model's behavior. Delta Lake's time travel or Iceberg's snapshot history can provide immutable dataset versions. Alternatively, training datasets are materialised and archived before each training run.

**Data quality before training**: The quality tests you apply to analytics models (not_null, referential integrity, range checks, statistical drift) are equally applicable to training data. A training dataset with 20% null values in a key feature, or with systematic labeling errors in the target variable, will produce a systematically incorrect model. Data quality testing should be a blocking gate in the ML pipeline before data reaches model training.

**Documentation**: Training datasets should document: what data sources were joined, what time period was covered, what preprocessing was applied, what the positive/negative class ratio is (for classification), what geographic or demographic coverage is included. This documentation is required both for model debugging and for regulatory explainability.

**Privacy and sensitive data**: Training data for models that will be deployed in customer-facing applications requires careful handling of PII. Models can memorise training data — a language model trained on customer emails may reproduce individual customers' text when prompted appropriately. Apply differential privacy techniques, anonymisation, or data minimisation to training datasets that contain sensitive information.

Feature store governance

A feature store is a centralised repository for ML features — derived columns computed from raw data, designed to be reusable across multiple models. Without a feature store, each ML team recomputes the same features independently, often with subtle definitional differences.

Feature store governance requires:

- Feature definitions documented with business meaning, calculation logic, and expected ranges

- Feature versioning (a feature definition change creates a new version; old models continue using the old version)

- Ownership assigned — who is responsible for keeping this feature current and correct

- Lineage from raw data through feature computation to model training

Feature stores (Feast open-source, Databricks Feature Store, AWS SageMaker Feature Store, Vertex AI Feature Store) implement this governance infrastructure. A mature ML organisation uses a feature store as the primary data access layer for model training, preventing feature drift and ensuring reproducibility.

Model cards and model documentation

A model card is a standardised documentation format for ML models, proposed by Google researchers. It records:

- Model purpose and intended use cases

- Training data description (sources, version, time period, coverage)

- Performance metrics on evaluation datasets

- Known limitations and failure modes

- Fairness and bias evaluation across demographic groups

- Acceptable and unacceptable uses

Model cards serve multiple governance purposes: they enable informed deployment decisions, provide the documentation required for AI audit, and communicate model limitations to downstream consumers who may not have ML expertise.

The EU AI Act requires risk-based documentation for AI systems — model cards provide a starting structure, though regulated industries will need more detailed documentation than the card format alone.

Audit trails for AI-driven decisions

When a model denies a credit application, assigns a risk score to an insurance claim, or triggers a fraud alert, the decision must be auditable. This requires:

- Logging the inputs to the model at the time of each decision

- Logging the model version used to make the decision

- Logging the model's output and the decision rule applied to that output

- Retaining logs for the regulatory retention period (often 7 years in financial services)

This audit trail must be immutable — it cannot be modified after the fact. Append-only storage (S3 with object lock, Delta Lake with retention policies) provides this guarantee.

Explainability tools (SHAP values, LIME, attention maps for neural networks) can add "why did the model produce this output?" context to the audit record. For regulated decisions, explainability is often required — not just the outcome, but the factors that drove it.

Data access governance for AI

AI systems accessing your data at training time need the same access governance as human users:

- Service accounts with minimum necessary permissions to the training data

- Row-level security where applicable (a model trained on all customer data when it should only use consented data is a compliance violation)

- Audit logging of data access by AI training pipelines (the same way you log analyst queries)

For models that consume live data at inference time (a recommendation model querying user history), the same RLS and access controls that apply to human users should apply to the model's service account.

For the foundational governance context, see data governance framework and ai-ready data infrastructure. Our data architecture consulting practice designs AI-ready data governance frameworks for organisations deploying ML in production — book a free architecture review.

Get your data architecture audit in 30 minutes.

A former Microsoft data architect audits your data foundation, identifies your top priorities, and sends you a written plan. Free. No pitch.

Book a Call →