基于Transformer-SHAP框架揭示气象因素对多层土壤湿度预测的贡献与滞后效应

《Journal of Hydrology》:Unraveling meteorological contributions and lagged effects on multi-depth soil moisture prediction using Transformer-SHAP framework

【字体: 大 中 小 】 时间:2026年09月08日 来源:Journal of Hydrology 7.3

编辑推荐:

   •开发Transformer-SHAP框架以预测和解释土壤水分(SM)。•量化气象因素对多层深土壤水分的贡献及滞后效应。•为土壤缓冲效应和SM过程提供可解释的证据。引言土壤水分(SM)是连接陆-气交换、植被功能和陆面水循环的关键生态水文变量(Yu et al., 2021a)

  
  • •
    开发Transformer-SHAP框架以预测和解释土壤水分(SM)。
  • •
    量化气象因素对多层深土壤水分的贡献及滞后效应。
  • •
    为土壤缓冲效应和SM过程提供可解释的证据。

引言

土壤水分(SM)是连接陆-气交换、植被功能和陆面水循环的关键生态水文变量(Yu et al., 2021a)。其时间变化影响蒸散分配、陆地表层附近温度条件、干旱发展和生态系统生产力(Seneviratne et al., 2010)。重要的是,SM动态随土壤深度显著变化。表层SM通常对降水脉冲和大气蒸散发需求快速响应,而深层SM受土壤蓄水、水力再分配、根系吸水及延迟垂直传输过程的影响更为显著。这些与深度相关的过程塑造了SM的记忆特征,导致在相同气象驱动条件下不同土层之间响应幅度与时间尺度的差异(Guswa et al., 2002, Rahmati et al., 2024)。因此,准确预测多个土壤深度上的SM对于理解根区水分可用性、干旱演化和生态系统水分调节至关重要。
已有多种方法被开发用于模拟或预测土壤剖面上SM动态。在土壤剖面尺度上,基于物理过程的模型如HYDRUS 1D通过求解Richards方程模拟非饱和或变饱和土壤中的水分运动。通过对土壤剖面进行数值离散化,这些模型表征了入渗、根系吸水、排水和垂直再分配过程(Richards, 1931, ?im?nek et al., 2008)。在更大空间尺度上,陆面模型和水文模型通常将土壤柱划分为多个层次,通过将垂直水分运动、蒸散发、径流、地下水交换、植被过程和地表能量平衡耦合来模拟不同深度的SM。例如,可变入渗容量模型(VIC)通常将土壤剖面划分为若干层,入渗水首先进入上层土壤,蒸散发主要发生在根区内,重力驱动的排水将水分输送至更深土层(Liang et al., 1994)。社区陆面模型第5版(CLM5)和Noah MP进一步将多层土壤水文过程与植被过程及陆-气能量交换耦合(Niu et al., 2011, Lawrence et al., 2019)。这些模型为土壤水分运动提供了物理意义上的表征,在研究土壤水文过程和陆面-大气相互作用方面发挥着重要作用。
然而,基于物理过程的模型在单个监测站的应用可能受限于其对土壤水力特性详细信息、不同土壤层初始条件、上下边界条件、根系分布、植被参数和校准数据的需求。当土壤水力特性和根系特征在垂直或空间方向上存在变化时,这些参数的不确定性可能传播至模拟的SM结果中。此外,预定义的土层结构、对局部异质性的简化表征以及模型参数化与单点观测之间的尺度不匹配可能限制这些模型在特定站点的SM动态再现能力。因此,数据驱动方法可以作为有用的补充工具,因为它们可以直接从气象和SM观测时间序列中学习非线性和滞后关系。然而,不应将其视为基于物理过程模型的替代方案。
随着机器学习和深度学习方法的不断发展,在SM预测方面已取得重大进展。早期的数据驱动研究通常采用人工神经网络、支持向量机、随机森林等传统机器学习模型来刻画气象变量与SM之间的非线性关系。循环神经网络,特别是长短期记忆网络(LSTM)和门控循环单元(GRU),随后因其循环结构能够保留前序信息并表征SM时间序列中的时序依赖关系而受到广泛关注(Yu et al., 2021b)。近年来,基于注意力机制和Transformer架构的模型也被应用于水文和环境时间序列预测。其自注意力机制可以直接评估不同时间步之间的关系,从而在不完全依赖顺序循环处理的情况下识别有用的历史信息(Vaswani et al., 2017, Wu et al., 2021, Zhou et al., 2021, Orozco López et al., 2024)。近期研究比较了Transformer、LSTM和耦合架构;将基于物理过程的模拟与深度学习模型整合;探索用于SM预测的时间序列基础模型;以及将神经算子与滑动窗口策略相结合用于地下SM重建和预测(Li et al., 2024, Wang and Zha, 2024, Singh et al., 2025)。这些研究表明人工智能模型在表征非线性关系、长期时序依赖性和多环境变量相互作用方面具备日益增强的能力。
尽管取得了这些进展,多深度SM预测仍存在一些局限性。首先,许多现有研究主要关注单一土层或强调整体预测精度,而不同土壤深度之间气象贡献的比较仍然有限。其次,历史输入窗口的长度通常仅被视为模型超参数,其与土壤深度的变化鲜少结合不同深度的SM记忆以及对前序气象条件的延迟响应进行解释。第三,深度学习模型的复杂内部结构使得确定单个气象预测因子如何影响特定预测变得困难,包括其贡献的方向、幅度、非线性特征和样本依赖性。虽然注意力机制可以识别重要的时序信息,但仅凭注意力权重无法提供预测因子贡献的完整定量归因。因此,需要一种可解释的框架来同时评估预测性能、识别有用的前序时间窗口并量化模型在不同土壤深度产生的气象贡献。
可解释人工智能为提升深度学习模型的可解释性提供了途径。现有的解释方法包括排列特征重要性、部分依赖图(PDP)、局部可解释模型无关解释(LIME)、SHAP等方法。其中,SHAP基于合作博弈论,能够提供全局特征重要性和样本级归因结果。这使得可以量化每个气象预测因子在不同土层上的贡献幅度和方向,从而支持对复杂模型的面向机制的解释(Lundberg and Lee, 2017, Ba?a?ao?lu et al., 2022)。因此,本研究开发了一种基于Transformer的模型用于日尺度多层深SM预测,并使用SHAP作为可解释性工具诊断气象预测因子贡献的深度依赖差异。我们旨在解决两个关键挑战:刻画SM对前序气象信息深度依赖的时序关系以及提高多层深SM预测中预测因子的可解释性。为此,针对不同土层构建了深度专属的Transformer模型,并使用深度自适应输入窗口来捕获表层快速响应和深层更缓慢、更具持续性的动态。同时,将基于SHAP的特征归因与预测结果相结合,为气象贡献的垂直变化提供可解释的见解。该框架不仅提高了多层深SM动态的预测精度,还有助于刻画滞后响应,并对深度依赖的SM行为提供更具可解释性的理解。

