One-Shot Active Learning for Vessel Segmentation
摘要
Vessel segmentation is crucial for analyzing brain vasculature and understanding cerebral functions and disease mechanisms. Current deep-learning models for segmenting blood vessels within brain images are supervised and depend on extensive labeled data, which requires expert annotation and is both time-consuming and resource-intensive. To address these challenges, we propose Vessel-Dictionary Selection Net (V-DiSNet), a one-shot active learning (OSAL) framework specifically designed for vessels that can be used to select a small, representative set of informative and diverse samples for expert annotation and training, given an unlabeled dataset in a single iteration. The selection process involves sampling from a latent space designed by leveraging the recurrent properties of brain vessel patterns. Specifically, we combine dictionary learning with k-means clustering to learn a latent representation integrating fundamental basis elements representing recurrent vessel features such as shape, connectivity, and structures. We experimentally demonstrate the effectiveness of our method on three publicly available 3D Magnetic Resonance Angiography datasets, showing that V-DisNet consistently outperforms random sampling and other state-of-the-art OSAL methods in terms of standard vessel segmentation metrics. Our code is available at github.com/i-vesseg/V-DiSNet .