Hierarchical Text-Guided Brain Tumor Segmentation via Sub-Region-Aware Prompts
Organizations: Macquarie University, Sydney, NSW, Australia · Adelaide University, Adelaide, SA, Australia · Flinders University, Adelaide, SA, Australia · University of Sydney, Sydney, NSW, Australia
Abstract
Brain tumor segmentation remains challenging because the three standard sub-regions, i.e., whole tumor (WT), tumor core (TC), and enhancing tumor (ET), often exhibit ambiguous visual boundaries. Integrating radiological description texts with imaging has shown promise. However, most multimodal approaches typically compress a report into a single global text embedding shared across all sub-regions, overlooking their distinct clinical characteristics. We propose TextCSP (text-modulated soft cascade architecture), a hierarchical text-guided framework that builds on the TextBraTS baseline with three novel components: (1) a text-modulated soft cascade decoder that predicts WT->TC->ET in a coarse-to-fine manner consistent with their anatomical containment hierarchy. (2) sub-region-aware prompt tuning, which uses learnable soft prompts with a LoRA-adapted BioBERT encoder to generate specialized text representations tailored for each sub-region; (3) text-semantic channel modulators that convert the aforementioned representations into channel-wise refinement signals, enabling the decoder to emphasize features aligned with clinically described patterns. Experiments on the TextBraTS dataset demonstrate consistent improvements across all sub-regions against state-of-the-art methods by 1.7% and 6% on the main metrics Dice and HD95.
Figures & tables
| Methods | Dice (%) | HD95 | ||||||
|---|---|---|---|---|---|---|---|---|
| ET | WT | TC | Avg. | ET | WT | TC | Avg. | |
| 3D-UNet [ 16 ] | 80.4 | 87.3 | 81.6 | 83.1 | 6.11 | 10.51 | 8.93 | 8.17 |
| nnU-Net [ 9 ] | 82.2 | 87.5 | 82.6 | 84.1 | 4.27 | 11.90 | 8.52 | 8.23 |
| SegResNet [ 5 ] | 80.9 | 88.4 | 82.3 | 83.8 | 6.18 | 7.28 | 7.41 | 6.95 |
| Swin UNETR [ 4 ] | 81.0 | 89.5 | 80.8 | 83.8 | 5.95 | 8.23 | 7.03 | 7.07 |
| Nestedformer [ 20 ] | 82.6 | 89.5 | 80.2 | 84.1 | 5.08 | 10.51 | 8.93 | 8.17 |
| Soft Cascade | Sub-Region-Aware Prompt | LoRA | Text Modulation | Dice (%) | HD95 |
| 85.9 | 5.67 | ||||
| 86.4 | 5.26 | ||||
| 86.6 | 5.04 | ||||
| 87.0 | 4.81 |