GraphMorph: Tubular Structure Extraction by Morphing Predicted Graphs
This addresses the challenge of topological accuracy in tubular structure extraction for applications like medical imaging or road network analysis, representing an incremental improvement over traditional pixel-level classification methods.
The paper tackles the problem of accurately restoring topology in tubular structure extraction tasks, such as blood vessel segmentation, by proposing GraphMorph, a method that focuses on branch-level features to achieve more topologically accurate predictions, with experimental evaluations showing significant reductions in false positives in segmentation tasks.
Accurately restoring topology is both challenging and crucial in tubular structure extraction tasks, such as blood vessel segmentation and road network extraction. Diverging from traditional approaches based on pixel-level classification, our proposed method, named GraphMorph, focuses on branch-level features of tubular structures to achieve more topologically accurate predictions. GraphMorph comprises two main components: a Graph Decoder and a Morph Module. Utilizing multi-scale features extracted from an image patch by the segmentation network, the Graph Decoder facilitates the learning of branch-level features and generates a graph that accurately represents the tubular structure in this patch. The Morph Module processes two primary inputs: the graph and the centerline probability map, provided by the Graph Decoder and the segmentation network, respectively. Employing a novel SkeletonDijkstra algorithm, the Morph Module produces a centerline mask that aligns with the predicted graph. Furthermore, we observe that employing centerline masks predicted by GraphMorph significantly reduces false positives in the segmentation task, which is achieved by a simple yet effective post-processing strategy. The efficacy of our method in the centerline extraction and segmentation tasks has been substantiated through experimental evaluations across various datasets. Source code will be released soon.