arXiv:2503.16635eess.IVcs.CV2025-03被引 2

联邦扩散模型提升低计数全身PET图像质量,保护隐私同时增强病灶检测。

Fed-NDIF: A Noise-Embedded Federated Diffusion Model For Low-Count Whole-Body PET Denoising

  • 在联邦学习框架中嵌入噪声特征,实现多中心低计数PET图像去噪。
  • 相比本地模型和联邦UNet,PSNR、SSIM显著提升,病灶量化更准确。
  • 适用于医疗数据分散且需保护患者隐私的多中心影像研究。

低计数正电子发射断层扫描(LCPET)可降低患者辐射暴露,但伴随图像噪声增加与病灶检出率下降,亟需有效去噪方法。扩散模型在恢复图像质量方面展现潜力,但其训练依赖大规模多样化数据,在医疗领域难以获取。为应对数据稀缺与隐私问题,本文结合扩散模型与联邦学习——一种分布式训练机制,各机构本地训练模型后由中心服务器聚合参数。不同机构间扫描仪类型及噪声水平差异给联邦学习带来挑战。为此,提出新型噪声嵌入联邦扩散模型(Fed-NDIF),基于多中心数据与多种计数水平进行训练。该方法将肝脏归一化标准差(NSTD)噪声嵌入2.5D扩散模型,并采用联邦平均(FedAvg)算法聚合本地模型得到全局模型,再在本地数据上微调以获得个性化模型。在伯尔尼大学、上海瑞金医院和耶鲁-纽黑文医院的数据集上验证显示,该方法在整体3D体积的PSNR、SSIM和NMSE指标上均优于本地扩散模型与联邦UNet模型,显著提升图像质量与病灶检出及量化能力。

原文摘要 · Abstract (English)

Low-count positron emission tomography (LCPET) imaging can reduce patients' exposure to radiation but often suffers from increased image noise and reduced lesion detectability, necessitating effective denoising techniques. Diffusion models have shown promise in LCPET denoising for recovering degraded image quality. However, training such models requires large and diverse datasets, which are challenging to obtain in the medical domain. To address data scarcity and privacy concerns, we combine diffusion models with federated learning -- a decentralized training approach where models are trained individually at different sites, and their parameters are aggregated on a central server over multiple iterations. The variation in scanner types and image noise levels within and across institutions poses additional challenges for federated learning in LCPET denoising. In this study, we propose a novel noise-embedded federated learning diffusion model (Fed-NDIF) to address these challenges, leveraging a multicenter dataset and varying count levels. Our approach incorporates liver normalized standard deviation (NSTD) noise embedding into a 2.5D diffusion model and utilizes the Federated Averaging (FedAvg) algorithm to aggregate locally trained models into a global model, which is subsequently fine-tuned on local datasets to optimize performance and obtain personalized models. Extensive validation on datasets from the University of Bern, Ruijin Hospital in Shanghai, and Yale-New Haven Hospital demonstrates the superior performance of our method in enhancing image quality and improving lesion quantification. The Fed-NDIF model shows significant improvements in PSNR, SSIM, and NMSE of the entire 3D volume, as well as enhanced lesion detectability and quantification, compared to local diffusion models and federated UNet-based models.

联邦学习PET去噪扩散模型医学影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。