4 个月前

DARTS:可微分架构搜索

DARTS:可微分架构搜索

摘要

本文通过将架构搜索任务公式化为可微分形式,解决了其扩展性挑战。与传统方法在离散且不可微分的搜索空间中应用进化算法或强化学习不同,我们的方法基于对架构表示的连续松弛,允许使用梯度下降高效地搜索架构。在CIFAR-10、ImageNet、Penn Treebank和WikiText-2上进行的大量实验表明,我们的算法在图像分类的高性能卷积架构和语言建模的递归架构发现方面表现出色,同时比最先进的非可微分技术快几个数量级。我们已将实现公开发布,以促进对高效架构搜索算法的进一步研究。

代码仓库

diff7/DARTS-devices
pytorch
GitHub 中提及
abcp4/MyDarts
pytorch
GitHub 中提及
liamcli/darts
pytorch
GitHub 中提及
siddikui/Fast-DARTS
pytorch
GitHub 中提及
brian-moser/dartsrenet
pytorch
GitHub 中提及
tianbaochou/CNAS
pytorch
GitHub 中提及
guoyongcs/NATv2
pytorch
GitHub 中提及
KunlinY/darts-parallel
pytorch
GitHub 中提及
IsmaelCesar/darts
pytorch
GitHub 中提及
nutellamok/advrush
pytorch
GitHub 中提及
jiangyingjunn/i-darts
pytorch
GitHub 中提及
aragakiyuiii/Gumbel-darts-master
pytorch
GitHub 中提及
JunrQ/NAS
pytorch
GitHub 中提及
guoyongcs/NAT
pytorch
GitHub 中提及
victoriaqiu/10707finalproject
pytorch
GitHub 中提及
D-X-Y/AutoDL-Projects
pytorch
GitHub 中提及
adammenges/DARTS-PyTorch
pytorch
GitHub 中提及
NVlabs/unas
pytorch
GitHub 中提及
jkooy/NAS-UDA
pytorch
GitHub 中提及
mlvc-lab/Classification-NAS
pytorch
GitHub 中提及
MarcAntoineAlex/darts
pytorch
GitHub 中提及
anonNo2/MulTeacher-KD
pytorch
GitHub 中提及
lorylei/DARTS-et
pytorch
GitHub 中提及
lippman1125/darts_notes
pytorch
GitHub 中提及
khanrc/pt.darts
pytorch
GitHub 中提及
dragen1860/DARTS-PyTorch
pytorch
GitHub 中提及
mayankk6196/darts-python
pytorch
GitHub 中提及
pipilurj/BONAS
pytorch
GitHub 中提及
osmr/imgclsmob
mxnet
GitHub 中提及
flymin/darts
pytorch
GitHub 中提及
sinamoradian/SNIP-NAS-2
pytorch
GitHub 中提及
veegee82/DARTS
GitHub 中提及
tally0818/DARTS
pytorch
GitHub 中提及
HeungChangLee/AutoML
pytorch
GitHub 中提及
D-X-Y/GDAS
pytorch
GitHub 中提及
KunlinY/darts
pytorch
GitHub 中提及
ahundt/sharpDARTS
pytorch
GitHub 中提及
DoctorLiQ/AutoML
pytorch
GitHub 中提及
grtzsohalf/pt.darts
pytorch
GitHub 中提及
ky941122/darts-experiment
pytorch
GitHub 中提及
leopard-ai/betty
pytorch
GitHub 中提及
quark0/darts
官方
pytorch
GitHub 中提及
kc-ml2/darts
pytorch
GitHub 中提及
scz233/darts
pytorch
GitHub 中提及
nikita270a/BayesianDARTS
pytorch
GitHub 中提及
D-X-Y/NAS-Projects
pytorch
GitHub 中提及
eml-eda/plinio
pytorch
GitHub 中提及
yochaiz/darts-UNIQ
pytorch
GitHub 中提及
NivNayman/XNAS
pytorch
GitHub 中提及
alphadl/darts.pytorch1.1
pytorch
GitHub 中提及

基准测试

基准方法指标
architecture-search-on-cifar-10-imageDARTS + c/o
Params: 3.4M
Percentage error: 2.83
Search Time (GPU days): 4
language-modelling-on-penn-treebank-wordDifferentiable NAS
Params: 23M
Test perplexity: 56.1
Validation perplexity: 58.3
neural-architecture-search-on-cifar-10DARTS (second order)
Parameters: 3.3
Search Time (GPU days): 4
Top-1 Error Rate: 2.76%
neural-architecture-search-on-cifar-10DARTS (first order)
Parameters: 3.3
Search Time (GPU days): 1.5
Top-1 Error Rate: 3%
neural-architecture-search-on-imagenetDARTS
Accuracy: 73.3
MACs: 595M
Params: 4.9
Top-1 Error Rate: 26.7
neural-architecture-search-on-nas-bench-201DARTS (second order)
Accuracy (Test): 16.43
Search time (s): 29902
neural-architecture-search-on-nas-bench-201DARTS (first order)
Accuracy (Test): 16.43
Search time (s): 10890

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DARTS:可微分架构搜索 | 论文 | HyperAI超神经