Search papers, labs, and topics across Lattice.
This paper introduces a manifold-anchored variational framework for learning unsupervised representations in medical imaging, addressing the limitations of existing deep latent-variable models that rely on Gaussian mixture priors. By employing a geometry-aware Expectation-Maximization algorithm, the method ensures that sub-population prototypes remain on the data manifold, leveraging heat-kernel-weighted latent graphs to select graph medoids based on diffusion centrality. The proposed approach outperforms existing methods on cardiac scar and brain MRI benchmarks, achieving the highest accuracy and producing sharper prototypes even as sub-population counts increase.
Prototypes derived from this new framework remain on the data manifold, leading to unprecedented accuracy and clarity in medical imaging representations.
Learning unsupervised representations of medical imaging cohorts can reveal clinically 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. \rev{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.
