“您可能也喜欢此模型”：用于为未知图像数据集推荐深度学习模型的数据驱动方法
机器学习
2020-05-22 v2 计算机视觉与模式识别
信息检索
图像与视频处理
机器学习
摘要
对于一个未知（新）分类数据集，选择合适的深度学习架构通常是一个递归、耗时且费力的过程。在本研究中，我们提出一种新技术，从已知模型仓库中推荐合适的架构。此外，我们预测所推荐架构在给定未知数据集上的性能准确率，而无需训练该模型。我们提出一种模型编码器方法，以无监督方式学习深度学习架构及其超参数的定长表示。我们手动策划了一个带有对应已知深度学习模型的图像数据集仓库，并表明预测准确率是实际准确率的一个良好估计。我们讨论了所提方法在三个基准图像数据集上的意义，以及将该用于文本模态时所面临的挑战。为了进一步提高所提方法的可复现性，完整实现与训练好的模型一并公开可用。
引用
@article{arxiv.1911.11433,
title = {"You might also like this model": Data Driven Approach for Recommending Deep Learning Models for Unknown Image Datasets},
author = {Ameya Prabhu and Riddhiman Dasgupta and Anush Sankaran and Srikanth Tamilselvam and Senthil Mani},
journal= {arXiv preprint arXiv:1911.11433},
year = {2020}
}
备注
NeurIPS 2019, New in ML Group