RefTr: Recurrent Refinement of Confluent Trajectories for 3D Tubular Tree Centerlines

Paper in proceeding, 2027

Tubular tree structures such as blood vessels and lung airways are central to many clinical tasks, including diagnosis, treatment planning, and surgical navigation. Accurate centerline extraction with correct topology is essential, as missing small branches can lead to incomplete assessments or overlooked abnormalities. We propose RefTr, a 3D image-to-graph framework that generates tubular tree centerlines via recurrent refinement of confluent trajectories. RefTr adopts a Transformer-based Producer–Refiner architecture in which the Producer predicts candidate trajectories and a shared Refiner iteratively aligns with target branches. The confluent trajectory representation enables whole-branch refinement while explicitly enforcing valid topology. This recurrent scheme improves precision and reduces decoder parameters by 2.4× compared to the state-of-the-art. We further introduce an efficient non-maximum suppression algorithm for spatial tree graphs to merge duplicate branches and extend evaluation metrics to be radius-aware for robust comparison. Experiments on multiple public datasets demonstrate higher recall and precision, faster inference, and substantially fewer parameters, highlighting the effectiveness of RefTr for 3D tubular tree analysis. Code is available at https://github.com/RomStriker/RefTr.