《计算机应用》唯一官方网站 ›› 2026, Vol. 46 ›› Issue (8): 2411-2420.DOI: 10.11772/j.issn.1001-9081.2026010044

• 人工智能 • 上一篇    下一篇

扩散引导的进化神经架构搜索算法

汤泽宇, 温晓易, 谢在鹏()   

  1. 河海大学 计算机与软件学院,南京 211100
  • 收稿日期:2026-01-20 修回日期:2026-03-19 接受日期:2026-04-02 发布日期:2026-04-23 出版日期:2026-08-10
  • 通讯作者: 谢在鹏
  • 作者简介:汤泽宇(2000—),男,江苏宿迁人,硕士研究生,CCF会员,主要研究方向:神经架构搜索、进化计算
    温晓易(2004—),男,福建龙岩人,主要研究方向:神经架构搜索
    谢在鹏(1982—),男,江苏扬州人,副教授,博士,CCF会员,主要研究方向:神经架构搜索、进化计算、生成式模型。
  • 基金资助:
    水灾害防御全国重点实验室“一带一路”水资源与可持续发展科技基金资助项目(2021490811)

Diffusion-guided evolutionary neural architecture search algorithm

Zeyu TANG, Xiaoyi WEN, Zaipeng XIE()   

  1. College of Computer Science and Software Engineering,Hohai University,Nanjing Jiangsu 211100,China
  • Received:2026-01-20 Revised:2026-03-19 Accepted:2026-04-02 Online:2026-04-23 Published:2026-08-10
  • Contact: Zaipeng XIE
  • About author:TANG Zeyu, born in 2000, M. S. candidate. His research interests include neural architecture search, evolutionary computation.
    WEN Xiaoyi, born in 2004. His research interests include neural architecture search.
  • Supported by:
    National Key Laboratory of Water Disaster Prevention “Belt and Road” Water Resources and Sustainable Development Science and Technology Fund(2021490811)

摘要:

进化算法(EA)在神经架构搜索(NAS)中因具备较强全局探索能力而被广泛采用。然而,传统变异方式常缺乏性能信息引导,且局部开发能力有限,导致搜索效率与架构质量提升受限。因此,提出一种扩散引导的进化NAS (DGEA)算法,利用分布感知与偏置控制机制共同提升变异质量和搜索能力。该算法的核心是利用离散扩散模型生成结构化扰动,通过从高性能架构分布中学习到的结构先验,执行可控的噪声添加与条件去噪以生成新的高性能架构。同时,引入自适应变异策略动态平衡探索与利用;此外,集成一个轻量级的和声搜索(HS)算子作为局部增强模块微调架构邻域,以抑制早熟收敛。在NAS-Bench-101、NAS-Bench-201以及ImageNet100迁移任务上的实验结果表明,DGEA显著提升了搜索效率与效果。在相近的计算预算下,相较于EG-NAS (Neural Architecture Search with Fast Evolutionary Exploration)、GEA (Guided Evolutionary NAS with Efficient Performance Estimation)和REA (Regularized Evolution for Image Classifier Architecture Search),DGEA达到相同准确率水平所需架构评估数量减少10%~84%,准确率整体提升0.28~0.42个百分点。消融实验证实了扩散引导变异(DGM)机制和HS模块的贡献以及协同作用。通过整合分布感知的生成式变异与自适应HS局部优化,DGEA有效提升了进化搜索的动态性能与收敛稳定性。

关键词: 神经架构搜索, 进化算法, 扩散引导变异, 分布建模, 和声搜索, 局部优化

Abstract:

Evolutionary Algorithms (EAs) are widely adopted in Neural Architecture Search (NAS) due to their strong global exploration capability. However, traditional mutation ways often lack performance information guidance and have limited local exploitation capability, thereby constraining search efficiency and architecture quality. Therefore, this paper proposed a Diffusion-Guided Evolutionary NAS (DGEA) algorithm to enhance mutation quality and search capability jointly via distribution-aware and bias control mechanisms. The core of DGEA is to use a discrete diffusion model to generate structural perturbations, aiding evolutionary search process. By learning structural priors from the distribution of high-performance architectures, DGEA performed controlled noise addition and conditional denoising to create new high-performance architecture. At the same time, DGEA introduced an adaptive mutation strategy to balance exploration and exploitation dynamically. Besides, DGEA integrated a lightweight Harmony Search (HS) operator as local enhancement module for fine-tuning neighborhoods, thereby suppressing premature convergence. Experimental results on NAS-Bench-101, NAS-Bench-201, and ImageNet100 transfer tasks show that DGEA boosts search efficiency and effectiveness significantly. Under similar computational budgets, compared to EG-NAS (Neural Architecture Search with Fast Evolutionary Exploration), GEA (Guided Evolutionary Neural Architecture Search with Efficient Performance Estimation) and REA (Regularized Evolution for Image Classifier Architecture Search), DGEA reduces the architecture evaluations needed to reach the same accuracy level by 10%-84% with the overall accuracy gains of 0.28 to 0.42 percentage points. Ablation studies verify the contributions and synergistic effect of Diffusion-Guided Mutation (DGM) mechanism and HS module. It can be seen that by integrating distribution-aware generative mutation with adaptive HS local optimization, DGEA improves evolutionary search’s dynamic performance and convergence stability.

Key words: Neural Architecture Search (NAS), Evolutionary Algorithm (EA), Diffusion-Guided Mutation (DGM), distribution modeling, Harmony Search (HS), local optimization

中图分类号: