arXiv:2507.02405cs.CV2025-07被引 4

用位置感知扩散自编码器,提升高分辨率脑组织分类与图像伪影修复

PosDiffAE: Position-aware Diffusion Auto-encoder For High-Resolution Brain Tissue Classification Incorporating Artifact Restoration

  • 引入位置感知机制,让模型学习脑图像块的位置信息以区分组织类型
  • 无监督修复撕裂和JPEG伪影,利用扩散模型的生成约束能力
  • 适合医学图像分析、脑组织分类及图像修复领域的研究人员

去噪扩散模型通过逐步建模图像分布生成高质量图像,但其采样过程无法提取图像特定语义表示,而自编码器则具备此能力。本文将编码器与扩散模型结合,构建扩散自编码框架,实现图像到隐空间的映射与结构化表征。首先,设计一种隐空间结构化机制,强制模型回归高分辨率脑图像块的位置信息,从而提升区域特异性细胞模式识别能力。其次,基于邻域感知提出无监督撕裂伪影修复方法,利用隐空间表示和扩散模型的生成约束能力。第三,通过表征引导,并利用扩散模型在推理阶段可调控的加噪与去噪能力,实现无监督的JPEG伪影修复。

原文摘要 · Abstract (English)

Denoising diffusion models produce high-fidelity image samples by capturing the image distribution in a progressive manner while initializing with a simple distribution and compounding the distribution complexity. Although these models have unlocked new applicabilities, the sampling mechanism of diffusion does not offer means to extract image-specific semantic representation, which is inherently provided by auto-encoders. The encoding component of auto-encoders enables mapping between a specific image and its latent space, thereby offering explicit means of enforcing structures in the latent space. By integrating an encoder with the diffusion model, we establish an auto-encoding formulation, which learns image-specific representations and offers means to organize the latent space. In this work, First, we devise a mechanism to structure the latent space of a diffusion auto-encoding model, towards recognizing region-specific cellular patterns in brain images. We enforce the representations to regress positional information of the patches from high-resolution images. This creates a conducive latent space for differentiating tissue types of the brain. Second, we devise an unsupervised tear artifact restoration technique based on neighborhood awareness, utilizing latent representations and the constrained generation capability of diffusion models during inference. Third, through representational guidance and leveraging the inference time steerable noising and denoising capability of diffusion, we devise an unsupervised JPEG artifact restoration technique.

扩散模型脑组织分类图像修复自编码器

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