2 个月前
Auto-DeepLab:用于语义图像分割的层次神经架构搜索
Chenxi Liu; Liang-Chieh Chen; Florian Schroff; Hartwig Adam; Wei Hua; Alan Yuille; Li Fei-Fei

摘要
近期,神经架构搜索(Neural Architecture Search, NAS)在大规模图像分类中成功识别出超越人工设计的神经网络架构。本文研究了用于语义图像分割的NAS。现有的工作通常集中在搜索可重复的单元结构,而手动设计控制空间分辨率变化的外部网络结构。这种选择简化了搜索空间,但对于表现出更多网络级架构变化的密集图像预测来说,逐渐成为一个问题。因此,我们提出除了搜索单元级结构外,还应搜索网络级结构,从而形成一个分层的架构搜索空间。我们介绍了一个包含许多流行设计的网络级搜索空间,并开发了一种允许高效基于梯度的架构搜索的方法(在Cityscapes图像上使用3个P100 GPU运行3天)。我们在具有挑战性的Cityscapes、PASCAL VOC 2012和ADE20K数据集上展示了所提方法的有效性。Auto-DeepLab是我们专门为语义图像分割设计的架构,在没有任何ImageNet预训练的情况下达到了最先进的性能。