用强化学习设计小模型,实现卫星火情实时检测。
Designing a Classifier for Active Fire Detection from Multispectral Satellite Imagery Using Neural Architecture Search
- 用强化学习搜索适合卫星的微型神经网络架构。
- 模型仅1716参数,推理仅需984微秒,功耗约800毫瓦。
- 适合资源受限的低轨纳米卫星,可实现实时火情识别。
本文展示了一种基于强化学习的神经架构搜索(NAS)代理,用于设计小型神经网络,以在多光谱卫星图像中进行活跃火情检测。目标是设计一个神经网络,判断单个像素是否属于火区,并在低地球轨道(LEO)纳米卫星有限功耗预算的约束下,实现传感器数据的机载处理。为使用强化学习,需定义奖励函数,本文通过回归模型预测特定架构在量化至INT8精度后的F1得分,该模型由随机采样的神经网络架构训练并收集其分类性能统计得到。除F1分数外,奖励函数还包含可训练参数总数,以限制模型大小,确保符合纳米卫星平台的资源约束。最终将最优神经网络部署于Google Coral Micro Dev Board,评估其推理延迟与功耗:模型含1,716个可训练参数,平均推理时间为984μs,推理功耗约为800mW。结果表明,基于强化学习的NAS方法可成功应用于此前未解决的新问题。
原文摘要 · Abstract (English)
This paper showcases the use of a reinforcement learning-based Neural Architecture Search (NAS) agent to design a small neural network to perform active fire detection on multispectral satellite imagery. Specifically, we aim to design a neural network that can determine if a single multispectral pixel is a part of a fire, and do so within the constraints of a Low Earth Orbit (LEO) nanosatellite with a limited power budget, to facilitate on-board processing of sensor data. In order to use reinforcement learning, a reward function is needed. We supply this reward function in the shape of a regression model that predicts the F1 score obtained by a particular architecture, following quantization to INT8 precision, from purely architectural features. This model is trained by collecting a random sample of neural network architectures, training these architectures, and collecting their classification performance statistics. Besides the F1 score, we also include the total number of trainable parameters in our reward function to limit the size of the designed model and ensure it fits within the resource constraints imposed by nanosatellite platforms. Finally, we deployed the best neural network to the Google Coral Micro Dev Board and evaluated its inference latency and power consumption. This neural network consists of 1,716 trainable parameters, takes on average 984μs to inference, and consumes around 800mW to perform inference. These results show that our reinforcement learning-based NAS approach can be successfully applied to novel problems not tackled before.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。