告别显存焦虑!PPDM让3D扩散模型在普通显卡上跑出SOTA
3D扩散模型做医学图像翻译效果虽好,但训练时显存随分辨率三次方暴增,一张H100都勉强。PPDM (Pixel Puzzling Diffusion Model) 用像素拼图妙招,把空间尺寸“平移”成通道数,显存从49GB直降到6.94GB,效果还更稳。不是炫技,是真正解决落地痛点的好工作。
LONGGE AI RESEARCH DAILY
每天筛选值得关注的 AI 论文,用工程视角拆解方法、实验和真正的落地价值。
目前已发表 1573 篇文章
3D扩散模型做医学图像翻译效果虽好,但训练时显存随分辨率三次方暴增,一张H100都勉强。PPDM (Pixel Puzzling Diffusion Model) 用像素拼图妙招,把空间尺寸“平移”成通道数,显存从49GB直降到6.94GB,效果还更稳。不是炫技,是真正解决落地痛点的好工作。
继去年8月聊过用高斯泼溅做无监督3D超声重建之后,本期带来一篇更硬核的工程优化论文——直接在GPU层面把4D超声重建速度拉满。作者没有引入花哨的网络结构,而是从内存访问模式下手,精准诊断出传统延迟求和算法让GPU“吃不饱”的症结,并用数据重排、混合精度和Tensor Core三招巧妙化解。实验直接在RTX 4080 Super上跑出了千赫兹帧率,比现有库快3
LONGGE AI COMMUNITY
加入「龙哥读论文」知识星球,持续获取 AI 论文、资讯、开源项目、招聘与研究思路。
加入龙哥读论文微信群:添加微信 kangjinlonghelper,备注“研究方向 + 地点 + 学校/公司 + 昵称”。
微信扫码加入知识星球