Transfer Learning Based Automatic Model Creation Tool For Resource Constraint Devices

Autor:	Bhat, Karthik, Bhandari, Manan, Oh, ChangSeok, Kim, Sujin, Yoo, Jeeho
Rok vydání:	2020
Předmět:	Computer Science - Machine Learning
Druh dokumentu:	Working Paper
Popis:	With the enhancement of Machine Learning, many tools are being designed to assist developers to easily create their Machine Learning models. In this paper, we propose a novel method for auto creation of such custom models for constraint devices using transfer learning without the need to write any machine learning code. We share the architecture of our automatic model creation tool and the CNN Model created by it using pretrained models such as YAMNet and MobileNetV2 as feature extractors. Finally, we demonstrate accuracy and memory footprint of the model created from the tool by creating an Automatic Image and Audio classifier and report the results of our experiments using Stanford Cars and ESC-50 dataset. Comment: 7 pages, 10 figures
Databáze:	arXiv
Externí odkaz:	http://arxiv.org/abs/2012.10056 Zobrazit plný text záznamu View this record from Arxiv