arrow
返回

TBEM: Testing-Based GPU-Memory Consumption Estimation for Deep Learning

delete2022-01-01
delete1
delete
OA
AI
H
Haiyi Liu
S
Shaoying Liu *
C
Chenglong Wen
W
W. Eric Wong
DOI:10.1109/ACCESS.2022.3164510delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Deep Learning (DL) has been successfully implemented and deployed to various software service applications. During the training process of DL, a large amount of GPU computing resources is required, but it is difficult for developers to accurately calculate the GPU resources that the model may consume before running, which brings great inconvenience to the development of DL systems. Especially in today's cloud-based model training. Therefore, it is very important to estimate the GPU memory resources that the DL model may use in a certain computing framework. Existing work has focused on static analysis methods to assess GPU memory consumption, highly coupled with the framework, and lack of research on low-coupled GPU memory consumption of the framework. In this article, we propose TBEM, which is a test-based method for estimating the memory usage of the DL model. First, TBEM generates enough DL models using an orthogonal array testing strategy and a classical neural network design pattern. Then, TBEM generates DL model tested in a real environment to obtain the real-time GPU memory usage values corresponding to the model. After obtaining the data of different models and corresponding GPU usage values, the data is analyzed by regression.
Keyword:
Deep learning
Neural networks
Graphics processing units
Training
Computational modeling
Analytical models
Static analysis
Deep learning
program static analysis
automated testing
bug detection

期刊

IEEE Access 封面图
IEEE Access
IF:
3.6
论文数:
9.8W
被引数:
29.4W

机构

U
University of Texas Dallas
学者数:
5.6K
论文数: 5.0K
被引数: 15
U
university of texas system
学者数:
18.5W
论文数: 15.6W
被引数: 210
H
Hiroshima University
学者数:
2.1W
论文数: 1.5W
被引数: 1.3W
学者 查看更多机构
引用论文

引用论文

err分享
err收藏
err分享
err收藏
High-throughput machining using high average power ultrashort pulse lasers and ultrafast polygon scanner
err2016-03-04
err0
PREAI
errJoerg Schille; Lutz Schneider; André Streek; Sascha Kloetzer; Udo Loeschner
err分享
err收藏
学者 查看更多内容