Managing ML Pipelines: Feature Stores and the Coming Wave of Embedding Ecosystems
Not on the map. Why?
Abstract (arXiv)
The industrial machine learning pipeline requires iterating on model features, training and deploying models, and monitoring deployed models at scale. Feature stores were developed to manage and standardize the engineer's workflow in this end-to-end pipeline, focusing on traditional tabular feature data. In recent years, however, model development has shifted towards using self-supervised pretrained embeddings as model features. Managing these embeddings and the downstream systems that use them introduces new challenges with respect to managing embedding training data, measuring embedding quality, and monitoring downstream models that use embeddings. These challenges are largely unaddressed in standard feature stores. Our goal in this tutorial is to introduce the feature store system and discuss the challenges and current solutions to managing these new embedding-centric pipelines.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 16 of its 29 references that are in the map. First posted on arXiv 11 Aug 2021.
Builds on
16 of 29 references.
References · 16
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,366
- 2020ColBERT: Efficient and Effective Passage Search via Contextualized Late Interaction over BERTcited by 1,160
- 2020Beyond Accuracy: Behavioral Testing of NLP models with CheckListcited by 979
- 2017Snorkel: Rapid Training Data Creation with Weak Supervisioncited by 370
- 2019Deep Learning Recommendation Model for Personalization and Recommendation Systemscited by 327
- 2021Robustness Gym: Unifying the NLP Evaluation Landscapecited by 107
- 2020Model Patching: Closing the Subgroup Performance Gap with Data Augmentationcited by 82
- 2019Rare Words: A Major Problem for Contextualized Embeddings And How to Fix it by Attentive Mimickingcited by 40
- 2018Factors Influencing the Surprising Instability of Word Embeddingscited by 31
- 2019Ludwig: a type-based declarative deep learning toolboxcited by 24
- 2020Bootleg: Chasing the Tail with Self-Supervised Named Entity Disambiguationcited by 24
- 2019Slice-based Learning: A Programming Model for Residual Learning in Critical Data Slicescited by 23
4 more
- 2019Overton: A Data System for Monitoring and Improving Machine-Learned Productscited by 21
- 2019On the Downstream Performance of Compressed Word Embeddingscited by 13
- 2020Understanding the Downstream Instability of Word Embeddingscited by 4
- 2020ARDA: Automatic Relational Data Augmentation for Machine Learningcited by 3
Built on by
Citing papers per year: 2021: 2; 2022: 1; 2023: 2; 2024: 0; 2025: 0; 2026: 1.
6 papers.
- 2021Looper: An end-to-end ML platform for product decisionscited by 9
- 2021Fix your Models by Fixing your Datasetscited by 3
- 2022SHiFT: An Efficient, Flexible Search Engine for Transfer Learningcited by 3
- 2023Efficient Multi-stage Inference on Tabular Datacited by 0
- 2023Managed Geo-Distributed Feature Store: Architecture and System Designcited by 0
- 2026Learning Source Acquisition Policies by Offline Planningcited by 0
- 2026Learning Source Acquisition Policies by Offline Planning13 Sep 2026
- 2023Managed Geo-Distributed Feature Store: Architecture and System Design31 May 2023
- 2023Efficient Multi-stage Inference on Tabular Data21 Mar 2023
- 2022SHiFT: An Efficient, Flexible Search Engine for Transfer Learning4 Apr 2022
- 2021Fix your Models by Fixing your Datasets15 Dec 2021
- 2021Looper: An end-to-end ML platform for product decisions14 Oct 2021