flexiflow
about
FlexiFlow is a new programming model that is aimed at automatic model switching to maximise performance in machine learning workflows.
With the increasing use of machine learning in production systems, engineers often encounter practical problems in deploying and maintaining ML models and workflows. A recurring concern is the issue of soft failures—situations where the model does not crash but delivers incorrect or degraded predictions, often due to factors like data drift. These soft failures compromise prediction accuracy and, in turn, the overall effectiveness of the deployed system. When workflows consist of multiple steps and models, such degradations must be promptly identified and addressed. Developers commonly rely on trace logs to identify problematic data batches, but exhaustively testing alternative model configurations is resource-intensive and time-consuming.
In practice, only one single model is typically selected for each step based on evaluation over training data. However, this static selection often fails to maintain high accuracy across all real-world inputs during deployment. If a single model underperforms, the entire workflow may need to be re-executed from the beginning with an alternative model, significantly increasing the management burden and latency.

As a solution we introduce FlexiFlow, a dataflow system that dynamically switches between alternate models when the current model exhibits low accuracy.
FlexiFlow learns to rank models using a novel multi-armed bandit approach that accounts for model runtimes, probability of passing user-defined assertions, and the computational structure of the ML workflow.
docs
FlexiFlow docs, add the link to the documentation present in the github repository.
code
You can find the entire repository here: Link to the github repository here. The Dockerfile used to build the FlexiFlow environment is present in root directory of the repository linked here.
papers
Add the upcoming papers here.
team
collaborators
-
Abhilash Jindal
Indian Institute of Delhi -
Todd Nief
UChicago -
Tvisha Malik
Indian Institute of Delhi -
Anshik Sahu
Indian Institute of Delhi -
Amitabh Chaudhary
UChicago -
Aaron Schein
UChicago