TreeGaussian: Tree-Guided Cascaded Contrastive Learning for Hierarchical Consistent 3D Gaussian Scene Segmentation and Understanding
Organizations: Institute of Computing Technology, Chinese Academy of Sciences, ICT · University of Chinese Academy of Sciences, UCAS · Beihang University, BUAA
Abstract
3D Gaussian Splatting (3DGS) has emerged as a real-time, differentiable representation for neural scene understanding. However, existing 3DGS-based methods struggle to represent hierarchical 3D semantic structures and capture whole-part relationships in complex scenes. Moreover, dense pairwise comparisons and inconsistent hierarchical labels from 2D priors hinder feature learning, resulting in suboptimal segmentation. To address these limitations, we introduce TreeGaussian, a tree-guided cascaded contrastive learning framework that explicitly models hierarchical semantic relationships and reduces redundancy in contrastive supervision. By constructing a multi-level object tree, TreeGaussian enables structured learning across object-part hierarchies. In addition, we propose a two-stage cascaded contrastive learning strategy that progressively refines feature representations from global to local, mitigating saturation and stabilizing training. A Consistent Segmentation Detection (CSD) mechanism and a graph-based denoising module are further introduced to align segmentation modes across views while suppressing unstable Gaussian points, enhancing segmentation consistency and quality. Extensive experiments, including open-vocabulary 3D object selection, 3D point cloud understanding, and ablation studies, demonstrate the effectiveness and robustness of our approach.
Figures & tables
| Scale | Methods | Type | mIoU(%) | mAcc(%) |
| whole | LangSplat [ 33 ] | Language distillation | 9.66 | 12.41 |
| whole | LEGaussians [ 37 ] | Language distillation | 16.21 | 23.82 |
| whole | OmniSeg3D-GS [ 44 ] | Fused contrastive learning | 38.55 | 63.29 |
| whole | OpenGaussian [ 40 ] | Flat contrastive learning | 38.36 | 51.43 |
| whole | InstanceGaussian [ 23 ] | Flat contrastive learning | 45.30 | 58.44 |
| whole | Ours | Cascaded contrastive learning | 51.78 | 70.51 |
| Methods | 19 Classes | 15 Classes | 10 Classes | ||||||
|---|---|---|---|---|---|---|---|---|---|
| whole | part | subpart | whole | part | subpart | whole | part | subpart | |
| OmniSeg3D-GS [ 44 ] | 20.56 | 15.29 | 12.11 | 23.74 | 17.83 | 18.79 | 32.41 | 25.73 | 18.79 |
| OpenGaussian [ 40 ] | 24.73 | 18.12 | 14.73 | 30.13 | 20.91 | 17.24 | 38.29 | 33.50 | 25.77 |
| InstanceGaussian [ 23 ] | 40.66 | 27.20 | 17.20 | 42.51 | 23.76 | 19.93 | 47.94 | 30.63 | 24.24 |
| Ours | 41.61 | 28.99 | 18.32 | 42.57 | 29.28 | 20.01 | 49.80 | 38.76 | 27.66 |
| Methods | 19 Classes | 15 Classes | 10 Classes | ||||||
|---|---|---|---|---|---|---|---|---|---|
| whole | part | subpart | whole | part | subpart | whole | part | subpart | |
| OmniSeg3D-GS [ 44 ] | 38.51 | 25.39 | 23.13 | 36.02 | 29.06 | 30.49 | 48.32 | 38.91 | 30.49 |
| OpenGaussian [ 40 ] | 41.54 | 28.89 | 24.66 | 48.25 | 31.89 | 27.87 | 55.19 | 45.17 | 39.88 |
| InstanceGaussian [ 23 ] | 54.01 | 40.73 | 27.78 | 59.15 | 37.63 | 29.07 | 64.01 | 48.89 | 37.50 |
| Ours | 54.38 | 42.11 | 28.82 | 59.94 | 42.95 | 32.61 | 65.41 | 54.55 | 40.35 |
| Scale | w/o | w/o Denoising | Ours | |||
|---|---|---|---|---|---|---|
| mIoU | mAcc | mIoU | mAcc | mIoU | mAcc | |
| whole | 47.05 | 62.92 | 48.23 | 66.63 | 51.78 | 70.51 |
| part | 42.37 | 59.02 | 43.45 | 59.66 | 44.10 | 60.44 |
| Scale | w/o | w/o | w/o CSD | Ours | ||||
|---|---|---|---|---|---|---|---|---|
| mIoU | mAcc | mIoU | mAcc | mIoU | mAcc | mIoU | mAcc | |
| whole | 56.63 | 69.49 | 48.22 | 69.49 | 47.51 | 59.32 | 62.61 | 77.97 |
| part | 52.14 | 69.49 | 52.51 | 69.49 | 51.54 | 69.49 | 53.68 | 74.58 |
| Scene | Annotated Object Categories |
|---|---|
| figurines | jake, pirate hat, pikachu, rubber duck with hat, porcelain hand, red apple, tesla door handle, waldo, bag, toy cat statue, miffy, green apple, pumpkin, rubics cube, old camera, rubber duck with buoy, red toy chair, pink ice cream, spatula, green toy chair , and toy elephant . |
| teatime | sheep, yellow pouf, stuffed bear, coffee mug, tea in a glass, apple, coffee, hooves, bear nose, dall-e brand, plate, paper napkin, three cookies , and bag of cookies . |
| ramen | nori, sake cup, kamaboko, corn, spoon, egg, onion segments, plate, napkin, bowl, glass of water, hand, chopsticks , and wavy noodles . |
| waldo_kitchen | stainless steel pots, dark cup, refrigerator, frog cup, pot, spatula, plate, spoon, toaster, ottolenghi, plastic ladle, sink, ketchup, cabinet, red cup, pour-over vessel, knife , and yellow desk . |
| 19 Classes | wall, floor, cabinet, bed, chair, sofa, table, door, window, bookshelf, picture, counter, desk, curtain, refrigerator, shower curtain, toilet, sink, bathtub |
|---|---|
| 15 Classes | wall, floor, cabinet, bed, chair, sofa, table, door, window, bookshelf, counter, desk, toilet, sink, bathtub |
| 10 Classes | wall, floor, bed, chair, sofa, table, door, window, toilet, sink |
| Methods | Type | mIoU (whole) | mAcc (whole) | mIoU (part) | mAcc (part) |
|---|---|---|---|---|---|
| GarField | Fused | 49.91 | 65.62 | 40.15 | 59.14 |
| Ours | Cascaded | 51.78 | 70.51 | 44.10 | 60.44 |
| Dataset | SAM (s/img) | Scale | Training time (min) | VRAM (GB) | CSD | Denoising |
|---|---|---|---|---|---|---|
| Lerf_ovs | 26.85 | whole / part | 30 / 50 | 7.6 / 10.9 | 1.0 min | 1.2 s/cluster |
| ScanNet-v2 | 19.15 | whole / part / subpart | 20 / 30 / 30 | 9.9 / 14.4 / 16.8 | 1.0 min | 0.8 s/cluster |
| Metric | w/o | w/o | w/o CSD | w/o Denoising | Ours |
|---|---|---|---|---|---|
| mIoU | 27.61 | 27.34 | 26.70 | 28.24 | 28.99 |
| mAcc | 38.82 | 38.41 | 37.78 | 39.75 | 42.11 |
| Scale | figurines | ramen | teatime | waldo_kitchen | Overall |
|---|---|---|---|---|---|
| whole | 56/56 (100.0%) | 62/71 (87.3%) | 57/59 (96.6%) | 14/22 (63.6%) | 189/208 (90.9%) |
| part | 53/56 (94.6%) | 62/71 (87.3%) | 57/59 (96.6%) | 14/22 (63.6%) | 186/208 (89.4%) |
| whole+part | 56/56 (100.0%) | 64/71 (90.1%) | 59/59 (100.0%) | 16/22 (72.7%) | 195/208 (93.8%) |
| 0.00 | 0.05 | 0.10 | 0.15 | 0.20 | |
|---|---|---|---|---|---|
| w/ position clustering | 53.68 | 53.10 | 52.57 | 51.77 | 50.51 |
| w/o position clustering | 48.60 | 47.89 | 45.88 | 42.71 | 40.33 |
3D Semantic Segmentation
3D semantic segmentation assigns meaning to points, voxels, meshes, or reconstructed scene elements for spatial understanding.
See all 137 papers