Bonnet: Ultra-fast whole-body bone segmentation from CT scans
Not on the map. Why?
Abstract (arXiv)
This work proposes Bonnet, an ultra-fast sparse-volume pipeline for whole-body bone segmentation from CT scans. Accurate bone segmentation is important for surgical planning and anatomical analysis, but existing 3D voxel-based models such as nnU-Net and STU-Net require heavy computation and often take several minutes per scan, which limits time-critical use. The proposed Bonnet addresses this by integrating a series of novel framework components including HU-based bone thresholding, patch-wise inference with a sparse spconv-based U-Net, and multi-window fusion into a full-volume prediction. Trained on TotalSegmentator and evaluated without additional tuning on RibSeg, CT-Pelvic1K, and CT-Spine1K, Bonnet achieves high Dice across ribs, pelvis, and spine while running in only 2.69 seconds per scan on an RTX A6000. Compared to strong voxel baselines, Bonnet attains a similar accuracy but reduces inference time by roughly 25x on the same hardware and tiling setup. The toolkit and pre-trained models will be released at https://github.com/HINTLab/Bonnet.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 15 of its 22 references that are in the map. First posted on arXiv 30 Jan 2026.
Builds on
15 of 22 references.
References · 15
- 2016PointNet: Deep Learning on Point Sets for 3D Classification and Segmentationcited by 5,223
- 2017PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Spacecited by 4,384
- 20173D Semantic Segmentation with Submanifold Sparse Convolutional Networkscited by 746
- 2022TotalSegmentator: robust segmentation of 104 anatomical structures in CT imagescited by 597
- 2020Searching Efficient 3D Architectures with Sparse Point-Voxel Convolutioncited by 422
- 2019From Points to Parts: 3D Object Detection from Point Cloud with Part-aware and Part-aggregation Networkcited by 366
- 2019Point-Voxel CNN for Efficient 3D Deep Learningcited by 343
- 2017Submanifold Sparse Convolutional Networkscited by 223
- 2019Modeling Point Clouds with Self-Attention and Gumbel Subset Samplingcited by 151
- 2023STU-Net: Scalable and Transferable Medical Image Segmentation Models Empowered by Large-Scale Supervised Pre-trainingcited by 96
- 2020Deep Learning to Segment Pelvic Bones: Large-scale CT Datasets and Baseline Modelscited by 45
- 2021CTSpine1K: A Large-Scale Dataset for Spinal Vertebrae Segmentation in Computed Tomographycited by 45
3 more
- 2021RibSeg Dataset and Strong Point Cloud Baselines for Rib Segmentation from CT Scanscited by 21
- 2022RibSeg v2: A Large-scale Benchmark for Rib Labeling and Anatomical Centerline Extractioncited by 10
- 2023Efficient Anatomical Labeling of Pulmonary Tree Structures via Deep Point-Graph Representation-based Implicit Fieldscited by 6
Built on by
Not cited yet.