{Fast, Correct, Simple} - pick three
Easily compare training and production ML data & model distributions
Goals
Boxkite is an instrumentation library designed from ground up for tracking concept drift in HA (Highly Available) model servers. It integrates well with existing DevOps tools (ie. Grafana, Prometheus, fluentd, kubeflow, etc.), and scales horizontally to multiple replicas with no code or infrastructure change.
- Fast
- 0.5 seconds to process 1 million data points (training)
- Sub millisecond p99 latency (serving)
- Supports sampling for large data sets
- Correct
- Aggregates histograms from multiple server replicas (using PromQL)
- Separate counters for discrete and continuous variables (ie. categorical and numeric features)
- Initialises serving histogram bins from training data set (based on Freedman-Diaconis rule)
- Handles unseen data,
nan,None,inf, and negative values
- Simple
- One metric for each counter type (no confusion over which metric to choose)
- Default configuration supports both feature and inference monitoring (easy to setup)
- Small set of dependencies: prometheus, numpy, and fluentd
- Extensible metric system (support for image classification coming soon)
Some non-goals of this project are:
- Adversarial detection
If you are interested in alternatives, please refer to our discussions in FAQ.
Getting Started
Follow one of our tutorials to easily get started and see how Boxkite works with other tools:
- Prometheus & Grafana in Docker Compose locally
- Kubeflow & MLflow on Kubernetes with easy online test drive in the browser
See Installation & User Guide for how to use Boxkite in any environment.
FAQ
- Does boxkite support anomaly / outlier detection?
Prometheus has supported outlier detection in time series data since 2015. Once you've setup KL divergence and K-S test metrics, outlier detection can be configured on top using alerting rules. For a detailed example, refer to this tutorial: https://prometheus.io/blog/2015/06/18/practical-anomaly-detection/.
- Does boxkite support adversarial detection?
Adversarial detection concerns with identifying single OOD (Out Of Distribution) samples rather than comparing whole distributions. The algorithms are also highly model specific. For these reasons, we do not have plans to support them in boxkite at the moment. As an alternative, you may look into Seldon for such capabilities https://github.com/SeldonIO/alibi-detect#adversarial-detection.
- Does boxkite support concept drift detection for text / NLP models?
Not yet. This is still an actively researched area that we are keeping an eye on.
- Does boxkite support tensorflow / pytorch?
Yes, our instrumentation library is framework agnostic. It expects input data to be a list or np.array regardless of how the model is trained.
Contributors
The following people have contributed to the original concept and code
A full list of contributors, which includes individuals that have contributed entries, can be found here.
Shameless plug
Boxkite is a project from BasisAI, who offer an MLOps Platform called Bedrock.
Bedrock helps data scientists own the end-to-end deployment of machine learning workflows. Boxkite was originally part of the Bedrock client library, but we've spun it out into an open source project so that it's useful for everyone!
4 Nov 28, 2021
2 Dec 28, 2021
60 Dec 20, 2022
31 Jan 6, 2023
23 Oct 9, 2022
14 Dec 28, 2022
4 Dec 9, 2022
9 May 11, 2022
4 Aug 16, 2021
6 Dec 8, 2022
3 Mar 30, 2022
148 Dec 29, 2022
24 Dec 9, 2022
13 Oct 14, 2022
2 Jun 21, 2022
1 Jan 14, 2022
3.2k Dec 31, 2022
97 Jan 3, 2023
109 Dec 28, 2022