TubeMLLM: A Foundation Model for Topology Knowledge Exploration in Vessel-like Anatomy
Organizations: Institute of Medical Robotics, Shanghai Jiao Tong University, Shanghai, China · Department of Automation, Shanghai Jiao Tong University, Shanghai, China
Abstract
Modeling medical vessel-like anatomy is challenging due to its intricate topology and sensitivity to dataset shifts. Consequently, task-specific models often suffer from topological inconsistencies, including artificial disconnections and spurious merges. Motivated by the promise of multimodal large language models (MLLMs) for zero-shot generalization, we propose TubeMLLM, a unified foundation model that couples structured understanding with controllable generation for medical vessel-like anatomy. By integrating topological priors through explicit natural language prompting and aligning them with visual representations in a shared-attention architecture, TubeMLLM significantly enhances topology-aware perception. Furthermore, we construct TubeMData, a pionner multimodal benchmark comprising comprehensive topology-centric tasks, and introduce an adaptive loss weighting strategy to emphasize topology-critical regions during training. Extensive experiments on fifteen diverse datasets demonstrate our superiority. Quantitatively, TubeMLLM achieves state-of-the-art out-of-distribution performance, substantially reducing global topological discrepancies on color fundus photography (decreasing the number error from 37.42 to 8.58 compared to baselines). Notably, TubeMLLM exhibits exceptional zero-shot cross-modality transferring ability on unseen X-ray angiography, achieving a Dice score of 67.50% while significantly reducing the error to 1.21. TubeMLLM also maintains robustness against degradations such as blur, noise, and low resolution. Furthermore, in topology-aware understanding tasks, the model achieves 97.38% accuracy in evaluating mask topological quality, significantly outperforming standard vision-language baselines.
Figures & tables
| Methods | Dice | clDice | Num | Mat |
| nnUNet [ 7 ] | 74.44 | 78.48 | 37.42 | 36.56 |
| nnUNet w/ clDice [ 10 ] | 75.19 | 79.83 | 25.98 | 27.62 |
| nnUNet w/ SR [ 11 ] | 74.76 | 79.68 | 34.49 | 30.52 |
| nnUNet w/ cbDice [ 12 ] | 74.94 | 79.34 | 25.46 | 30.11 |
| nnUNet w/ CAL [ 13 ] | 72.63 | 77.85 | 56.91 | 29.79 |
| FR-UNet [ 8 ] | 71.59 | 76.76 | 40.13 | 27.94 |
| Task | Method | Dice | clDice | Num | Mat |
| Task: Single Step Refinement(SSR) | |||||
| SSR | nnUNet [ 7 ] | 74.44 | 78.48 | 37.42 | 36.56 |
| w/ TubeMLLM | 76.05 | 80.43 | 8.49 | 26.51 | |
| Task: Cross Dataset | |||||
| ZS | nnUNet [ 7 ] | 9.07 | 11.74 | 238.26 | 5.29 |
| TubeMLLM | 67.50 | 69.17 | 1.21 | 7.69 | |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
Medical VLMs
Medical vision-language models connect clinical images with text for diagnosis, reporting, question answering, and image grounding.
See all 474 papers