用单向变换的无监督生成模型,实现高质量低剂量CT重建。
Unsupervised Low-dose CT Reconstruction with One-way Conditional Normalizing Flows
- 采用单向变换策略,避免细节丢失和伪影产生。
- 无需标注数据即可在高分辨率图像上训练,速度与质量俱佳。
- 适合临床场景中缺乏标签数据的低剂量CT重建任务。
深度学习在低剂量计算机断层扫描(LDCT)重建中表现优异。然而,监督方法面临临床数据缺乏标签的问题,基于CNN的无监督去噪方法又常导致图像过度平滑。近年来,归一化流(NFs)方法在生成细节丰富的图像和避免过平滑方面展现出优势,但仍存在两个问题:(1) 当前在数据空间与潜在空间交替优化时采用双向变换,会导致细节损失和二次伪影;(2) 在高分辨率CT图像上训练NFs计算量巨大。尽管条件归一化流(CNFs)可降低计算负担,但现有方法需依赖标注数据进行条件化,无监督的CNFs用于LDCT重建仍属难题。为此,本文提出一种新型基于CNFs的无监督迭代重建算法。该方法在双空间交替优化中采用严格单向变换,有效避免细节丢失与伪影;同时提出新颖的无监督条件化策略,使CNFs可在高分辨率图像上高效训练,实现快速且高质量的无监督重建。在多个数据集上的实验表明,所提方法性能超越部分先进无监督甚至监督方法。
原文摘要 · Abstract (English)
Deep-learning methods have shown promising performance for low-dose computed tomography (LDCT) reconstruction. However, supervised methods face the problem of lacking labeled data in clinical scenarios, and the CNN-based unsupervised denoising methods would cause excessive smoothing in the reconstructed image. Recently, the normalizing flows (NFs) based methods have shown advantages in producing detail-rich images and avoiding over-smoothing, however, there are still issues: (1) Although the alternating optimization in the data and latent space can well utilize the regularization and generation capabilities of NFs, the current two-way transformation strategy of noisy images and latent variables would cause detail loss and secondary artifacts; and (2) Training NFs on high-resolution CT images is hard due to huge computation. Though using conditional normalizing flows (CNFs) to learn conditional probability can reduce the computational burden, current methods require labeled data for conditionalization, and the unsupervised CNFs-based LDCT reconstruction remains a problem. To tackle these problems, we propose a novel CNFs-based unsupervised LDCT iterative reconstruction algorithm. It employs strict one-way transformation when performing alternating optimization in the dual spaces, thus effectively avoiding the problems of detail loss and secondary artifacts. By proposing a novel unsupervised conditionalization strategy, we train CNFs on high-resolution CT images, thus achieving fast and high-quality unsupervised reconstruction. Experiments on different datasets suggest that the performance of the proposed algorithm could surpass some state-of-the-art unsupervised and even supervised methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。