字节跳动开源最新GAN压缩算法 算力消耗可减少至1/46

9月2日消息，字节跳动近期开源了一项代号为OMGD的压缩技术。这是字节自研的GAN压缩算法，在保证生成效果不变的前提下，算力消耗最低可以减少到原来的1/46。据悉，这项技术的论文已入选国际计算机视觉会议ICCV2021。

GAN是人工智能领域重要的深度学习模型，在图像生成、音乐生成和视频生成等方面应用广泛，还可以提高图像质量，实现图像风格化、图像着色等任务。漫画特效等人们常用的短视频道具，就是通过GAN实现的。

OMGD意为在线多粒度蒸馏。据字节跳动技术团队的论文显示，该算法能灵活地在训练过程中优化并压缩GAN模型，从而实现更好的图像效果和更少的计算成本。

目前，OMGD压缩算法已在抖音等产品中落地，相关技术代码也已发布在开源社区。

根据公开报道，截至目前，字节跳动已开源了机器学习平台Klever、联邦学习平台Fedlearner、高性能分布式训练框架BytePS、LightSeq推理和训练引擎等重磅项目。