基于迁移学习的资源受限设备自动模型创建工具
机器学习
2020-12-21 v1
摘要
随着机器学习的增强，许多工具被设计出来以协助开发者轻松创建其机器学习模型。本文提出一种利用迁移学习自动创建此类面向受限设备的定制模型的新方法，无需编写任何机器学习代码。我们分享了自动模型创建工具的架构以及其使用YAMNet和MobileNetV2等预训练模型作为特征提取器所创建的CNN模型。最后，我们通过创建自动图像与音频分类器展示了该工具所创建模型的准确率和内存占用，并报告了使用Stanford Cars和ESC-50数据集的实验结果。
引用
@article{arxiv.2012.10056,
title = {Transfer Learning Based Automatic Model Creation Tool For Resource Constraint Devices},
author = {Karthik Bhat and Manan Bhandari and ChangSeok Oh and Sujin Kim and Jeeho Yoo},
journal= {arXiv preprint arXiv:2012.10056},
year = {2020}
}
备注
7 pages, 10 figures