Models and microservices should be running on the same continuous delivery stackChris Riccomini on October 29, 2018
I’ve been interested in data science platforms for a long time. My fascination began when I was at LinkedIn, and helped build out the first model building and deployment system for People You May Know. At the time, we did feature engineering and training on Hadoop, job scheduling on Azkaban, and model deployment on Voldemort.
Things have changed a lot since then. Two posts that have caught my attention lately are the machine learning platforms from Airbnb (here) and Uber (here). Airbnb has also been talking about Bighead, their end-to-end ML system. All of these posts do a great job breaking down all of the different steps involved in machine learning, and how each company built a platform to help developers with the process.
Something that caught my eye, though, is that the deployment pipeline for their models is custom built. Both Airbnb and Uber have a fairly standard continuous deployment process for building and deploying web services. Yet, either deliberately or by accident, they chose to build a separate system for their model deployment pipeline.
This is an interesting peculiarity. I can’t recall ever seeing a unified model and service deployment stack. Yet, in the abstract, the deployment needs for models and services are strikingly similar.
Continuous deployment for models
Let’s start by reviewing the continuous delivery process for a microservice.
- Version control
- Unit test
- Deploy to staging
- Test in staging
- Deploy to production
- Measure and validate
This looks very similar to what you’re doing with your models.
Models need to live somewhere when they’re being developed. You need version control. You need to see who made what changes. Sounds a lot like Git.
A model needs to be built before it can be deployed. The process of building a model is definitely different from building a microservice. Microservices are built when code changes. Models are built when either data changes or when code changes. In both cases, though, there is a trigger (new data arrival, or a code merge) that causes a model to be retrained or a microservice to be rebuilt.
Microservices are compiled. Models are trained. The training often involves a much more involved sequence that usually requires some kind of orchestration (like Airbnb’s Airflow).
Still, in the abstract, there is an event that triggers a build, and then a sequence of steps that lead to an artifact. During the build, models, like microservices, need to be assigned a version. Perhaps semantic versioning isn’t quite the best fit, but a versioned artifact is required nonetheless. This makes it easier to track changes, roll forward, roll back, maintain history, etc. This Bighead slide calls out the challenge of model management explicitly. Versioning (and version control), and things like release notes, are how we handle this on the microservice side of the world.
Both microservices and models need to be deployed to production. They’re usually both built (or trained) in a non-production environment. For models, this is usually a modeling (or data warehousing) environment. For microservices, secure build infrastructure is used. The artifacts then get published into a repository. Things get even more interesting if you deploy microservices through Docker. You can layer your data on top of a service’s base image to quickly update and deploy model changes. Data containers also look interesting.
After a build is complete, it must be tested. For models, this usually means validating that the model is performing adequately against holdout data. Microservices need to pass their unit tests.
Deploy to staging
Though it’s not common practice, it makes sense to deploy your model to your staging environment before it goes to production. This will exercise the model deployment machinery, and make sure that your staging environment continues to look like production.
Test in staging
Any integration or user acceptance tests that need to be run can validate that changes in model behavior didn’t break anything obvious.
Deploy to production
After tests pass in staging, it’s time to deploy to production. On the microservice side of the fence, this usually involves either a blue/green deployment or a canary release. Most model deployments follow a similar pattern. You deploy the new model, shift traffic over, then remove the old model.
Measure and validate
Measuring and validating microservices usually involves looking at metric and log information. Are there any anomalies in the logs? Higher WARNs then usual? More exceptions? Has latency spiked, are there fewer/more threads being used, etc?
Again, the same pattern holds for models. The prediction success rate of the model can be measured, and if it dips beyond a threshold, it can be rolled back.
I’m not entirely sure why we ended up here. Two potential explanations are:
- The jump from data science to data engineering to DevOps makes it difficult for teams building data infrastructure to recognize the commonalities with the microservice environment at their organization (c.f. Conway’s law).
- Microservice tooling isn’t flexible enough. For example, CI systems tend to model build steps as linear or parallelizable. Model building usually involves much more complex DAGs where tasks have dependencies on one another.
Same process or same stack
Just because the process for releasing a model can be mapped to a continuous deployment lifecycle doesn’t necessarily mean that the same stack should be used for both pipelines. Airflow is a decent orchestrator for model building, but I don’t think I’d want to use it to build Java microservices. Likewise, CircleCI probably isn’t what you want to use to build a model.
The logical place to start is probably the deployment manager itself. The system that models the continuous deployment lifecycle. It should be possible to use systems like Spinnaker to track the deployment of a model through various environments (testing, staging, production, etc).
Artifact repositories and version control are other places that seem to be likely targets for integration. If you’re building models as pickled objects, tarballs, or Docker images, sticking them in a standard artifact repository is quite doable. Storing the models in version control is also obvious.
Measurement and validation also seem to be likely candidates. Placing model performance in your operational charts and graphs is something that should be done anyway, and adding alerting on top of it shouldn’t be controversial.
When you add all of this up, it paints a compelling argument for a shared stack. Yes, build and testing probably need to be done on separate systems, but versioning, deployment, measurement and validation all fit nicely with existing devops infrastructure.
I admit that it’s not all roses. I’ve already mentioned the differences in the build and test steps. There are other differences, too. Model building is very data-centric. Versioning your data (and providing lineage) is something that’s not really modeled in a traditional continuous deployment pipeline (though, static content deployment is perhaps an analogue). Lambda-style, or realtime model training also don’t map cleanly to a continuous deployment flow. Still, I argue that there exists a large enough overlap to warrant a shared stack for most of the infrastructure, as I outline above.
There are signs of hope in the devops and machine learning spaces. GitLab’s site has an interesting issue entitled GitLab/Devops for AI/ML. Gitlab does a fantastic job modeling continuous integration and delivery, and I’d be excited to progress in this area. There are a lot of posts about deploying models through Docker, version control in JupyterHub, and so on. I’m also pretty excited about how Prefect is thinking about problems like this.
In the meantime, the best path forward is probably to be opportunistic about shared infrastructure. If you find yourself deploying pre-built models, ask why they can’t be deployed via the same repository as your microservices. Likewise, when measuring the performance of a model in production, why not see how easy it’d be to use your existing metrics and monitoring infrastructure?