可逆扩散重建模型

提速14倍

以可逆两级设计将扩散采样过程直接改造为端到端重建网络,在压缩感知重建中兼得高精度与低显存。

Bin, Chen · Zhenyu, Zhang · Weiqi, Li · Zhao, Chen · Jiwen, Yu · Shijie, Zhao · 陈杰 · 长江

IEEE Transactions on Pattern Analysis and Machine Intelligence 2025

参数信息

推理加速
14.54 times
PSNR 提升
10.09 dB
GPU 内存降低
93.8 %
PSNR 提升
2.64 dB

技术优势

推理快14倍

将扩散采样过程重新用作重建模型,端到端微调,推理速度相比 DDNM 提升14.54倍。

显存降低93.8%

通过两级可逆设计,训练时大部分中间特征被清除,GPU内存降低高达93.8%。

精度提升10dB

相比现有最优压缩感知网络,PSNR 提升最高2.64dB;相比 DDNM,提升最高10.09dB。

端到端可微调

无需从头训练,可将大规模扩散模型作为重建模型进行端到端微调,部署更便捷。

应用场景