用双约束扩散模型,实现超低剂量PET图像高质量重建。
Double-Constraint Diffusion Model with Nuclear Regularization for Ultra-low-dose PET Reconstruction
- 冻结预训练扩散模型,仅训练双约束控制器,参数量极低。
- 在1%全剂量下仍保持高图像质量,优于现有方法。
- 适合临床快速适配不同剂量,无需重新训练。
超低剂量正电子发射断层扫描(PET)重建可显著降低患者辐射暴露并缩短检查时间,但可能导致噪声增加和细节丢失,影响图像质量。本文提出双约束扩散模型(DCDM),冻结预训练扩散模型权重,在编码架构中引入可训练的双约束控制器,大幅减少可训练参数。该模型通过核范数约束(NTC)和编码连接约束(ENC)协同优化:NTC利用核范数近似矩阵秩最小化,将低秩特性融入Transformer,高效提取低剂量图像信息并生成压缩特征表示;ENC则基于这些表示控制预训练扩散模型,生成像素空间的重建图像。在公开的UDPET数据集和临床数据集上的实验表明,DCDM在已知剂量缩减因子(DRF)下表现超越当前最优方法,并能良好泛化至未知DRF场景,即使在1%全剂量条件下仍具实用价值。
原文摘要 · Abstract (English)
Ultra-low-dose positron emission tomography (PET) reconstruction holds significant potential for reducing patient radiation exposure and shortening examination times. However, it may also lead to increased noise and reduced imaging detail, which could decrease the image quality. In this study, we present a Double-Constraint Diffusion Model (DCDM), which freezes the weights of a pre-trained diffusion model and injects a trainable double-constraint controller into the encoding architecture, greatly reducing the number of trainable parameters for ultra-low-dose PET reconstruction. Unlike full fine-tuning models, DCDM can adapt to different dose levels without retraining all model parameters, thereby improving reconstruction flexibility. Specifically, the two constraint modules, named the Nuclear Transformer Constraint (NTC) and the Encoding Nexus Constraint (ENC), serve to refine the pre-trained diffusion model. The NTC leverages the nuclear norm as an approximation for matrix rank minimization, integrates the low-rank property into the Transformer architecture, and enables efficient information extraction from low-dose images and conversion into compressed feature representations in the latent space. Subsequently, the ENC utilizes these compressed feature representations to encode and control the pre-trained diffusion model, ultimately obtaining reconstructed PET images in the pixel space. In clinical reconstruction, the compressed feature representations from NTC help select the most suitable ENC for efficient unknown low-dose PET reconstruction. Experiments conducted on the UDPET public dataset and the Clinical dataset demonstrated that DCDM outperforms state-of-the-art methods on known dose reduction factors (DRF) and generalizes well to unknown DRF scenarios, proving valuable even at ultra-low dose levels, such as 1% of the full dose.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。