小波条件扩散模型

高速低耗图像增强

利用小波变换大幅加速扩散模型推理,并设计高频恢复模块提升细节保真度,实现稳定且高效的弱光图像增强。

Hai, Jiang · Luo, Ao · 范浩强 · 韩松臣 · Liu, Shuaicheng

ACM Transactions on Graphics 2023

技术优势

推理大幅加速

在小波域中执行扩散过程,利用小波变换的信息保持特性压缩空间维度,显著减少推理步数和计算量,同时不牺牲信息。

输出稳定可控

训练期间同时进行前向扩散和去噪,使模型学会稳定去噪,避免推理时产生随机混乱的内容,输出结果更加稳定一致。

细节恢复更精细

引入高频恢复模块,利用垂直和水平方向细节补偿对角线方向信息,实现更精细的细节重建,避免细节丢失。

应用场景