用几何先验让医学图像原型更准确,避免传统方法的偏差。
Deep Image Prototype Learning with Geometric Heat-Kernel Priors

- 基于热核加权图的图中位点选择原型,保持在数据曲面上。
- 在心脏瘢痕和脑部MRI上达到最高精度,且支持大量亚群。
- 无需标签即可评估质量,适用于多疾病亚型分析。
无监督学习医学影像队列的表征可揭示解剖学上有意义的原型,而无需依赖常含噪声且无法捕捉真实病理异质性的专家标签。然而,现有深度潜变量模型通过欧氏平均估计高斯混合先验,导致原型偏离弯曲的数据流形,并随亚群数量增加而退化。本文提出一种基于几何感知期望最大化(EM)算法的流形锚定变分框架,其M步在热核加权潜空间图上选择具有最高扩散中心性的图中位点作为每个亚群的原型,确保原型始终位于流形上。引入狄利克雷能量正则项以保证潜空间的几何平滑性,并为每个亚群生成不确定性评分,实现无标签的质量评估。该流形锚定EM是一种通用几何工具,可推广至其他潜变量模型。在心脏瘢痕和脑部MRI基准测试中,本框架在所有对比方法中表现最优,生成了迄今最清晰的原型,且在大量亚群下仍稳定,而所有基线方法均出现退化。代码与实现细节见https://github.com/jr-xing/On-Manifold-Variational-Learning-with-Heat-Kernel-Priors。
原文摘要 · Abstract (English)
Learning unsupervised representations of medical imaging cohorts can reveal anatomically meaningful prototypes without expert labels, which are often noisy and fail to capture true pathological heterogeneity. However, existing deep latent-variable models estimate Gaussian mixture priors via Euclidean averaging, producing prototypes that drift off the curved data manifold and degenerate as the number of sub-populations grows. We propose a manifold-anchored variational framework built on a geometry-aware Expectation-Maximization (EM) algorithm, whose M-step selects each sub-population prototype as the graph medoid with the highest diffusion centrality on a heat-kernel-weighted latent graph, ensuring that every prototype remains on-manifold. A Dirichlet energy regularizer enforces geometric smoothness of the latent space, and a per-sub-population uncertainty score enables label-free quality assessment. The manifold-anchored EM is a general-purpose geometric tool that extends standard EM and applies readily to other latent-variable models beyond this setting. On cardiac scar and brain MRI benchmarks, our framework attains the highest accuracy among all compared methods, produces the sharpest prototypes reported to date, and remains stable at large sub-population counts where all baselines degenerate. Code and implementation details are available at https://github.com/jr-xing/On-Manifold-Variational-Learning-with-Heat-Kernel-Priors.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。