{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!
281 Dec 21, 2022
22 Nov 23, 2022
32 Jun 14, 2022
192 Dec 20, 2022
184 Dec 23, 2022
3 Sep 18, 2022
44 Nov 17, 2021
97 Dec 23, 2022
47 Nov 25, 2022
3.4k Jan 07, 2023
769 Dec 25, 2022
90 Dec 01, 2022
3 Nov 09, 2021
7.7k Jan 05, 2023
482 Dec 18, 2022
78 Nov 02, 2022
0 Jan 04, 2022
3 Dec 17, 2022
957 Dec 21, 2022
17 Dec 22, 2022