perf4sight: a toolflow to model CNN training performance on Edge GPUs
File(s)2108.05580v1.pdf (799.44 KB)
Accepted version
Author(s)
Rajagopal, Aditya
Bouganis, Christos-Savvas
Type
Conference Paper
Abstract
The increased memory and processing capabilities of today’s edge devices create opportunities for greater edge intelligence. In the domain of vision, the ability to adapt a Convolutional Neural Network’s (CNN) structure and parameters to the input data distribution leads to systems with lower memory footprint, latency and power consumption. However, due to the limited compute resources and memory budget on edge devices, it is necessary for the system to be able to predict the latency and memory footprint of the training process in order to identify favourable training configurations of the network topology and device combination for efficient network adaptation. This work proposes perf4sight, an automated methodology for developing accurate models that predict CNN training memory footprint and latency given a target device and network. This enables rapid identification of network topologies that can be retrained on the edge device with low resource consumption. With PyTorch as the framework and NVIDIA Jetson TX2 as the target device, the developed models predict training memory footprint and latency with 95% and 91% accuracy respectively for a wide range of networks, opening the path towards efficient network adaptation on edge GPUs.
Date Issued
2021-11-24
Date Acceptance
2021-10-11
Citation
2021 IEEE/CVF International Conference on Computer Vision Workshops (ICCVW), 2021, pp.963-971
ISBN
978-1-6654-0191-3
ISSN
2473-9936
Publisher
IEEE COMPUTER SOC
Start Page
963
End Page
971
Journal / Book Title
2021 IEEE/CVF International Conference on Computer Vision Workshops (ICCVW)
Copyright Statement
Copyright © 2021 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.
Identifier
https://www.webofscience.com/api/gateway?GWVersion=2&SrcApp=PARTNER_APP&SrcAuth=LinksAMR&KeyUT=WOS:000739651101005&DestLinkType=FullRecord&DestApp=ALL_WOS&UsrCustomerID=a2bf6146997ec60c407a63945d4e92bb
Source
18th IEEE/CVF International Conference on Computer Vision (ICCV)
Subjects
Computer Science
Computer Science, Artificial Intelligence
Computer Science, Interdisciplinary Applications
Imaging Science & Photographic Technology
Science & Technology
Technology
Publication Status
Published
Start Date
2021-10-11
Finish Date
2021-10-17
Coverage Spatial
Montreal, BC, Canada
Date Publish Online
2021-11-24