RibSeg Dataset and Strong Point Cloud Baselines for Rib Segmentation from CT Scans

Jiancheng Yang,Shixuan Gu,Donglai Wei,Hanspeter Pfister,Bingbing Ni

DOI: https://doi.org/10.1007/978-3-030-87193-2_58

2021-09-18

Abstract:Manual rib inspections in computed tomography (CT) scans are clinically critical but labor-intensive, as 24 ribs are typically elongated and oblique in 3D volumes. Automatic rib segmentation methods can speed up the process through rib measurement and visualization. However, prior arts mostly use in-house labeled datasets that are publicly unavailable and work on dense 3D volumes that are computationally inefficient. To address these issues, we develop a labeled rib segmentation benchmark, named \emph{RibSeg}, including 490 CT scans (11,719 individual ribs) from a public dataset. For ground truth generation, we used existing morphology-based algorithms and manually refined its results. Then, considering the sparsity of ribs in 3D volumes, we thresholded and sampled sparse voxels from the input and designed a point cloud-based baseline method for rib segmentation. The proposed method achieves state-of-the-art segmentation performance (Dice~$\approx95\%$) with significant efficiency ($10\sim40\times$ faster than prior arts). The RibSeg dataset, code, and model in PyTorch are available at <a class="link-external link-https" href="https://github.com/M3DV/RibSeg" rel="external noopener nofollow">this https URL</a>.

Image and Video Processing,Artificial Intelligence,Computer Vision and Pattern Recognition,Graphics,Machine Learning

What problem does this paper attempt to address?

The problem this paper attempts to address is:

In clinical practice, manually inspecting ribs in CT scans is a time-consuming and labor-intensive process. Although automated rib segmentation methods can speed up this process, existing methods mostly rely on internally annotated datasets that are not available to the public and are computationally inefficient when processing dense 3D volumetric data. To address these issues, the authors developed an annotated dataset named RibSeg, which includes 490 CT scans (a total of 11,719 individual ribs), and designed a point cloud-based baseline method to achieve rib segmentation.

Specifically, the paper mainly addresses the following issues:
1. **Lack of publicly annotated datasets**: Existing rib segmentation studies mostly use internal datasets, making it impossible to conduct fair benchmarking and method comparison.
2. **Low computational efficiency**: Existing methods are computationally inefficient when processing dense 3D volumetric data, requiring a significant amount of time and computational resources.
3. **Accuracy of rib segmentation**: Improving the accuracy of rib segmentation to better support downstream applications such as rib fracture detection and bone lesion diagnosis.

By developing the RibSeg dataset and proposing an efficient point cloud baseline method, the paper aims to provide a public, efficient, and accurate solution to promote the development of detection and diagnosis technologies for rib-related diseases.