4 个月前

学习用于 diffeomorphic 注册的概率模型

学习用于 diffeomorphic 注册的概率模型

摘要

我们提出了一种从数据中学习低维概率变形模型的方法,该模型可用于配准和变形分析。潜在变量模型在编码空间中将相似的变形映射到彼此附近。这使得可以比较不同的变形,为任何新图像生成正常或病理性的变形,或将一个图像对的变形传输到其他任意图像上。我们的无监督方法基于变分推断。具体而言,我们使用了条件变分自编码器(CVAE)网络,并通过应用具有对称损失函数的可微指数层来约束变换为对称且微分同胚的。我们还提出了一个包含空间正则化的公式,例如基于扩散的滤波器。此外,我们的框架提供了多尺度速度场估计。我们在334例心脏电影磁共振成像(cine-MRI)数据集上进行了3D同个体配准评估。在这个数据集中,我们的方法在使用32个潜在维度的情况下表现出了一流的性能,平均DICE得分为81.2%,平均Hausdorff距离为7.3毫米,优于三种现有的先进方法,并且展示了更加规则的变形场。每次配准的平均时间为0.32秒。此外,我们可视化了所学的潜在空间,并展示了编码后的变形可以用于传输变形以及通过线性投影聚类疾病,分类准确率达到83%。

基准测试

基准方法指标
diffeomorphic-medical-image-registration-on-1cVAE Diffeomorphic (S3)
Dice: 0.812
Grad Det-Jac: 1.4
Hausdorff Distance (mm): 7.3
RMSE: 0.30

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
学习用于 diffeomorphic 注册的概率模型 | 论文 | HyperAI超神经