中文

基于连续编码的高效多保真度多目标神经网络结构搜索表示

机器学习 2025-09-03 v1 人工智能

摘要

神经网络结构搜索(NAS)是一种自动化设计优化体系结构的吸引力方法,但在针对多个重要且相互冲突的目标进行优化时,受限于高计算预算。为此,提出一种自适应的 Co-Kriging 辅助多保真度多目标 NAS 算法,通过引入基于聚类的局部多保真度填充抽样策略来进一步降低NAS的计算成本,实现对搜索空间的高效探索以加快收敛速度。该算法通过一种新颖的连续编码方法来表示每个 cell 中节点的连接关系,这种编码方法基于通用的 cell-based U-Net 主干网络,进而减少了搜索维度(变量数量)。结果表明,在三个数值基准测试中(包括2D Darcy 流回归问题和CHASE_DB1生物医学图像分割问题),在有限计算预算下,所提出的NAS算法超越了以往发表的领先方法。随后,该方法被用于创建用于城市建模的风速回归模型,该模型能够以较低的计算复杂度实现良好的预测。进一步分析表明,NAS算法独立识别了支配优越U-Net体系结构的原理,例如每个 cell Incorporate 信息来自先前 cell 的重要性。

关键词

引用

@article{arxiv.2509.01943,
  title  = {A Continuous Encoding-Based Representation for Efficient Multi-Fidelity Multi-Objective Neural Architecture Search},
  author = {Zhao Wei and Chin Chun Ooi and Yew-Soon Ong},
  journal= {arXiv preprint arXiv:2509.01943},
  year   = {2025}
}