互联网资讯 / 人工智能 · 2024年4月4日

OMGD GAN 压缩算法实现显著算力降耗

该团队近期开源了一项代号为OMG D的压缩技术。这是自研的GAN压缩算法,在保证生成效果不变的前提下,算力消耗最低可以减少到原来的1/46。据悉,相关论文已入选国际计算机视觉会议ICCV 2021。

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

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

OMG D(Online Multi-Granularity Distillation)意为“在线多粒度蒸馏”。据论文显示,该算法能在训练过程中灵活优化并压缩GAN模型,从而实现更好的图像效果和更低的计算成本。

目前,OMG D 压缩算法已在部分产品场景中落地,相关技术代码也已公开在开源社区。

根据公开报道,目前该团队已开源了机器学习平台、联邦学习平台、高性能分布式训练框架等重量级项目。