arXiv:2504.20080cs.LGcs.AI2025-04被引 3

通过删减与模仿双策略,高效生成性能优越的轻量化神经网络架构。

DNAD: Differentiable Neural Architecture Distillation

  • 基于删减思想的SNPS算法,逐步压缩超网络以发现多样结构。
  • 融合知识蒸馏的DNAD在ImageNet上达23.7%错误率,仅600万参数。
  • 适合追求模型效率与精度平衡的研究者或工程落地场景。

为满足高效神经网络设计中性能与计算复杂度之间的权衡需求,本文提出可微神经架构蒸馏(DNAD)算法,其核心为‘删减’与‘模仿’两种搜索机制。首先,基于可微架构搜索(DARTS)框架,提出超网络渐进收缩(SNPS)算法,通过强制动态超网络从密集结构逐步稀疏化,实现同一类型细胞间拓扑各异的神经架构搜索,突破传统DARTS仅生成单一简单结构的局限。其次,结合知识蒸馏(KD)的优势,将教师网络与超网络的行为差异最小化,有效避免单级DARTS的过拟合问题。在CIFAR-10和ImageNet分类任务上的实验表明,SNPS与DNAD均能生成一组性能优异的架构,在参数量与浮点运算量(FLOPs)更少的前提下达到甚至低于现有方法的误差率。特别地,DNAD在ImageNet上以600万参数、5.98亿FLOPs实现23.7%的Top-1错误率,显著优于多数基于DARTS的方法。

原文摘要 · Abstract (English)

To meet the demand for designing efficient neural networks with appropriate trade-offs between model performance (e.g., classification accuracy) and computational complexity, the differentiable neural architecture distillation (DNAD) algorithm is developed based on two cores, namely search by deleting and search by imitating. Primarily, to derive neural architectures in a space where cells of the same type no longer share the same topology, the super-network progressive shrinking (SNPS) algorithm is developed based on the framework of differentiable architecture search (DARTS), i.e., search by deleting. Unlike conventional DARTS-based approaches which yield neural architectures with simple structures and derive only one architecture during the search procedure, SNPS is able to derive a Pareto-optimal set of architectures with flexible structures by forcing the dynamic super-network shrink from a dense structure to a sparse one progressively. Furthermore, since knowledge distillation (KD) has shown great effectiveness to train a compact network with the assistance of an over-parameterized model, we integrate SNPS with KD to formulate the DNAD algorithm, i.e., search by imitating. By minimizing behavioral differences between the super-network and teacher network, the over-fitting of one-level DARTS is avoided and well-performed neural architectures are derived. Experiments on CIFAR-10 and ImageNet classification tasks demonstrate that both SNPS and DNAD are able to derive a set of architectures which achieve similar or lower error rates with fewer parameters and FLOPs. Particularly, DNAD achieves the top-1 error rate of 23.7% on ImageNet classification with a model of 6.0M parameters and 598M FLOPs, which outperforms most DARTS-based methods.

神经架构搜索轻量化模型知识蒸馏可微搜索

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。