Bayesian Optimization-Aided Hybrid Deep Learning Model for Lightweight UAV-Based Smoke Detection

Uncrewed aerial vehicles (UAVs) play a crucial role in various applications, including detecting environmental hazards, e.g., wildfire smoke detection. However, the limited computational capabilities and battery life of UAVs present barriers to deploying complex artificial intelligence (AI) models onboard. To address this challenge, we propose a novel hybrid deep learning framework for UAVs to carry out light-weight yet efficient smoke detection. The framework combines a lightweight model for initial image assessment and a depth-wise model for selective processing of uncertain cases. Bayesian optimization is employed to determine the optimal threshold values for activating the depth-wise model, striking a balance between accuracy and computational efficiency. The proposed approach eliminates the need for cloud server connectivity, enabling onboard decision-making. Experimental results demonstrate that the hybrid framework achieves significant reductions in processing time and the number of calls to the depth-wise model while maintaining high accuracy. The framework’s adaptability and robustness make it suitable for real-time smoke detection applications in resource-constrained environments.