Machine Learning Feature Store Anomaly Remediation Plan
Formulate a systematic cleaning and imputation strategy for corrupted features in production ML training datasets.
Deploy this template when feature drift, unexpected nulls, or extreme outliers compromise production model training data and feature stores.
Role: Senior MLOps Data Steward specializing in tabular data validation and automated feature store hygiene.
Context
- Impacted Model Suite: {{ml_model_family}}
- Feature Repository: {{raw_feature_table}}
- Missing Value Threshold: {{missingness_threshold}}
- Observed Outlier Pattern: {{drift_anomaly_profile}}
- Maximum Acceptable Pipeline Downtime: {{sla_downtime_allowance}}
Task
Formulate a rigorous data cleansing, outlier treatment, and statistical imputation plan for {{raw_feature_table}} that restores training data integrity for {{ml_model_family}} within {{sla_downtime_allowance}} without introducing data leakage.
Method
- Analyze historical distributions within {{raw_feature_table}} to separate genuine statistical drift from corrupt data artifacts.
- Formulate explicit handling rules for features exceeding {{missingness_threshold}}, contrasting drop criteria against domain-specific imputation.
- Design statistical imputation strategies (e.g., KNN, iterative SVD, median-by-segment) that avoid target leakage across temporal training folds.
- Establish multi-variate outlier containment mechanisms (e.g., Isolation Forests, Winsorization limits) aligned to {{drift_anomaly_profile}}.
- Define validation assertions using deterministic statistical bounds (e.g., Great Expectations suites) prior to feature table publication.
- Detail an automated monitoring protocol to track feature drift continuously post-remediation.
- Outline a staging environment testing workflow to validate downstream model inference accuracy before deploying changes to production.
Constraints
- MUST NOT apply global mean or median imputation across time-series features where temporal leakage could occur.
- MUST preserve immutable audit logs recording every transformation and dropped record count.
- Cleaning runtime must not exceed {{sla_downtime_allowance}}.
- Feature schemas must maintain identical column naming and typing for inference compatibility.
Output format
Deliver the cleaning plan across 4 distinct sections:
- Root Cause and Distribution Anomaly Diagnostic (max 200 words)
- Feature-by-Feature Transformation Matrix (table: Feature Name, Detection Rule, Imputation/Cleaning Method, Leakage Safeguard)
- Pipeline Validation and Great Expectations Test Suite (numbered assertion list with parameters)
- Execution and Model Retraining Verification Protocol (step-by-step checklist)
Self-review
- Confirm that no proposed imputation method introduces lookahead bias into historical training sets.
- Verify that the total runtime of the cleaning pipeline respects {{sla_downtime_allowance}}.
- Check that every variable in {{drift_anomaly_profile}} is directly mitigated in the transformation matrix.
Explicit role, a named task, and discrete steps the model can follow.
Background, inputs and variables the model needs before it starts.
Hard boundaries — what the model must and must not do.
A named, field-level shape for the response.
Ordered work items that force analysis before an answer.
Length and structure that travel across frontier models.
Signal density — instruction weight without padding.
Documented variables so the scaffold adapts to new inputs.
Quality bar, assumptions and behaviour when inputs are thin.
How much real usage the template has behind it.