ML model production checklist
The purpose of this checklist is to make sure that:
- The team assessed if the model is ready for production before moving to the scoring process
- The team has prepared a production plan for the model
The checklist provides guidelines for creating this production plan. It should be used by teams/organizations that already built/trained an ML model and are now considering putting it into production.
Before putting an individual ML model into production, the following aspects should be considered:
- Is there a well defined baseline? Is the model performing better than the baseline?
- Are machine learning performance metrics defined for both training and scoring?
- Is the model benchmarked?
- Can ground truth be obtained or inferred in production?
- Has the data distribution of training, testing and validation sets been analyzed?
- Have goals and hard limits for performance, speed of prediction and costs been established so they can be considered if trade-offs need to be made?
- How will the model be integrated into other systems, and what impact will it have?
- How will incoming data quality be monitored?
- How will drift in data characteristics be monitored?
- How will performance be monitored?
- Have any ethical concerns been taken into account?
Please note that there might be scenarios where it is not possible to check all the items on this checklist. However, it is advised to go through all items and make informed decisions based on your specific use case.
Will your model performance be different in production than during training phase
Once deployed into production, the model might be performing much worse than expected. This poor performance could be a result of:
- The data to be scored in production is significantly different from the train and test datasets
- The feature engineering steps are different or inconsistent in production compared to the training process
- The performance measure is not consistent (for example your test set covers several months of data where the performance metric for production has been calculated for one month of data)
Is there a well-defined baseline? Is the model performing better than the baseline?
A good way to think of a model baseline is the simplest model one can come up with: either a simple threshold, a random guess or a very basic linear model. This baseline is the reference point your model needs to outperform. A well-defined baseline is different for each problem type and there is no one size fits all approach.
As an example, let's consider some common types of machine learning problems:
- Classification: Predicting between a positive and a negative class. Either the class with the most observations or a simple logistic regression model can be the baseline.
- Regression: Predicting the house prices in a city. The average house price for the last year or last month, a simple linear regression model, or the previous median house price in a neighborhood could be the baseline.
- Image classification: Building an image classifier to distinguish between cats and no cats in an image. If your classes are unbalanced: 70% cats and 30% no cats and if you always predict cats, your naive classifier has 70% accuracy and this can be your baseline. If your classes are balanced: 52% cats and 48% no cats, then a simple convolutional architecture can be the baseline (1 conv layer + 1 max pooling + 1 dense). Additionally, human accuracy at labelling can also be the baseline in an image classification scenario.
Some questions to ask when comparing to a baseline:
- How does your model compare to a random guess?
- How does your model performance compare to applying a simple threshold?
- How does your model compare with always predicting the most common value?
Note: In some cases, human parity might be too ambitious as a baseline, but this should be decided on a case by case basis. Human accuracy is one of the available options, but not the only one.
- "How To Get Baseline Results And Why They Matter" article
- "Always start with a stupid model, no exceptions." article
Are machine learning performance metrics defined for both training and scoring?
The methodology of translating the training metrics to scoring metrics should be well-defined and understood. Depending on the data type and model, the model metrics calculation might differ in production and in training. For example, the training procedure calculated metrics for a long period of time (a year, a decade) with different seasonal characteristics while the scoring procedure will calculate the metrics per a restricted time interval (for example a week, a month, a quarter). Well-defined ML performance metrics are essential in production so that a decrease or increase in model performance can be accurately detected.
Things to consider:
- In forecasting, if you change the period of assessing the performance, from one month to a year for example, then you might get a different result. For example, if your model is predicting sales of a product per day and the RMSE (Root Mean Squared Error) is very low for the first month the model is in production. As the model is live for longer, the RMSE is increasing, becoming 10x the RMSE for the first year compared to the first month.
- In a classification scenario, the overall accuracy is good, but the model is performing poorly for some subgroups. For example, a classifier has an accuracy of 80% overall, but only 55% for the 20-30 age group. If this is a significant age group for the production data, then your accuracy might suffer greatly when in production.
- In scene classification scenario, the model is trying to identify a specific scene in a video, and the model has been trained and tested (80-20 split) on 50000 segments where half are segments containing the scene and half of the segments do not contain the scene. The accuracy on the training set is 85% and 84% on the test set. However, when an entire video is scored, scores are obtained on all segments, and we expect few segments to contain the scene. The accuracy for an entire video is not comparable with the training/test set procedure in this case, hence different metrics should be considered.
- If sampling techniques (over-sampling, under-sampling) are used to train model when classes are imbalanced, ensure the metrics used during training are comparable with the ones used in scoring.
- If the number of samples used for training and testing is small, the performance metrics might change significantly as new data is scored.
Is the model benchmarked?
The trained model to be put into production is well benchmarked if machine learning performance metrics (such as accuracy, recall, RMSE or whatever is appropriate) are measured on the train and test set. Furthermore, the train and test set split should be well documented and reproducible.
Can ground truth be obtained or inferred in production?
Without a reliable ground truth, the machine learning metrics cannot be calculated. It is important to identify if the ground truth can be obtained as the model is scoring new data by either manual or automatic means. If the ground truth cannot be obtained systematically, other proxies and methodology should be investigated in order to obtain some measure of model performance.
One option is to use humans to manually label samples. One important aspect of human labelling is to take into account the human accuracy. If there are two different individuals labelling an image, the labels will likely be different for some samples. It is important to understand how the labels were obtained to assess the reliability of the ground truth (that is why we talk about human accuracy).
For clarity, let's consider the following examples (by no means an exhaustive list):
- Forecasting: Forecasting scenarios are an example of machine learning problems where the ground truth could be obtained in most cases even though a delay might occur. For example, for a model predicting the sales of ice cream in a local shop, the ground truth will be obtained as the sales are happening, but it might appear in the system at a later time than as the model prediction.
- Recommender systems: For recommender system, obtaining the ground truth is a complex problem in most cases as there is no way of identifying the ideal recommendation. For a retail website for example, click/not click, buy/not buy or other user interaction with recommendation can be used as ground truth proxies.
- Object detection in images: For an object detection model, as new images are scored, there are no new labels being generated automatically. One option to obtain the ground truth for the new images is to use people to manually label the images. Human labelling is costly, time-consuming and not 100% accurate, so in most cases, only a subset of images can be labelled. These samples can be chosen at random or by using active learning techniques of selecting the most informative unlabeled samples.
Has the data distribution of training, testing and validation sets been analyzed?
The data distribution of your training, test and validation (if applicable) dataset (including labels) should be analyzed to ensure they all come from the same distribution. If this is not the case, some options to consider are: re-shuffling, re-sampling, modifying the data, more samples need to be gathered or features removed from the dataset.
Significant differences in the data distributions of the different datasets can greatly impact the performance of the model. Some potential questions to ask:
- How much does the training and test data represent the end result?
- Is the distribution of each individual feature consistent across all your datasets? (i.e. same representation of age groups, gender, race etc.)
- Is there any data lineage information? Where did the data come from? How was the data collected? Can collection and labelling be automated?
Have goals and hard limits for performance, speed of prediction and costs been established, so they can be considered if trade-offs need to be made?
Some machine learning models achieve high ML performance, but they are costly and time-consuming to run. In those cases, a less performant and cheaper model could be preferred. Hence, it is important to calculate the model performance metrics (accuracy, precision, recall, RMSE etc), but also to gather data on how expensive it will be to run the model and how long it will take to run. Once this data is gathered, an informed decision should be made on what model to productionize.
System metrics to consider:
- CPU/GPU/memory usage
- Cost per prediction
- Time taken to make a prediction
How will the model be integrated into other systems, and what impact will it have?
Machine Learning models do not exist in isolation, but rather they are part of a much larger system. These systems could be old, proprietary systems or new systems being developed as a results of the creation a new machine learning model. In both of those cases, it is important to understand where the actual model is going to fit in, what output is expected from the model and how that output is going to be used by the larger system. Additionally, it is essential to decide if the model will be used for batch and/or real-time inference as production paths might differ.
Possible questions to assess model impact:
- Is there a human in the loop?
- How is feedback collected through the system? (for example how do we know if a prediction is wrong)
- Is there a fallback mechanism when things go wrong?
- Is the system transparent that there is a model making a prediction and what data is used to make this prediction?
- What is the cost of a wrong prediction?
How will incoming data quality be monitored?
As data systems become increasingly complex in the mainstream, it is especially vital to employ data quality monitoring, alerting and rectification protocols. Following data validation best practices can prevent insidious issues from creeping into machine learning models that, at best, reduce the usefulness of the model, and at worst, introduce harm. Data validation, reduces the risk of data downtime (increasing headroom) and technical debt and supports long-term success of machine learning models and other applications that rely on the data.
Data validation best practices include:
- Employing automated data quality testing processes at each stage of the data pipeline
- Re-routing data that fails quality tests to a separate data store for diagnosis and resolution
- Employing end-to-end data observability on data freshness, distribution, volume, schema and lineage
Note that data validation is distinct from data drift detection. Data validation detects errors in the data (ex. a datum is outside of the expected range), while data drift detection uncovers legitimate changes in the data that are truly representative of the phenomenon being modeled (ex. user preferences change). Data validation issues should trigger re-routing and rectification, while data drift should trigger adaptation or retraining of a model.
How will drift in data characteristics be monitored?
Data drift detection uncovers legitimate changes in incoming data that are truly representative of the phenomenon being modeled,and are not erroneous (ex. user preferences change). It is imperative to understand if the new data in production will be significantly different from the data in the training phase. It is also important to check that the data distribution information can be obtained for any of the new data coming in. Drift monitoring can inform when changes are occurring and what their characteristics are (ex. abrupt vs gradual) and guide effective adaptation or retraining strategies to maintain performance.
Possible questions to ask:
- What are some examples of drift, or deviation from the norm, that have been experience in the past or that might be expected?
- Is there a drift detection strategy in place? Does it align with expected types of changes?
- Are there warnings when anomalies in input data are occurring?
- Is there an adaptation strategy in place? Does it align with expected types of changes?
How will performance be monitored?
It is important to define how the model will be monitored when it is in production and how that data is going to be used to make decisions. For example, when will a model need retraining as the performance has degraded and how to identify what are the underlying causes of this degradation could be part of this monitoring methodology.
Ideally, model monitoring should be done automatically. However, if this is not possible, then there should be a manual periodical check of the model performance.
Model monitoring should lead to:
- Ability to identify changes in model performance
- Warnings when anomalies in model output are occurring
- Retraining decisions and adaptation strategy