Apr, 2024

可解释的人工智能整合特征工程用于野火预测

TL;DR通过深入研究,我们评估了各种机器学习算法在预测野火相关任务中的分类和回归任务方面的有效性。我们发现,对于不同类型或阶段的野火分类,XGBoost 模型在准确性和稳健性方面优于其他模型。与此同时,Random Forest 回归模型在预测受野火影响范围方面展现出更好的结果,同时在预测误差和解释方差两方面表现出色。此外,我们还开发了一个集成数值数据和图像信息的混合神经网络模型,用于同时进行分类和回归任务。为了深入了解这些模型的决策过程并确定重要的特征,我们利用了可解释的人工智能(XAI)技术,包括 TreeSHAP、LIME、Partial Dependence Plots(PDP)和 Gradient-weighted Class Activation Mapping(Grad-CAM)。这些解释工具揭示了各种特征的重要性和相互作用,突出了影响野火预测的复杂因素。我们的研究不仅证明了特定机器学习模型在野火相关任务中的有效性,而且强调了模型透明度和可解释性在环境科学应用中的关键作用。