arXiv:2608.14472cs.LGcs.AI2026-08

用线性规划优化神经网络结构搜索,提升训练效率和泛化性能。

LP-NAS: Linear Programming-based Neural Architecture Search

论文配图:LP-NAS: Linear Programming-based Neural Architecture Search
图 1 · 摘自论文原文
  • 基于验证损失梯度与训练损失海塞矩阵构建线性规划模型,指导架构更新方向。
  • 在CIFAR-10/100上比标准DARTS更快收敛,且早期即取得更高验证精度。
  • 适用于多种连续搜索空间,适合追求高效架构搜索的研究者与工程师。

神经网络架构搜索(NAS)旨在自动化网络设计,减少对人工经验的依赖。其中,可微分NAS因效率与准确性优于传统方法而受到关注。由于可微分NAS将搜索空间松弛为连续域,可借鉴连续优化原理。本文提出基于线性规划的神经架构搜索(LP-NAS),一种适用于广泛连续搜索空间的数学规划框架。LP-NAS利用验证损失梯度与训练损失海塞矩阵构造线性规划(LP),求解能提升泛化能力同时保持模型参数最优性的架构更新方向。沿该方向进行搜索,显著提升架构优化效率。我们提出两个计算高效的变体:S-LP-NAS与R-LP-NAS。应用于DARTS搜索空间时,得到S-LP-DARTS与R-LP-DARTS,二者在早期迭代中均实现更快收敛与更优验证性能。在CIFAR-10与CIFAR-100上的实验表明,LP-DARTS在搜索与评估阶段均优于标准DARTS。与P-DARTS、PC-DARTS、STO-DARTS等变体对比,亦展现优越性。进一步在ImageNet上验证了所发现架构的迁移能力。

原文摘要 · Abstract (English)

Neural Architecture Search (NAS) aims to automate neural network architecture design, reducing reliance on human expertise. Among the various NAS methods, differentiable NAS has gained prominence due to its efficiency and accuracy compared to conventional NAS approaches. Since differentiable NAS relaxes the architecture search space into a continuous domain, it is possible to apply principles from continuous optimization to NAS. In this paper, we propose Linear Programming-based NAS (LP-NAS), a mathematical programming-based framework for differentiable NAS that is applicable to a wide range of continuous search spaces. LP-NAS formulates a linear program (LP) using the validation-loss gradient and the training-loss Hessian to compute an architecture update direction that improves generalization while preserving the optimality of the model parameters. By following this LP-derived descent direction, LP-NAS efficiently navigates the architecture search space, leading to faster and more effective architecture optimization. We introduce two computationally efficient variants of LP-NAS, namely S-LP-NAS and R-LP-NAS. Applying LP-NAS to the Differentiable Architecture Search (DARTS) search space results in two algorithmic variants, S-LP-DARTS and R-LP-DARTS. Both variants achieve faster convergence and significantly higher validation performance during the early search iterations than the standard DARTS algorithm. Extensive experiments on CIFAR-10 and CIFAR-100 show that LP-DARTS outperforms standard DARTS in both the architecture search and evaluation phases. Additionally, we compare our approach with several DARTS variants (P-DARTS, PC-DARTS, and STO-DARTS) on the CIFAR-10 dataset and demonstrate its effectiveness. Furthermore, we validate the transferability of the discovered architectures through experiments on the ImageNet dataset.

神经架构搜索线性规划可微分搜索DARTS

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