mAP SSD vs SSD-EDL
Current State and Next Steps
Discussing the current limitations and suggesting further directions for experiments (29.03.2025)
Förderjahr 2023 / Stipendien Call #18 / ProjektID: 6885 / Projekt: Increasing Trustworthiness of Edge AI by Adding Uncertainty Estimation to Object Detection

Current experiments showed severe degradations in performance due to a combination of limitations in object detection architecture, training routine, and uncertainty estimation methods.

In our interim report, we conducted the first experiments on adding uncertainty estimation through Evidential Deep Learning for classification [6] to the Single Shot Detector (SSD) [3], as seen in the figure below:

mAP SSD vs SSD-EDL

Results showed that the detector was able to filter out uncertain predictions, resulting in increased precision (fewer false positives), however, the recall and also the overall mean Average Precision (mAP) decreased, showing limitations of directly applying EDL to object detection.

In our following experiments, we took another shot at adapting EDL to SSD by implementing a newer approach that uses Beta distributions for modeling evidence [4], compared to the default Dirichlet distribution. While Beta distributions resulted in a better mAP score than Dirichlet, both EDL versions still lacked mAP compared to the base SSD detector without uncertainty estimation.

Limitations

The insights gained so far suggested limitations in several possible ways:

  • EDL is not suitable for the SSD architecture, where different detectors might be more compatible.
  • Training an EDL network from scratch is not feasible, instead, it needs to learn on top of a pre-trained object detector.
  • EDL needs to be further adapted to make it more compatible with object detection.
  • Although promising, EDL might not be the overall best method for (fast) uncertainty estimation in object detection, alternatives may surpass it.

Next Steps

We tackle all those points by restructuring our experimental setup from the ground up. First, the state-of-the-art YOLO11 [2] object detection framework will be chosen as a base, coming with up-to-date architecture and training routines. Second, transfer learning will be applied to learn uncertainty estimation on top of an already well-performing object detector. Third, for EDL implementation, we switch to the approach of Park et al. [5], which learns EDL on a separate branch in the model output (head), leaving the base object detection intact, allowing it to build up on the existing performance level and improve from there. Lastly, we also include a well-known uncertainty estimation approach in our evaluation, Monte-Carlo Dropout [1], with its modified version Monte-Carlo Dropblock being applied in object detection [7]. While the Dropout technique performs well at estimating the uncertainty of the predictions, its downside is the significant computational overhead, which we will have to balance in order to make it a feasible solution for real-time edge devices with low computational power. The next blog post will feature updates on the performance of those new implementations.

References

[1] Yarin Gal and Zoubin Ghahramani. Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning. In Proceedings of The 33rd International Conference on Machine Learning, pages 1050–1059. PMLR, June 2016. ISSN: 1938-7228.

[2] Glenn Jocher and Jing Qiu. Ultralytics YOLO11, 2024. tex.orcid: 0000-0001-5950-6979, 0000-0002-7603-6750, 0000-0003-3783-7069.

[3] Wei Liu, Dragomir Anguelov, Dumitru Erhan, Christian Szegedy, Scott Reed, Cheng-Yang Fu, and Alexander C. Berg. SSD: Single Shot MultiBox Detector. In Computer Vision – ECCV 2016, pages 21–37. Springer International Publishing, 2016. ISSN: 1611-3349.

[4] Tejas Pandey, Nick Pears, William A P Smith, and John Alexander McDermid. E-DETR: Evidential deep learning for end-to-end uncertainty estimation in object detection, 2025.

[5] Younghyun Park, Wonjeong Choi, Soyeong Kim, Dong-Jun Han, and Jaekyun Moon. Active learning for object detection with evidential deep learning and hierarchical uncertainty aggregation. In The eleventh international conference on learning representations, 2023.

[6] Murat Sensoy, Lance Kaplan, and Melih Kandemir. Evidential deep learning to quantify classification uncertainty. In S. Bengio, H. Wallach, H. Larochelle, K. Grauman, N. Cesa-Bianchi, and R. Garnett, editors, Advances in neural information processing systems, volume 31. Curran Associates, Inc., 2018.

[7] Sai Harsha Yelleni, Deepshikha Kumari, Srijith P.K., and Krishna Mohan C. Monte Carlo DropBlock for modeling uncertainty in object detection. Pattern Recognition, 146:110003, February 2024.

Tags:

Machine Learning Object Detection Edge computing Uncertainty Estimation Evaluation Artificial Intelligence
CAPTCHA
Diese Frage dient der Überprüfung, ob Sie ein menschlicher Besucher sind und um automatisierten SPAM zu verhindern.

    Weitere Blogbeiträge

    Datenschutzinformation
    Der datenschutzrechtliche Verantwortliche (Internet Privatstiftung Austria - Internet Foundation Austria, Österreich) würde gerne mit folgenden Diensten Ihre personenbezogenen Daten verarbeiten. Zur Personalisierung können Technologien wie Cookies, LocalStorage usw. verwendet werden. Dies ist für die Nutzung der Website nicht notwendig, ermöglicht aber eine noch engere Interaktion mit Ihnen. Falls gewünscht, können Sie Ihre Einwilligung jederzeit via unserer Datenschutzerklärung anpassen oder widerrufen.