论文标题

使用神经普通微分方程的学习锥束CT重建

Learned Cone-Beam CT Reconstruction Using Neural Ordinary Differential Equations

论文作者

Thies, Mareike, Wagner, Fabian, Gu, Mingxuan, Folle, Lukas, Felsner, Lina, Maier, Andreas

论文摘要

学到的反问题的迭代重建算法提供了将有关问题的分析知识与从数据中学到的模块结合在一起的灵活性。这样,它们实现了高重建性能,同时确保与测量数据保持一致。在计算机断层扫描中,由于训练此类模型所需的高度高GPU记忆,将这种方法从2D风扇梁扩展到3D锥束数据很具有挑战性。本文建议使用神经普通微分方程来通过数值集成在残留公式中解决重建问题。对于培训,无需通过几个展开的网络块或求解器的内部进行倒退。取而代之的是,在神经ode设置中获得非常有效的内存,允许在单个消费者图形卡上进行训练。与最佳性能的经典迭代重建算法相比,该方法能够将均方根误差降低30%以上,即使在稀疏的视图方案中,也会产生高质量的锥形束重建。

Learned iterative reconstruction algorithms for inverse problems offer the flexibility to combine analytical knowledge about the problem with modules learned from data. This way, they achieve high reconstruction performance while ensuring consistency with the measured data. In computed tomography, extending such approaches from 2D fan-beam to 3D cone-beam data is challenging due to the prohibitively high GPU memory that would be needed to train such models. This paper proposes to use neural ordinary differential equations to solve the reconstruction problem in a residual formulation via numerical integration. For training, there is no need to backpropagate through several unrolled network blocks nor through the internals of the solver. Instead, the gradients are obtained very memory-efficiently in the neural ODE setting allowing for training on a single consumer graphics card. The method is able to reduce the root mean squared error by over 30% compared to the best performing classical iterative reconstruction algorithm and produces high quality cone-beam reconstructions even in a sparse view scenario.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源