首页 > 新闻 > 正文
17

让AI自己调整超参数,谷歌大脑新优化器火了,自适应不同任务,83个任务训

2022-11-25 08:47:22 来源:IT之家   阅读量:8242   

还在担心如何为优化器调整更好的参数吗。

现在Google Brain又出了一个新的优化器VeLO,可以直接使用,不需要手动调整任何超级参数。

与其他人工设计的算法不同,VeLO完全基于AI,可以适应各种任务。

有网友看到了优化器进步的曙光:

亚当之后出现了很多优化师,但都失败了这个优化器可能确实表现得更好

那么,这个基于人工智能的优化器是如何构建的呢。

VeLO到底是怎么建成的。

在训练神经网络的过程中,优化器是必不可少的一部分。

但是AI模型应用广泛,用来训练AI模型的优化器还是手工设计,听起来有些不合理。

所以谷歌大脑的研究人员灵机一动:为什么不用AI做优化器。

在设计上,optimizer的原理是基于元学习的思想,即从相关任务中学习经验来帮助学习目标任务。

与迁移学习相比,元学习强调获取元知识,元知识是一种关于任务的一般性知识,可以推广到更多的任务。

基于这一思路,VeLO还会吸收渐变,自动输出参数更新,无需任何超参数调谐,适应各种需要优化的任务。

在架构上,人工智能优化器由LSTM和超级网络MLP组成。

每个LSTM负责设置多个MLP的参数,每个LSTM通过全局上下文信息相互协作。

在训练中,AI优化器采用元训练,以参数值和梯度为输入,输出需要更新的参数。

经过4000个月的TPU训练,结合各种优化任务的长处,VeLO终于诞生了。

效果优于手动调优优化器。

结果表明,VeLO对83个任务的加速效果超过了现有的一系列优化器。

y轴是加速度与Adam的比值,x轴是任务的比例。

结果表明,VeLO不仅优于不需要调整超参数的优化器,甚至优于一些仔细调整超参数的优化器:

与经典老大哥亚当相比,VeLO在所有任务上的训练速度都更快,其中超过50%的训练速度比调整自己学习速度4倍以上的亚当更快,超过14%的任务学习速度甚至快16倍。

在六种学习任务的优化效果上,VeLO在五种任务上的绩效效果与Adam相当甚至更好:

值得一提的是,VeLO这次也部署在JAX看来谷歌真的推广了这个新框架

巴特,有网友认为训练VeLO需要4000 TPU月,计算成本太高:

虽然这个进步很重要,但它甚至可以赶上GPT—3的训练量。

目前VeLO是开源的,有兴趣的朋友可以试试这个新的AI优化器。

还有一点

前段时间,哈佛的一个博士生提出了一个有趣的想法,得到了很多人的赞同:

Kareem Carr是生物统计学的博士生,他的贡献在生物论文中很常见,之前在AI论文中不多。

不知道以后会不会成为机器学习圈的新趋势。

GitHub地址:

论文地址:

参考链接:

声明:本网转发此文章,旨在为读者提供更多信息资讯,所涉内容不构成投资、消费建议。文章事实如有疑问,请与有关方核实,文章观点非本网观点,仅供读者参考。

本网站由 财经产业网 版权所有 网站地图
Copyright @ 2001-  www.catai.com.cn All Rights Reserved 财经产业网 版权所有  备案号:皖ICP备2022015281号
网站所登新闻、资讯等内容, 均为相关单位具有著作权,未经书面授权,转载注明出处  邮箱:bgm1231@sina.com