同时优化网络结构与参数,用梯度下降自动发现紧凑高效模型。
Simultaneous Weight and Architecture Optimization for Neural Networks
- 用嵌入空间统一表示网络,实现结构与参数联合优化
- 在多个数据集上发现稀疏紧凑模型,性能接近传统训练结果
- 适合需要轻量化部署的场景,如边缘计算
神经网络通常通过选择架构并训练参数来完成。架构选择常依赖试错或神经架构搜索(NAS)方法,而现有NAS多采用离散步骤,先优化架构再训练参数。本文提出一种新框架,通过梯度下降同时学习网络架构与参数,从根本上改变训练流程。核心是多尺度编码器-解码器结构:编码器将功能相似的网络(无论架构与权重差异)映射到相近嵌入空间;训练时从嵌入空间随机采样网络嵌入,使用定制损失函数(含稀疏性惩罚项)进行梯度下降,解码器生成对应网络。实验表明,该框架能发现稀疏且紧凑的神经网络,在保持高精度的同时显著降低复杂度。
原文摘要 · Abstract (English)
Neural networks are trained by choosing an architecture and training the parameters. The choice of architecture is often by trial and error or with Neural Architecture Search (NAS) methods. While NAS provides some automation, it often relies on discrete steps that optimize the architecture and then train the parameters. We introduce a novel neural network training framework that fundamentally transforms the process by learning architecture and parameters simultaneously with gradient descent. With the appropriate setting of the loss function, it can discover sparse and compact neural networks for given datasets. Central to our approach is a multi-scale encoder-decoder, in which the encoder embeds pairs of neural networks with similar functionalities close to each other (irrespective of their architectures and weights). To train a neural network with a given dataset, we randomly sample a neural network embedding in the embedding space and then perform gradient descent using our custom loss function, which incorporates a sparsity penalty to encourage compactness. The decoder generates a neural network corresponding to the embedding. Experiments demonstrate that our framework can discover sparse and compact neural networks maintaining a high performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。