CVPRMar, 2024

利用可解释性知识构建最优神经网络架构

TL;DR通过对操作和体系结构模块的潜在嵌入与其所出现的架构的实际性能进行对齐,AutoBuild 能够为体系结构模块分配可解释的重要性评分,从而构建高性能的神经网络,无需进行搜索。通过对图像分类、分割和稳定扩散模型的实验,我们展示了 AutoBuild 可以学习构建高质量体系结构或减少搜索空间以关注相关区域,找到超过原始标记的体系结构和搜索基准的体系结构的更好结果。