Understanding the Challenges When 3D Semantic Segmentation Faces Class Imbalanced and OOD Data
Not on the map. Why?
Abstract (arXiv)
3D semantic segmentation (3DSS) is an essential process in the creation of a safe autonomous driving system. However, deep learning models for 3D semantic segmentation often suffer from the class imbalance problem and out-of-distribution (OOD) data. In this study, we explore how the class imbalance problem affects 3DSS performance and whether the model can detect the category prediction correctness, or whether data is ID (in-distribution) or OOD. For these purposes, we conduct two experiments using three representative 3DSS models and five trust scoring methods, and conduct both a confusion and feature analysis of each class. Furthermore, a data augmentation method for the 3D LiDAR dataset is proposed to create a new dataset based on SemanticKITTI and SemanticPOSS, called AugKITTI. We propose the wPre metric and TSD for a more in-depth analysis of the results, and follow are proposals with an insightful discussion. Based on the experimental results, we find that: (1) the classes are not only imbalanced in their data size but also in the basic properties of each semantic category. (2) The intraclass diversity and interclass ambiguity make class learning difficult and greatly limit the models' performance, creating the challenges of semantic and data gaps. (3) The trust scores are unreliable for classes whose features are confused with other classes. For 3DSS models, those misclassified ID classes and OODs may also be given high trust scores, making the 3DSS predictions unreliable, and leading to the challenges in judging 3DSS result trustworthiness. All of these outcomes point to several research directions for improving the performance and reliability of the 3DSS models used for real-world applications.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 49 of its 76 references that are in the map. First posted on arXiv 1 Mar 2022.
Builds on
49 of 76 references.
References · 49
- 2016PointNet: Deep Learning on Point Sets for 3D Classification and Segmentationcited by 5,223
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2019nuScenes: A multimodal dataset for autonomous drivingcited by 4,485
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2017PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Spacecited by 4,384
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
- 2018A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attackscited by 1,472
- 2017Enhancing The Reliability of Out-of-distribution Image Detection in Neural Networkscited by 1,325
- 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,274
- 2019Class-Balanced Loss Based on Effective Number of Samplescited by 1,239
37 more
- 2019KPConv: Flexible and Deformable Convolution for Point Cloudscited by 1,202
- 2020Meta-Learning in Neural Networks: A Surveycited by 944
- 2018Deep Anomaly Detection with Outlier Exposurecited by 939
- 2014Deep Neural Networks are Easily Fooled: High Confidence Predictions for Unrecognizable Imagescited by 774
- 20173D Semantic Segmentation with Submanifold Sparse Convolutional Networkscited by 746
- 2016Multi-View 3D Object Detection Network for Autonomous Drivingcited by 721
- 2018PointConv: Deep Convolutional Networks on 3D Point Cloudscited by 656
- 2019RandLA-Net: Efficient Semantic Segmentation of Large-Scale Point Cloudscited by 584
- 2020Learning from Noisy Labels with Deep Neural Networks: A Surveycited by 580
- 2019Deep Learning for 3D Point Clouds: A Surveycited by 557
- 2017A systematic study of the class imbalance problem in convolutional neural networkscited by 510
- 2019Generalizing from a Few Examples: A Survey on Few-Shot Learningcited by 462
- 2021Deep Long-Tailed Learning: A Surveycited by 416
- 2020Are we done with ImageNet?cited by 329
- 2017SqueezeSeg: Convolutional Neural Nets with Recurrent CRF for Real-Time Road-Object Segmentation from 3D LiDAR Point Cloudcited by 231
- 2018SqueezeSegV2: Improved Model Structure and Unsupervised Domain Adaptation for Road-Object Segmentation from a LiDAR Point Cloudcited by 228
- 2019Learning to Discover Novel Visual Categories via Deep Transfer Clusteringcited by 223
- 2018Deep Learning: A Critical Appraisalcited by 214
- 2018To Trust Or Not To Trust A Classifiercited by 205
- 2017SEGCloud: Semantic Segmentation of 3D Point Cloudscited by 178
- 2020SqueezeSegV3: Spatially-Adaptive Convolution for Efficient Point-Cloud Segmentationcited by 170
- 2021A Unified Survey on Anomaly, Novelty, Open-Set, and Out-of-Distribution Detection: Solutions and Future Challengescited by 154
- 2020SemanticPOSS: A Point Cloud Dataset with Large Quantity of Dynamic Instancescited by 123
- 2021Re-labeling ImageNet: from Single to Multi-Labels, from Global to Localized Labelscited by 117
- 2020Cylinder3D: An Effective 3D Framework for Driving-scene LiDAR Semantic Segmentationcited by 112
- 2016Vehicle Detection from 3D Lidar Using Fully Convolutional Networkcited by 105
- 2018PointSIFT: A SIFT-like Network Module for 3D Point Cloud Semantic Segmentationcited by 58
- 2018Fully-Convolutional Point Networks for Large-Scale Point Cloudscited by 44
- 2018Exploring Spatial Context for 3D Semantic Segmentation of Point Cloudscited by 33
- 2019Linking Points With Labels in 3D: A Review of Point Cloud Semantic Segmentationcited by 29
- 2018Know What Your Neighbors Do: 3D Semantic Segmentation of Point Cloudscited by 22
- 2018PointSeg: Real-Time Semantic Segmentation Based on 3D LiDAR Point Cloudcited by 19
- 2019DeepTemporalSeg: Temporally Consistent Semantic Segmentation of 3D LiDAR Scanscited by 13
- 2020Are We Hungry for 3D LiDAR Data for Semantic Segmentation? A Survey and Experimental Studycited by 12
- 2020Compositional Prototype Network with Multi-view Comparision for Few-Shot Point Cloud Semantic Segmentationcited by 12
- 2019PyramNet: Point Cloud Pyramid Attention Network and Graph Embedding Module for Classification and Segmentationcited by 6
- 2021On the combined effect of class imbalance and concept complexity in deep learningcited by 2
Built on by
Citing papers per year: 2024: 2; 2025: 1; 2026: 1.
4 papers.
- 2024The METRIC-framework for assessing data quality for trustworthy AI in medicine: a systematic reviewcited by 13
- 2025CrossFuse: Learning Infrared and Visible Image Fusion by Cross-Sensor Top-K Vision Alignment and Beyondcited by 5
- 2024Semantic Scene Completion with Multi-Feature Data Balancing Networkcited by 0
- 2026How Sampling Strategy Affects Imbalance Mitigation in LiDAR Segmentation: A Study of Structured vs. Random Point-Based Architecturescited by 0
- 2026How Sampling Strategy Affects Imbalance Mitigation in LiDAR Segmentation: A Study of Structured vs. Random Point-Based Architectures17 Aug 2026
- 2025CrossFuse: Learning Infrared and Visible Image Fusion by Cross-Sensor Top-K Vision Alignment and Beyond20 Feb 2025
- 2024Semantic Scene Completion with Multi-Feature Data Balancing Network2 Dec 2024
- 2024The METRIC-framework for assessing data quality for trustworthy AI in medicine: a systematic review21 Feb 2024