arXiv:2605.30631cs.CVcs.AI2026-05

用直方图约束生成肺结节,让合成结果更真实且保持亚型特征。

Controllable Lung Nodule Synthesis via Histogram-Regularized Latent Diffusion Models

论文配图:Controllable Lung Nodule Synthesis via Histogram-Regularized Latent Diffusion Models
图 1 · 摘自论文原文
  • 引入直方图正则化项控制生成过程中的密度分布。
  • 合成的结节在视觉真实性和亚型一致性上显著提升。
  • 适合用于数据增强,尤其改善罕见亚型的诊断性能。

尽管基于CT的肺癌筛查自动化诊断系统已取得显著进展,但其发展仍受限于多样且标注完善的肺结节数据集稀缺。基于扩散的生成模型为数据合成提供了可行方案;然而,现有条件方法主要优化空间重建损失,仅促进体素级相似性,难以有效约束病灶级的强度分布,导致生成结果纹理过平滑,且未能充分体现不同类型结节(实性、部分实性、磨玻璃)的特异性衰减特征。为此,我们提出一种可控的潜在扩散模型,在完整的3D CT体积中合成肺结节,并准确建模结节特定的强度分布。具体而言,不依赖单一空间损失,而是引入基于直方图的正则化项,约束生成过程中的体素强度分布。模型结合亚型、空间掩码与亨氏单位(HU)直方图条件,以及可微特征空间直方图正则化项,更精准对齐病灶级强度分布,提升合成结节的视觉合理性与亚型一致性。大量实验表明,该框架在肺部CT数据上实现了强视觉真实感,通过定量指标与视觉图灵测试验证。此外,用于数据增强时,生成结节可提升下游临床任务表现,特别是对低频结节亚型,并展现出对亚型指导恶性度分类的潜力。

原文摘要 · Abstract (English)

While automated diagnosis systems have achieved remarkable success in computed tomography (CT)-based lung cancer screening, their development remains limited by the scarcity of diverse, annotated pulmonary nodule datasets. Diffusion-based generative models offer a promising strategy for data synthesis; however, many existing conditional approaches primarily optimize spatial reconstruction losses, which encourage voxel-wise similarity but may inadequately constrain lesion-level intensity distributions. As a result, these methods may produce over-smoothed texture profiles and underrepresent the distinct attenuation characteristics of different nodule subtypes, including solid, part-solid, and ground-glass nodules. To address this challenge, we propose a controllable latent diffusion model that synthesizes pulmonary nodules within full 3D CT volumes while accurately modeling nodule-specific intensity distributions. Specifically, rather than relying solely on spatial losses, we introduce a histogram-based regularization term that constrains voxel intensity distributions during the generative process. The model combines subtype, spatial mask, and Hounsfield unit (HU) histogram conditioning with the differentiable feature-space histogram regularization term to better align lesion-level intensity distributions, improving the visual plausibility and subtype consistency of synthesized nodules. Extensive experiments on lung CT data demonstrate that our framework achieves strong visual realism, validated through both quantitative metrics and a visual Turing test. Furthermore, when used for data augmentation, the generated nodules improve performance in downstream clinical tasks, particularly for underrepresented nodule subtypes, and show a potential benefit for subtype-informed malignancy classification.

肺结节生成扩散模型数据增强医学图像

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