Motivation
Computing resources on cloud such as [Amazon AWS](https://aws.amazon.com/cn/)、[Baidu Cloud](https://cloud.baidu.com/) have multi-tenancy. Deep learning model training and inference with elastic resources will be common on cloud. We propose Elastic Deep Learning (EDL) that makes training and inference of deep learning models on cloud easier and more efficient. Now EDL is an incubation-stage project of the [LF AI Foundation](https://lfai.foundation).Installation
EDL package support python2.7/3.6/3.7. You can install with ```pip install paddle_edl```. But we highly **recommend** you use it in our docker: ``` docker pull hub.baidubce.com/paddle-edl/paddle_edl:latest-cuda9.0-cudnn7 nvidia-docker run -name paddle_edl hub.baidubce.com/paddle-edl/paddle_edl:latest-cuda9.0-cudnn7 /bin/bash ```Latest Release(0.3.1)
- Support elastic training with inference type services during training, such as knowledge distillation - Inference type services are automatically registered through service discovery in EDL - Knowledge distillation examples in computer vision and natural language processingQuick start Demo
- Install Paddle Serving ``` bash pip install paddle-serving-server-gpu ``` - The Teacher Model: [ResNeXt101_32x16d_wsl](https://github.com/facebookresearch/WSL-Images). Start teacher on gpu 1. ``` bash cd example/distill/resnet wget --no-check-certificate https://paddle-edl.bj.bcebos.com/distill_teacher_model/ResNeXt101_32x16d_wsl_model.tar.gz tar -zxf ResNeXt101_32x16d_wsl_model.tar.gz python -m paddle_serving_server_gpu.serve \ --model ResNeXt101_32x16d_wsl_model \ --mem_optim \ --port 9898 \ --gpu_ids 1 ``` - The Student Model: [ResNet50_vd](https://arxiv.org/pdf/1812.01187.pdf)(that is ResNet-D in paper). Train student on gpu 0. ``` bash python -m paddle.distributed.launch --selected_gpus 0 \ ./train_with_fleet.py \ --model=ResNet50_vd \ --data_dir=./ImageNet \ --use_distill_service=True \ --distill_teachers=127.0.0.1:9898 ``` - To run distillation on clusters, please reference [Run EDL distillation training](./example/distill/README.md) - Performance benchmark on industrial cluster | mode | teacher resource | student resource | total batch size | acc1 | acc5 | speed(img/s) | | :----: | :-----: | :----: | :----: | :----: | :----: | :----: | | pure train | None | 8 * v100 | 256 | 77.1 | 93.5 | 1828 | | teacher and student on the same gpus | 8 * v100 | 8 * v100 | 256 | 79.0 | 94.3 | 656 | | EDL service distill | 40 * P4 | 8 * v100 | 256 | 79.0 | 94.5 | 1514 |About Knowledge Distillation in EDL
- Theory: [Distilling the Knowledge in a Neural Network](https://arxiv.org/abs/1503.02531) - Knowledge distillation consists of two parts in general, i.e. strong teachers and weak students. - Student model learns from a teacher or mixture-of-teachers model's feed-forward results to achieve better results. - Application scenarios of EDL knowledge distillation - Teacher models and student models are running on the same GPU devices that training throughputs are not maximized - Offline GPU cluster has limited resources but some online GPU resources can be used during training. - Heterogenous teacher models can improve student model's performance but are hard to deploy on a single GPU card due to memory limitation. - Computation burden of teacher models and student models is hard to balance to maximize the training throughputs. - Solution: - Deploy teacher models as online inference service through [Paddle Serving](https://github.com/PaddlePaddle/Serving) - Online inference services are elastic and are registered to EDL service management modules. - Dynamical adaptation of teacher models' online instance to maximize students' training throughputs and resource utilization.
Release 0.2.0
Checkpoint based elastic training on multiple GPUs
- We have several training nodes running on each GPU. - A master node is responsible for checkpoint saving and all the other nodes are elastic nodes. - When elastic nodes join or leave current training job, training hyper-parameter will be adjusted automatically. - Newly comming training nodes will load checkpoint from remote FS automatically. - A model checkpoint is saved every serveral steps given by user
