arXiv AI By Junho Moon, Ikbeom Jang

TRACE: Topology-Aware Reconnection via Prediction-Adaptive Connectivity Enforcement for Thin-Structure Segmentation

Read the original on arXiv AI →

TRACE is a topology‑aware, prediction‑adaptive objective for thin‑structure segmentation that uses a target skeleton to define required local connections. For each connection, it identifies the easiest valid reconnection under the current prediction and jointly supervises its bottleneck with the corresponding skeleton point, allowing supervision to adapt to the evolving prediction rather than remaining fixed. A topology‑aware skeleton carrier preserves both foreground and class‑specific topology, enabling reliable definition of these connections, and extensive experiments on six 2D and 3D benchmarks demonstrate improved connectivity and topology while maintaining strong segmentation accuracy with low training overhead.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv Computer Vision
Oct 22

Disconnect to Connect: A Data Augmentation Method for Improving Topology Accuracy in Image Segmentation

The paper introduces CoLeTra, a data augmentation technique designed to enhance topology accuracy in image segmentation of thin, tubular structures such as blood vessels. By artificially breaking these structures in training images while preserving the original labels, CoLeTra teaches models that continuous structures can appear disconnected, improving segmentation quality even when training labels are topologically inaccurate. Experiments across various architectures, loss functions, and datasets show that CoLeTra yields more topologically accurate segmentations and often boosts Dice and clDice scores, and the authors also release a dedicated dataset for topology-focused segmentation.

By Juan Miguel Valverde, Maja {\O}stergaard, Adrian Rodriguez-Palomo, Peter Alling Strange Vibe, Nina K{\o}lln Wittig, Henrik Birkedal, Anders Bjorholm Dahl
arXiv Computer Vision
Oct 5

Branch-Centric Tokenization and Test-Time Augmentation for Skeleton Generation

The paper presents a new approach to automatic skeleton generation that tackles two key challenges: encoding branch structures into token sequences and leveraging test-time computation. It introduces branch‑centric tokenization, a representation that groups structurally related elements together and directly encodes connectivity, producing more compact sequences than traditional BFS serialization. Additionally, the authors propose view‑augmented generation, a test‑time augmentation technique that rotates the input mesh, aligns predictions, and selects the best skeleton based on mesh coverage and consistency across views, leading to improved accuracy on benchmark datasets.

By Zhengyuan Li, Chuanyu Pan, Yuanming Hu, Raymond A. Yeh
arXiv Computer Vision
Oct 9

Skeleton-Guided Progressive Test-Time Adaptation for Thin Curvilinear Structures

The paper introduces Skeleton‑Guided Progressive Test‑Time Adaptation (SGP‑TTA) for accurately segmenting thin curvilinear structures such as vessels and roads. It combines Progressive Batch Normalization (ProgBN) to gradually shift normalization statistics toward target data, and Consensus Skeleton Recall (CSR) to enforce structural connectivity by updating only BN affine parameters. Experiments demonstrate that SGP‑TTA outperforms existing test‑time adaptation methods, especially under cross‑modality domain shifts.

By Boa Jang, JunGyu Lee, Gwanho Lee, Jinwook Choi, Young-Gon Kim
arXiv Computer Vision
Oct 9

Refine Connections, Close the Gap: A Reliable Enhancement Framework for Driving Scene Topology

The paper introduces TopoEnhance, a framework that improves the reliability of scene topology graphs for autonomous driving by treating topology enhancement as a denoising reconstruction task. It learns to recover structural consistency from corrupted ground‑truth graphs, thereby correcting logical inconsistencies and unreliable connections. Experiments show consistent gains in both continuous topology metrics (TOP score) and discrete connectivity (Topology Jaccard Similarity) across multiple baselines without retraining.

By Xiaoqi Wang, Dingyi Zhaung, David Paz, Wenbin He, Yucai Bai, Peng Zhou, Rui Zhang, Jinhua Zhao, Liu Ren
arXiv Computer Vision
22h ago

OcTreeMamba: Topology-Aware Mamba for 3D Medical Segmentation via Octree Sparsification

OcTreeMamba introduces a hierarchical framework that integrates Octree structures into a Mamba-based segmentation model for 3D medical images. It uses a Structure Learner to sparsify anatomy-adaptive regions via hierarchical gating and Gumbel-Softmax sampling, then builds a Dynamic Octree to preserve topological connectivity. A bidirectional Tree-SSM propagates contextual features along leaf–parent–root and root–parent–leaf paths, achieving competitive performance on four benchmark datasets, especially for vascular structures.

By Kaixiang Mai, Jiaqi Yu, Yuzhu Ji, Yiqun Zhang, Zhijie Wang, Haijun Zhang, Yiu-ming Cheung