BayeTopo: Bayesian-based Topology-guided Learning for Vascular Imaging Segmentation.
Authors
Abstract
Vascular segmentation is a critical task in clinical medical image processing and a prerequisite for accurately diagnosing vascular-related diseases. The development of automated segmentation methods is challenged by internal variability in vessel representations. Recently, topology guidance has shown potential for capturing semantically consistent representations. However, current topology-guided methods lack modeling of global-to-local dependencies. This limitation forces latent representations subject to a trade-off between learning global topology and local geometries within the vascular network. In this paper, we propose a Bayesian-based topology-guided (BayeTopo) learning approach to capture global-to-local dependencies. It introduces a prior that explicitly models local geometry as a probability conditioned on global topology within topology-sensitive regions of the vascular network. We further implement a topology-guided diffusion model to optimize the conditional probability. It gradually infers local geometry from the restored global topology with multi-scale noise, enabling rich global-to-local representations. Then, an inhomogeneous diffusion process is involved, where noise initially accumulates in topology-sensitive regions before achieving uniformity. It ensures an orderly degradation of information from global topology to local geometry, thereby enabling effective global-to-local supervision. Extensive experiments on six datasets, involving three types of vascular networks under four imaging modalities, demonstrate the superior performance and generalization capability of our method compared to previous topology-guided learning and diffusion-based models. A series of case studies further validates the effectiveness of our designs in enhancing semantic consistency within local vascular regions, thereby improving topological accuracy.