Katib is a Kubernetes-native project for automated machine learning (AutoML). Katib supports Hyperparameter Tuning, Early Stopping and Neural Architecture Search

Katib is the project which is agnostic to machine learning (ML) frameworks. It can tune hyperparameters of applications written in any language of the users’ choice and natively supports many ML frameworks, such as TensorFlow, MXNet, PyTorch, XGBoost, and others.

Table of Contents

• Getting Started
• Name
• Concepts in Katib
  • Experiment
  • Suggestion
  • Trial
  • Worker Job
  • Search Algorithms
    • Hyperparameter Tuning
    • Neural Architecture Search
• Components in Katib
• Web UI
• GRPC API documentation
• Installation
  • TF operator
  • PyTorch operator
  • Katib
  • Running examples
  • Katib SDK
  • Cleanups
• Quick Start
• Who are using Katib?
• CONTRIBUTING
• Citation

Created by doctoc.

Getting Started

Follow the getting-started guide on the Kubeflow website.

Name

Katib stands for secretary in Arabic.

Concepts in Katib

For a detailed description of the concepts in Katib and AutoML, check the Kubeflow documentation.

Katib has the concepts of Experiment, Suggestion, Trial and Worker Job.

Experiment

An Experiment represents a single optimization run over a feasible space. Each Experiment contains a configuration:

1. Objective: What you want to optimize.
2. Search Space: Constraints for configurations describing the feasible space.
3. Search Algorithm: How to find the optimal configurations.

Katib Experiment is defined as a CRD. Check the detailed guide to configuring and running a Katib Experiment in the Kubeflow docs.

Suggestion

A Suggestion is a set of hyperparameter values that the hyperparameter tuning process has proposed. Katib creates a Trial to evaluate the suggested set of values.

Katib Suggestion is defined as a CRD.

Trial

A Trial is one iteration of the hyperparameter tuning process. A Trial corresponds to one worker job instance with a list of parameter assignments. The list of parameter assignments corresponds to a Suggestion.

Each Experiment runs several Trials. The Experiment runs the Trials until it reaches either the objective or the configured maximum number of Trials.

Katib Trial is defined as a CRD.

Worker Job

The Worker Job is the process that runs to evaluate a Trial and calculate its objective value.

The Worker Job can be any type of Kubernetes resource or Kubernetes CRD. Follow the Trial template guide to support your own Kubernetes resource in Katib.

Katib has these CRD examples in upstream:

• Kubernetes Job

• Kubeflow TFJob

• Kubeflow PyTorchJob

• Kubeflow MPIJob

• Tekton Pipeline

Thus, Katib supports multiple frameworks with the help of different job kinds.

Search Algorithms

Katib currently supports several search algorithms. Follow the Kubeflow documentation to know more about each algorithm.

Hyperparameter Tuning

• Random Search
• Tree of Parzen Estimators (TPE)
• Grid Search
• Hyperband
• Bayesian Optimization
• Covariance Matrix Adaptation Evolution Strategy (CMA-ES)

Neural Architecture Search

• Efficient Neural Architecture Search (ENAS)
• Differentiable Architecture Search (DARTS)

Components in Katib

Katib consists of several components as shown below. Each component is running on Kubernetes as a deployment. Each component communicates with others via GRPC and the API is defined at pkg/apis/manager/v1beta1/api.proto.

• Katib main components:
  • katib-db-manager - the GRPC API server of Katib which is the DB Interface.
  • katib-mysql - the data storage backend of Katib using mysql.
  • katib-ui - the user interface of Katib.
  • katib-controller - the controller for the Katib CRDs in Kubernetes.

Web UI

Katib provides a Web UI. You can visualize general trend of Hyper parameter space and each training history. You can use random-example or other examples to generate a similar UI. Follow the Kubeflow documentation to access the Katib UI.

GRPC API documentation

Check the Katib v1beta1 API reference docs.

Installation

