中文

ResBuilder:基于残差结构的深度自动化学习

机器学习 2023-08-17 v1 计算机视觉与模式识别

摘要

本工作中,我们开发了一种称为ResBuilder的神经架构搜索算法,可从零开始构建ResNet架构,以适中的计算成本实现高准确率。它也可用于修改现有架构,并具备移除和插入ResNet块的能力,从而在ResNet架构空间中搜索合适架构。我们在不同图像分类数据集上的实验表明,与现成的ResNet相比,ResBuilder在节省计算成本的同时取得了接近最先进(SOTA)的性能。值得注意的是,我们曾在CIFAR10上调整一次参数,便为所有其他数据集得出合适的默认选择。我们通过将该方法以默认参数应用于一个专有的欺诈检测数据集,证明了该特性甚至可推广至工业应用。

关键词

引用

@article{arxiv.2308.08504,
  title  = {ResBuilder: Automated Learning of Depth with Residual Structures},
  author = {Julian Burghoff and Matthias Rottmann and Jill von Conta and Sebastian Schoenen and Andreas Witte and Hanno Gottschalk},
  journal= {arXiv preprint arXiv:2308.08504},
  year   = {2023}
}