English

Probabilistic Object Classification using CNN ML-MAP layers

Machine Learning 2020-08-25 v2 Computer Vision and Pattern Recognition Machine Learning

Abstract

Deep networks are currently the state-of-the-art for sensory perception in autonomous driving and robotics. However, deep models often generate overconfident predictions precluding proper probabilistic interpretation which we argue is due to the nature of the SoftMax layer. To reduce the overconfidence without compromising the classification performance, we introduce a CNN probabilistic approach based on distributions calculated in the network's Logit layer. The approach enables Bayesian inference by means of ML and MAP layers. Experiments with calibrated and the proposed prediction layers are carried out on object classification using data from the KITTI database. Results are reported for camera (RGBRGB) and LiDAR (range-view) modalities, where the new approach shows promising performance compared to SoftMax.

Keywords

Cite

@article{arxiv.2005.14565,
  title  = {Probabilistic Object Classification using CNN ML-MAP layers},
  author = {G. Melotti and C. Premebida and J. J. Bird and D. R. Faria and N. Gonçalves},
  journal= {arXiv preprint arXiv:2005.14565},
  year   = {2020}
}

Comments

Accepted at the Workshop on Perception for Autonomous Driving (PAD), European Conference on Computer Vision (ECCV) 2020

R2 v1 2026-06-23T15:54:36.865Z