arXiv:2506.11183eess.IVcs.CV2025-06

通过解耦频段学习,用扩散模型修复相位图细节

DiffPR: Diffusion-Based Phase Reconstruction via Frequency-Decoupled Learning

  • 分两阶段设计:先用无高频跳接的U-Net提取低频结构
  • 再用无条件扩散模型逐步还原高频残差,提升细节清晰度
  • 在4个细胞数据集上显著改善相位重建质量,适合医学成像应用

深度学习在离轴定量相位成像(QPI)中仍面临过平滑问题。端到端的U-Net倾向于保留低频信息,忽略细粒度诊断特征。我们发现这是谱偏差所致,且由高层跳跃连接强化。移除最深层的跳跃连接,仅在低分辨率监督网络,可显著提升泛化性和保真度。基于此,提出DiffPR:第一阶段,采用取消高频跳接的非对称U-Net,从干涉图预测四分之一尺度的相位图,捕捉可靠低频结构;第二阶段,将上采样预测结果轻度添加高斯噪声后,输入无条件扩散模型,通过反向去噪迭代恢复缺失高频残差。在四个QPI数据集(B-Cell、WBC、HeLa、3T3)上的实验表明,DiffPR优于强基线,峰值信噪比(PSNR)最高提升1.1 dB,平均绝对误差(MAE)降低11%,膜嵴和斑点图案明显更锐利。结果证明,取消高层跳接并交由扩散先验处理细节,是缓解传统相位重建网络谱偏差的有效方法。

原文摘要 · Abstract (English)

Oversmoothing remains a persistent problem when applying deep learning to off-axis quantitative phase imaging (QPI). End-to-end U-Nets favour low-frequency content and under-represent fine, diagnostic detail. We trace this issue to spectral bias and show that the bias is reinforced by high-level skip connections that feed high-frequency features directly into the decoder. Removing those deepest skips thus supervising the network only at a low resolution significantly improves generalisation and fidelity. Building on this insight, we introduce DiffPR, a two-stage frequency-decoupled framework. Stage 1: an asymmetric U-Net with cancelled high-frequency skips predicts a quarter-scale phase map from the interferogram, capturing reliable low-frequency structure while avoiding spectral bias. Stage 2: the upsampled prediction, lightly perturbed with Gaussian noise, is refined by an unconditional diffusion model that iteratively recovers the missing high-frequency residuals through reverse denoising. Experiments on four QPI datasets (B-Cell, WBC, HeLa, 3T3) show that DiffPR outperforms strong U-Net baselines, boosting PSNR by up to 1.1 dB and reducing MAE by 11 percent, while delivering markedly sharper membrane ridges and speckle patterns. The results demonstrate that cancelling high-level skips and delegating detail synthesis to a diffusion prior is an effective remedy for the spectral bias that limits conventional phase-retrieval networks.

相位重建扩散模型频段解耦医学成像

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