跳到正文
arXiv · cs.LG· Yize Sun, Hanzhang Ma, Zixin Wu, Xuechun Li, Volker Tresp, Yunpu Ma·· 3 小时前

DIFFNAS:基于扩散模型与零成本评估的无预测器神经架构搜索

DIFFNAS: Predictor-Free Neural Architecture Search via Diffusion Model and Zero-Cost Evaluation

AI 导读

DIFFNAS 提出一种无需性能预测器的 NAS 框架,用离散图扩散模型结合 MeCo、SWAP 等免训练指标引导架构生成。在 NAS-Bench-201 上,它为 CIFAR-10 和 CIFAR-100 找到最优架构,准确率分别为 94.37% 和 73.51%,搜索成本相比 DiffusionNAG 最多降低 17.5x。

来源:arXiv · cs.LG · arxiv.org