For standard installation of Katib with support for all job operators, install Kubeflow. Follow the documentation:

• Kubeflow installation guide
• Kubeflow Katib guides.

If you install Katib with other Kubeflow components, you can't submit Katib jobs in Kubeflow namespace. Check the Kubeflow documentation to know more about it.

Alternatively, if you want to install Katib manually with TF and PyTorch operators support, follow these steps:

Create Kubeflow namespace:

kubectl create namespace kubeflow

Clone Kubeflow manifest repository:

git clone [email protected]:kubeflow/manifests.git
Set `MANIFESTS_DIR` to the cloned folder.
export MANIFESTS_DIR=<cloned-folder>

TF operator

For installing TF operator, run the following:

cd "${MANIFESTS_DIR}/tf-training/tf-job-crds/base"
kustomize build . | kubectl apply -f -
cd "${MANIFESTS_DIR}/tf-training/tf-job-operator/base"
kustomize build . | kubectl apply -f -

PyTorch operator

For installing PyTorch operator, run the following:

cd "${MANIFESTS_DIR}/pytorch-job/pytorch-job-crds/base"
kustomize build . | kubectl apply -f -
cd "${MANIFESTS_DIR}/pytorch-job/pytorch-operator/base/"
kustomize build . | kubectl apply -f -

Katib

Finally, you can install Katib:

git clone [email protected]:kubeflow/katib.git
make deploy

Check if all components are running successfully:

kubectl get pods -n kubeflow

Expected output:

NAME                                READY   STATUS    RESTARTS   AGE
katib-controller-858d6cc48c-df9jc   1/1     Running   1          20m
katib-db-manager-7966fbdf9b-w2tn8   1/1     Running   0          20m
katib-mysql-7f8bc6956f-898f9        1/1     Running   0          20m
katib-ui-7cf9f967bf-nm72p           1/1     Running   0          20m
pytorch-operator-55f966b548-9gq9v   1/1     Running   0          20m
tf-job-operator-796b4747d8-4fh82    1/1     Running   0          21m

Running examples

After deploy everything, you can run examples to verify the installation.

This is an example for TF operator:

kubectl create -f https://raw.githubusercontent.com/kubeflow/katib/master/examples/v1beta1/tfjob-example.yaml

This is an example for PyTorch operator:

kubectl create -f https://raw.githubusercontent.com/kubeflow/katib/master/examples/v1beta1/pytorchjob-example.yaml

Check the Kubeflow documentation how to monitor your Experiment status.

You can view your results in Katib UI. If you used standard installation, access the Katib UI via Kubeflow dashboard. Otherwise, port-forward the katib-ui:

kubectl -n kubeflow port-forward svc/katib-ui 8080:80

You can access the Katib UI using this URL: http://localhost:8080/katib/.

Katib SDK

Katib supports Python SDK:

• Check the Katib v1beta1 SDK documentation.

Run make generate to update Katib SDK.

Cleanups

To delete installed TF and PyTorch operator run kubectl delete -f on the respective folders.

To delete Katib run make undeploy.

Quick Start

Please follow the Kubeflow documentation to submit your first Katib experiment.

Who are using Katib?

Please see ADOPTERS.md.

CONTRIBUTING

Please feel free to test the system! developer-guide.md is a good starting point for developers.

Citation

If you use Katib in a scientific publication, we would appreciate citations to the following paper:

A Scalable and Cloud-Native Hyperparameter Tuning System, George et al., arXiv:2006.02085, 2020.

Bibtex entry:

@misc{george2020katib,
    title={A Scalable and Cloud-Native Hyperparameter Tuning System},
    author={Johnu George and Ce Gao and Richard Liu and Hou Gang Liu and Yuan Tang and Ramdoot Pydipaty and Amit Kumar Saha},
    year={2020},
    eprint={2006.02085},
    archivePrefix={arXiv},
    primaryClass={cs.DC}
}