arrow
返回

MMDataLoader: Reusing Preprocessed Data Among Concurrent Model Training Tasks

delete2024-02-01
delete2
PRE
AI
金
金海 (Hai Jin)
Z
Zhanyang Zhu
L
Ligang He
Y
Yuhao Li
Y
Yusheng Hua
X
Xuanhua Shi *
DOI:10.1109/TC.2023.3336161delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Data preprocessing plays an important role in deep learning, which directly affects the training efficiency. Data preprocessing is performed on the CPU. The preprocessed data are then fed to the models that are trained on the GPU. We observe that data preprocessing on the CPU can potentially create a bottleneck in the entire process of a model training task. In order to tackle this issue, we have developed MMDataLoader, which enables reusing preprocessed data among multiple model training tasks. MMDataLoader automatically constructs a data preprocessing pipeline based on each task's specific preprocessing workflow, allowing for maximum data reuse and reduced computing workload on the CPU. Unlike conventional data loaders that operate at the task level and provide data provision services to specific training tasks, MMDataLoader operates at the server level and provides data for all concurrently running tasks. We have conducted extensive experiments. The results show that MMDataLoader can significantly increase preprocessing throughput without affecting model convergence when compared to conventional methods where model training tasks are executed concurrently. For instance, with three tasks running, the preprocessing throughput can increase by 1.6x to 3.15x, depending on the tasks being executed and the proportion of preprocessing operations that are shared among them.
Keyword:
Deep learning
Training data
preprocessing
concurrent training
branch pipeline

期刊

IEEE Transactions on Computers 封面图
IEEE Transactions on Computers
IF:
3.8
论文数:
5.4K
被引数:
9.8K

机构

U
University of Warwick
学者数:
2.2W
论文数: 2.2W
被引数: 85
引用论文

引用论文

Ada-boundary: accelerating DNN training via adaptive boundary batch selection
err2020-09-04
err11
errOAAI
errSong, Hwanjun; Kim, Sundong; Kim, Minseok; Lee, Jae-Gil
err分享
err收藏
err分享
err收藏
Metal hydride technologies for renewable energy
err2019-09-04
err0
errOAAI
errAlexey Kazakov; Dmitry Blinov; Ivan Romanov; Dmitry Dunikov; Vasily Borzenko
err分享
err收藏
学者 查看更多内容