专栏名称: 机器学习研究会
机器学习研究会是北京大学大数据与机器学习创新中心旗下的学生组织,旨在构建一个机器学习从事者交流的平台。除了及时分享领域资讯外,协会还会举办各种业界巨头/学术神牛讲座、学术大牛沙龙分享会、real data 创新竞赛等活动。
目录
相关文章推荐
爱可可-爱生活  ·  晚安~ #晚安# -20250122224548 ·  16 小时前  
爱可可-爱生活  ·  本文证明了多种生成模型的ELBO在驻点处会收 ... ·  昨天  
宝玉xp  ·  转发微博-20250122004120 ·  昨天  
黄建同学  ·  腾讯混元 3D ... ·  昨天  
51好读  ›  专栏  ›  机器学习研究会

【论文】(论文+代码)加速的PixelCNN ++,图像生成效率提升了183倍

机器学习研究会  · 公众号  · AI  · 2017-02-22 22:00

正文



点击上方“机器学习研究会”可以订阅哦
摘要
 

转自:视觉机器人

论文《FAST GENERATION FOR CONVOLUTIONAL AUTORGRESSIVE MODELS》摘要:

Convolutional autoregressive models have recently demonstrated state-of-the-art performance on a number of generation tasks. While fast, parallel training methods have been crucial for their success, generation is typically implemented in a näıve fashion where redundant computations are unnecessarily repeated. This results in slow generation, making such models infeasible for production environments. In this work, we describe a method to speed up generation in convolutional autoregressive models. The key idea is to cache hidden states to avoid redundant computation. We apply our fast generation method to the Wavenet and PixelCNN++ models and achieve up to 21x and 183x speedups respectively.


论文链接:

https://openreview.net/pdf?id=rkdF0ZNKl


代码链接:

https://github.com/PrajitR/fast-pixel-cnn


原文链接:

http://weibo.com/5501429448/EwI42zZzt?type=comment#_rnd1487752595359

“完整内容”请点击【阅读原文】
↓↓↓