On the Development of a Neural Network Architecture for Magnetometer-Based UXO Classification

The classification of Unexploded Ordnance (UXO) from magnetometer data is a critical but challenging task, frequently hindered by the data scarcity required for training robust machine learning models. To address this, we leverage a high-fidelity digital twin to generate a comprehensive dataset of magnetometer signals from both UXO and non-UXO objects, incorporating complex remanent magnetization effects. In this study, we design and evaluate a custom Convolutional Neural Network (CNN) for UXO classification and compare it against classical machine learning baseline, including Random Forest and kNN. Our CNN model achieves a balanced accuracy of 84.65%, significantly outperforming traditional models that exhibit performance collapse under slight distortions such as additive noise, drift, and time-wrapping. Additionally, we present a compact two-block CNN variant that retains competitive accuracy while reducing the number of learnable parameters by approximately 33%, making it suitable for real-time onboard classification in underwater vehicle missions. Through extensive ablation studies, we confirm that architectural components, such as residual skip connections and element-wise batch normalization, are crucial for achieving model stability and performance. The results also highlight the practical implications of underwater vehicles for survey design, emphasizing the need to mitigate signal drift and maintain constant survey speeds. This work not only provides a robust deep learning model for UXO classification, but also offers actionable suggestions for improving both model deployment and data acquisition protocols in the field.