章节片段

研究区

研究站点位于俄克拉荷马州立大学研究牧场,地处美国南大平原东部森林向西过渡至西部草原的植被生态过渡带,位于俄克拉荷马州斯蒂尔沃特约11公里西南方向(图1)。该地区具有典型的大陆性气候,年均降水量875毫米,年均气温15.5°C,平均湿度69%(Liu et al., 2017)。自20世纪30年代以来,该地区经历了

SM数据

SM数据于2010年8月16日至2016年10月30日按日尺度采集。数据来源于2010年在草原区安装8个SM站所获取的数据。每个SM站设有4支水平安装的ECH2O EC-5 SM探针,分别置于5、20、45和80厘米土壤深度,配备EM50数据记录仪(Zou et al., 2014)。各站点位于同一草原监测区域内,间距从数十米到数百米不等,且土壤类型大致相似

不同土壤深度的模型性能

图7展示了四个土壤深度(5、20、45和80厘米)的观测值和预测值。图8展示了对应的散点图,其中蓝色虚线代表1:1参考线。散点越接近该线表明预测值与观测值之间的一致性越好。在训练期内,模型表现出较高的预测性能,散点密集分布在1:1线附近。在5、20厘米处的NSE值分别为0.93、0.99、0.98和0.96,

结论

本研究针对不同土壤深度开发了Transformer-SHAP框架,用于预测日尺度SM并解释土壤剖面中气象贡献和滞后效应。结果表明,Transformer在5、20、45和80厘米处的SM预测表现可靠。最优历史输入窗口随土壤深度逐步增加,表明深层SM对更长时间前序期内累积的气象信息具有依赖性。这一模式与

CRediT作者贡献声明

董建康:撰写—审阅与编辑,撰写—原稿撰写,可视化,验证,软件,资源,方法论,形式化分析,数据管理,概念化。夏文雪:撰写—审阅与编辑,可视化,指导监督,软件,资源。郝慧晴:撰写—审阅与编辑。练怡:指导监督。余富荣:指导监督。郝永红:指导监督,获取研究资助。Tian-Chyi Jim Yeh:指导监督。

利益冲突声明

作者声明不存在可能影响本文所报告工作的已知竞争性经济利益或个人关系。

致谢

郝永红的工作部分得到国家自然科学基金42577059的支持。作者亦衷心感谢四位匿名审稿人提出的详细且具有建设性的意见,以改进本文稿的质量。
董建康|夏文雪|郝慧晴|练怡|余富荣|郝永红|Tian-Chyi Jim Yeh
相关新闻
生物通微信公众号
微信
新浪微博
  • 搜索
  • 国际
  • 国内
  • 人物
  • 产业
  • 热点
  • 科普

热点排行

    今日动态 | 人才市场 | 新技术专栏 | 中国科学人 | 云展台 | BioHot | 云讲堂直播 | 会展中心 | 特价专栏 | 技术快讯 | 免费试用

    版权所有 生物通

    Copyright© eBiotrade.com, All Rights Reserved

    联系信箱:

    粤ICP备09063491号