arXiv:2502.20979cs.CV2025-02被引 13

用知识蒸馏压缩模型,让无人机实时识别火灾

Real-Time Aerial Fire Detection on Resource-Constrained Devices Using Knowledge Distillation

  • 用强教师模型蒸馏轻量MobileViT-S,提升小设备性能
  • 在火情数据集上准确率比顶尖模型高0.44%,速度最快
  • 适合部署在无人机、卫星等算力受限的设备上

野火灾害造成严重环境破坏、人员伤亡和经济损失。为减轻影响,早期火情检测系统至关重要。现有系统多依赖固定摄像头,视野有限。将智能检测与遥感结合可提升覆盖范围与移动性,适用于偏远难测区域。当前方法主要采用卷积神经网络与视觉变换器,虽精度高,但计算复杂,难以在无人机等边缘设备实现实时运行。本文提出基于MobileViT-S的轻量化火情检测模型,通过强教师模型的知识蒸馏进行压缩。消融实验表明教师模型与蒸馏策略对性能提升显著。利用Grad-CAM生成激活图,验证模型能聚焦于火源区域。实验结果表明,该模型在常见火情基准测试中准确率优于现有最优模型0.44%、2.00%,同时保持极小模型体积。其处理速度在同类工作中最高,可在资源受限设备上实现真正实时检测,适用于卫星、无人机及物联网设备部署。

原文摘要 · Abstract (English)

Wildfire catastrophes cause significant environmental degradation, human losses, and financial damage. To mitigate these severe impacts, early fire detection and warning systems are crucial. Current systems rely primarily on fixed CCTV cameras with a limited field of view, restricting their effectiveness in large outdoor environments. The fusion of intelligent fire detection with remote sensing improves coverage and mobility, enabling monitoring in remote and challenging areas. Existing approaches predominantly utilize convolutional neural networks and vision transformer models. While these architectures provide high accuracy in fire detection, their computational complexity limits real-time performance on edge devices such as UAVs. In our work, we present a lightweight fire detection model based on MobileViT-S, compressed through the distillation of knowledge from a stronger teacher model. The ablation study highlights the impact of a teacher model and the chosen distillation technique on the model's performance improvement. We generate activation map visualizations using Grad-CAM to confirm the model's ability to focus on relevant fire regions. The high accuracy and efficiency of the proposed model make it well-suited for deployment on satellites, UAVs, and IoT devices for effective fire detection. Experiments on common fire benchmarks demonstrate that our model suppresses the state-of-the-art model by 0.44%, 2.00% while maintaining a compact model size. Our model delivers the highest processing speed among existing works, achieving real-time performance on resource-constrained devices.

火情检测知识蒸馏边缘计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。