Enhanced Transformer-based Deep Semantic Segmentation Architecture for Lidar 3D Point Clouds /
Mohammed Moustafa Mohamed Hassoubah
- 2022
- 94 p. ill. 21 cm.
Abstract: For the task of semantic segmentation of 2D or 3D inputs, transformer architecture suffers limitations in the ability of localization because of lacking low-level details. Also, for the transformer to function well, it has to be pre-trained first. Still pre-training transformers is an open area of research. In this work, we introduce a novel architecture for semantic segmentation of 3D point clouds generated from Light Detection and Ranging (LiDAR) sensors. A transformer is integrated into the U-Net 2D segmentation network [1] and the new architecture is trained to conduct semantic segmentation of 2D spherical images generated from projecting 3D LiDAR point clouds. Such integration allows capturing the local and region level dependencies from CNN backbone processing of the input, followed by transformer processing to capture the long range dependencies. Obtained results demonstrate that the new architecture provides enhanced segmentation results over existing state-of-theart approaches. Furthermore, to define the best pre-training settings, multiple ablations have been conducted on network architecture, self-training loss function and self-training procedures. It’s proved that, the integrated architecture, pre-trained over the augmented version of the training dataset to reconstruct the original data from the corrupted input, while randomly initializing the batch normalization layers when fine-tuning, all these together outperforms the SalsaNext [2] (to our knowledge it’s the best projection based semantic segmentation network), where results are reported on the SemanticKITTI [3] iv validation dataset with 2D input dimension 1024 × 64. In our evaluation it is found that self-supervision pre-training much reduces the epistemic uncertainty (the model weights uncertainty ex.introducing new input example to the model different from those seen in training dataset would increase such uncertainty) of the output of the segmentation model.