Mind the Gap: Mesh-Guided Repair of Broken Vessels
Authors: Gniewosz Drwiega, Wojciech Szymanski, Marek Wodzinski
Organizations: Sano – Centre for Computational Personalised Medicine International Research Foundation, Krakow, Poland · AGH University of Krakow, Krakow, Poland
Vessel segmentation is commonly optimized as voxel-wise classification, but small local errors can strongly disrupt vascular connectivity while having little effect on overlap scores. This is particularly problematic for downstream analyses that rely on centerlines, branches, connected components, or graph structure. We propose a mesh-guided post-processing framework for repairing broken vessel segmentations produced by nnU-Net. For each predicted binary mask, a deformable template mesh is fitted to the mask surface in physical space and used as a case-specific geometric scaffold. The fitted mesh is not voxelized as the final segmentation; instead, it guides conservative reconnection of disconnected components by proposing or validating thin bridge candidates under foreground-growth constraints. We evaluated this approach in three vascular anatomies using AortaSeg24 and SEGA for the aorta, TopCoW for the Circle of Willis, and PARSE for the pulmonary arteries. Performance is measured using Dice, connected-component Dice (ccDice), and the Betti-0 number. Across these datasets, repair substantially improved connectivity while preserving overlap: Dice remained nearly unchanged, whereas ccDice increased from 0.596 to 0.992 for aorta, from 0.722 to 0.835 for TopCoW, and from 0.028 to 0.862 for PARSE. The FOMAML meta-initialization further accelerated the fitting per-case, supporting practical mesh-based repair of the vascular topology. These results suggest that explicit mesh representations can provide a useful geometric prior for correcting topological failures in otherwise accurate voxel segmentations.
Figures & tables
Figure 1: Examples of broken-vessel failure modes across vascular anatomies. Top row: reference vessel structures. Bottom row: broken nnU-Net predictions, with missing local connections highlighted. Such small local errors can fragment otherwise accurate masks.
Figure 2: Overview of the proposed mesh-guided nnU-Net repair pipeline. A CT volume is first processed by a CNN segmentation branch based on nnU-Net, producing segmented vessel sections and an initial nnU-Net mask. When the predicted mask is fragmented or topologically inconsistent, its surface is extracted and used as the target for mesh fitting. Then, a fixed template mesh is passed through a GCN deformation branch. The mesh is progressively refined over multiple deformation stages, where each stage predicts bounded vertex offsets and forwards the updated mesh to the next refinement stage. The final overfitted mesh approximates the surface geometry of the nnU-Net prediction while preserving the template mesh connectivity. This fitted mesh is then used as a geometric prior for topology-aware mask repair, producing a repaired nnU-Net mask with improved connectivity.
Figure 3: Qualitative examples of mesh-guided repair across vascular anatomies. The method reconnects fragmented vessel components using local mesh-guided bridges while preserving the original mask away from the repaired regions. Blue arrows indicate broken vessel regions repaired by the proposed method.
Dataset
Mask
Dice ↑
ccDice ↑
β0↓
FB ↓
Aorta ( n=145 )
Raw
0.934±0.007
0.596±0.045
3.35±0.43
0.056±0.009
Filtered
0.935±0.007
0.695±0.043
2.47±0.30
0.044±0.008
Repaired
0.934±0.007
0.992±0.009
1.01±0.02
0.055±0.009
TopCoW ( n=125 )
Raw
0.870±0.010
0.722±0.043
2.58±0.20
0.040±0.006
Filtered
0.870±0.010
0.723±0.043
2.58±0.20
0.040±0.006
Repaired
0.867±0.010
0.835±0.034
1.02±0.02
0.063±0.007
Table 3: Segmentation and connectivity metrics before and after repair. Values are reported as mean ± 95% confidence interval. Best values within each dataset are shown in bold. FB denotes false-branch fraction.
Figure 4: Quantitative summary of repair effects and mesh-fitting convergence. Mesh-guided repair improves connectivity-oriented metrics while preserving Dice, and meta-initialization accelerates per-case mesh fitting.
Figure 5: Effect of FOMAML meta-initialization on per-case mesh-fitting convergence for 25 held-out TopCoW cases. Lines show mean Chamfer loss across cases, and shaded regions indicate ± SEM. The black horizontal line indicates a matched Chamfer-distance level, allowing comparison of the number of optimization steps required by FOMAML and standard initialization to reach the same reconstruction error.
Department of Computer Science, Kennesaw State University, Marietta, 30060, GA, USA · Department of Cardiology, Medical University of South Carolina, Charleston, SC, USA