A team wants to track the lineage of ML pipeline runs, including which datasets, parameters, and models were used in each execution. Which Vertex AI service should they use?
Vertex AI Metadata stores artefacts and executions in a lineage graph, recording which datasets, parameters and models each pipeline run consumed and produced. This directly satisfies the requirement to track lineage across ML pipeline executions.
Why this answer
Vertex AI Metadata (part of Vertex ML Metadata) is the service designed to record and query ML metadata, including artifacts (datasets, models), executions (pipeline runs), and events, forming a lineage graph. It captures which datasets, parameters, and models were used in each pipeline execution, enabling reproducibility and auditability. This directly matches the requirement to track lineage of ML pipeline runs.
Exam trap
PMLE often tests the confusion between Vertex AI Experiments (run tracking and comparison) and Vertex ML Metadata (lineage and artifact relationships), causing candidates to pick Experiments when lineage is the requirement.
How to eliminate wrong answers
Option B (Vertex AI Feature Store) is wrong because it manages and serves feature values for training and online serving, not pipeline run lineage or artifact tracking. Option C (Vertex AI Model Registry) is wrong because it stores and versions trained models and their metadata, but it does not track the full pipeline execution lineage including datasets and parameters. Option D (Vertex AI Experiments) is wrong because it tracks experiment runs, metrics, and parameters for comparison, but it is not the underlying lineage/metadata service that records artifact relationships across pipeline executions.