Uncertainty Quantification In Surface Landmines and UXO Classification Using MC Dropout
By: Sagar Lekhak , Emmett J. Ientilucci , Dimah Dera and more
Potential Business Impact:
Finds hidden bombs more reliably, even in bad conditions.
Detecting surface landmines and unexploded ordnances (UXOs) using deep learning has shown promise in humanitarian demining. However, deterministic neural networks can be vulnerable to noisy conditions and adversarial attacks, leading to missed detection or misclassification. This study introduces the idea of uncertainty quantification through Monte Carlo (MC) Dropout, integrated into a fine-tuned ResNet-50 architecture for surface landmine and UXO classification, which was tested on a simulated dataset. Integrating the MC Dropout approach helps quantify epistemic uncertainty, providing an additional metric for prediction reliability, which could be helpful to make more informed decisions in demining operations. Experimental results on clean, adversarially perturbed, and noisy test images demonstrate the model's ability to flag unreliable predictions under challenging conditions. This proof-of-concept study highlights the need for uncertainty quantification in demining, raises awareness about the vulnerability of existing neural networks in demining to adversarial threats, and emphasizes the importance of developing more robust and reliable models for practical applications.
Similar Papers
An Empirical Study on MC Dropout--Based Uncertainty--Error Correlation in 2D Brain Tumor Segmentation
Machine Learning (CS)
Finds where computer mistakes tumor scans.
Unreliable Uncertainty Estimates with Monte Carlo Dropout
Machine Learning (CS)
Makes AI less sure when it might be wrong.
A Quad-Step Approach to Uncertainty-Aware Deep Learning for Skin Cancer Classification
CV and Pattern Recognition
Helps doctors spot skin cancer more accurately.