Oct, 2023

TS-ENAS:基于细胞的网络架构搜索的两阶段演化

TL;DR我们提出了一种基于叠加单元的神经网络结构搜索的两阶段进化方法(TS-ENAS),包括基于堆叠单元的一阶段搜索和对这些单元进行调整的二阶段。在算法中,我们设计了一种新的基于单元的搜索空间和一种有效的两阶段编码方法来表示单元和神经网络结构。此外,我们设计了一种基于单元的权重继承策略来初始化网络的权重,这显著减少了算法的运行时间。我们在四个图像分类数据集Fashion-MNIST、CIFAR10、CIFAR100和ImageNet上对所提出的方法进行了广泛的测试和比较,并与包括手工设计网络和NAS网络在内的22个最新算法进行了比较。实验结果表明,TS-ENAS能更有效地找到性能相当的神经网络结构。