Valohai Model Hub is a centralized solution for managing machine learning models throughout their entire lifecycle.
Key features include:
- Unified view to all your models and their versions
- Lineage tracing to see what data was used to create each model version
- Model states (pending, approved, rejected) for seamless workflow management
- Training metrics to track and compare model version performance
- Model access control to manage who can see and use the models
- Centralized documentation for use cases, explanations, and usage instructions
- Automated workflows for reproducibility and transparency
Unified View
Model Hub offers a centralized view of all your models. Valohai UI allows you to easily access detailed information, such as model version history, training performance, and lineage. You can compare versions, download artifacts, or trigger workflows without needing to search through multiple tools.
Lineage Tracing
Model Hub automatically tracks the entire lineage of each model version—from data and parameters to code and pipeline steps. By leveraging Valohai’s reproducibility-first design, every model’s origin can be traced back through the pipeline that generated it.
You can see the exact data, code, and environment that was used to create the model.
Model States
Model Hub allows you to manage the state of each model version. You can mark a model as “Pending” for review, “Approved” for production, or “Rejected” if it doesn’t meet the requirements.
This workflow management feature ensures that only the best models are promoted to production.
Training Metrics
Valohai makes performance tracking seamless by capturing training metrics automatically, without the need for a custom centralized logging setup. Model Hub leverages this to provide an easy-to-understand view of metrics like accuracy, loss, and F1 scores across model versions.
This enables teams to monitor model progress, identify trends, and make informed decisions about iteration and refinement.
Model Access Control
Admins can control who can view or modify models and their versions by setting access permissions. This prevents unauthorized changes and ensures team members have the right access to the right models.
This feature is especially useful for organizations with strict privacy and security requirements.
Centralized Documentation
Clear documentation is essential for collaboration and compliance. Model Hub allows you to document each model and model version with its intended use, limitations, and other relevant information, ensuring that all team members have the context they need.
Automated Workflows
To make deployment smoother, Model Hub supports triggers that can be used to automate workflows, like deploying a new model version or rolling back to a previous one. This automation support ensures seamless transitions from development to production.