China Deep Learning Distributed Hydrologic Model (CDLM) 全国物理-数据双驱动分布式水文模型 (CDLM)

A national high-resolution distributed hydrologic modeling framework coupling differentiable physical operators with neural regionalization.

一个面向全国尺度高分辨率水文模拟的分布式模型框架，耦合可微物理算子与神经网络参数区域化方法。

Home首页 / CDLM

Model Architecture模型架构

CDLM is developed for national-scale high-resolution streamflow simulation across China. The framework couples 3-km gridded GR4J runoff generation and 1-km Muskingum river routing as recurrent differentiable operators, while neural parameter generators infer spatially varying runoff and routing parameters from dynamic forcings and static landscape attributes. The full runoff-routing chain is optimized through streamflow errors at gauges, enabling process-embedded regionalization for ungauged basins.

CDLM 面向全国尺度高分辨率径流模拟而开发。该框架将 3 km 网格 GR4J 产流模块与 1 km Muskingum 河道汇流模块构建为递归可微算子，并通过神经网络参数生成器从动态气象驱动和静态地理属性中推断空间变化的产流与汇流参数。完整的产流-汇流链条通过水文站流量误差进行优化，从而支持面向无资料流域的过程嵌入式参数区域化。

CDLM model architecture diagram from the manuscript

Architecture of CDLM extracted from the manuscript. National spatial discretization, dynamic forcings, static attributes, differentiable GR4J runoff generation, 1-km Muskingum routing, neural parameterization, streamflow loss, and back-propagation are integrated in one end-to-end computational graph.

图为论文中的 CDLM 模型架构。全国空间离散、动态气象驱动、静态属性、可微 GR4J 产流、1 km Muskingum 汇流、神经网络参数化、流量损失函数和反向传播被整合在同一个端到端计算图中。

3-km GR4J runoff RNN3 km GR4J 产流 RNN 1-km Muskingum routing RNN1 km Muskingum 汇流 RNN Static-attribute parameter generators静态属性参数生成器 Dynamic PET correction动态 PET 校正 Back-propagation through physics物理过程中的反向传播

Core Ideas核心思想

Physics-encoded differentiability物理过程可微化

Instead of treating runoff generation and routing as black-box predictors, CDLM implements GR4J and Muskingum equations as differentiable recurrent operators. This keeps the model interpretable while allowing gradient-based training at national scale.

CDLM 不将产流和汇流视为黑箱预测器，而是将 GR4J 和 Muskingum 方程实现为可微递归算子，在保留物理可解释性的同时支持全国尺度的梯度优化训练。

Neural regionalization神经网络区域化

Model parameters are generated from grid-cell and river-reach attributes. This shifts regionalization from donor-parameter transfer toward spatially continuous parameter learning for gauged and ungauged basins.

模型参数由网格单元和河道河段属性生成，使参数区域化从传统的相似流域参数移植，转向面向有资料和无资料流域的空间连续参数学习。

Runoff-routing coupling产汇流联合建模

Runoff from 3-km cells is mapped to a 1-km river network and routed downstream. The runoff generator and routing model are calibrated jointly rather than as isolated components.

3 km 网格产流被映射到 1 km 河网并向下游汇流。产流模块和汇流模块在同一计算链条中联合校准，而不是作为彼此独立的模块处理。

National training data全国尺度训练数据

The framework uses streamflow observations from 1,167 near-natural gauges in China to learn parameter generators and evaluate spatial transfer performance.

该框架利用中国 1,167 个近自然水文站流量观测学习参数生成器，并评估模型在空间迁移和无资料流域模拟中的表现。

CDLM Code ArchiveCDLM 代码归档

cdlm.zip

ZIP
Size大小
124 KB
Files文件数
45
Version版本
1.0
Modified更新时间
2026-06-04
SHA-256
E053E3ACD0144DFDE2D84E6DF55678593BE824850AB4BCDB85ED86B2BA73B75E
License许可协议
CC BY-NC-ND 4.0

The active public download will be enabled upon manuscript acceptance.

论文接收后将开放公众下载。

Package structure压缩包结构 Size大小 Modified修改时间
  • ▣cdlm.zip124 KB2026-06-04 15:55
  • ▸upload/-2026-06-04 15:52
  • ·  README.md1.9 KB2026-06-04 15:47
  • ·  pyproject.toml390 B2026-06-04 15:47
  • ▸  configs/-2026-06-04 15:46
  • ·    train_example.toml1.8 KB2026-06-04 15:46
  • ▸  data/-2026-06-04 15:47
  • ·    README.md906 B2026-06-04 15:47
  • ▸    sample/-2026-06-04 15:47
  • ·      .gitkeep1 B2026-06-04 15:47
  • ▸  scripts/-2026-06-04 15:47
  • ·    build_pm_forcing_cache.sh248 B2026-06-04 15:47
  • ·    export_checkpoint_outputs.sh230 B2026-06-04 15:47
  • ·    train.sh2.4 KB2026-06-04 15:47
  • ▸  src/-2026-06-04 15:44
  • ▸    diff_gr4j_reachkx/-2026-06-04 15:49
  • ·      __init__.py1.4 KB2026-05-30 12:51
  • ·      attributes.py7.3 KB2026-05-30 12:51
  • ·      build_3k_cell_attributes.py2.8 KB2026-06-04 14:12
  • ·      build_basin_attributes.py929 B2026-05-30 12:51
  • ·      build_basin_hierarchy.py2.6 KB2026-06-04 14:13
  • ·      build_basin_runoff_attributes.py3.4 KB2026-05-30 12:51
  • ·      build_forcing_cache.py2.1 KB2026-06-04 14:12
  • ·      build_obs_tensor.py6.4 KB2026-05-30 12:51
  • ·      build_pm_pet_cache.py1.7 KB2026-05-30 12:51
  • ·      build_precip_seasonality.py8.5 KB2026-05-30 12:51
  • ·      build_reach_attributes.py9.4 KB2026-06-04 14:12
  • ·      build_runoff_signature_target.py6.2 KB2026-06-04 14:12
  • ·      compute_detailed_station_metrics.py10.6 KB2026-05-31 10:17
  • ·      compute_metrics_from_streamflow.py9.3 KB2026-05-30 12:51
  • ·      config.py2.7 KB2026-05-30 12:51
  • ·      data_config.py4.5 KB2026-06-04 15:45
  • ·      export_checkpoint_outputs.py25.6 KB2026-06-04 14:13
  • ·      forcing.py25.4 KB2026-05-30 12:51
  • ·      gr4j.py5.7 KB2026-05-30 12:51
  • ·      graph.py18.5 KB2026-05-30 12:51
  • ·      hierarchy.py9.6 KB2026-06-04 14:18
  • ·      import_fortran_muskingum.py4.2 KB2026-06-04 14:13
  • ·      losses.py1.8 KB2026-05-30 12:51
  • ·      model.py7.5 KB2026-05-30 12:51
  • ·      obs.py3.5 KB2026-05-30 12:51
  • ·      params.py28.4 KB2026-06-04 14:12
  • ·      remap.py4.0 KB2026-05-30 12:51
  • ·      routing.py5.9 KB2026-05-30 12:51
  • ·      routing_numba.py27.7 KB2026-05-30 12:51
  • ·      run_national_mean_runoff.py24.3 KB2026-06-04 14:15
  • ·      scale.py18.9 KB2026-05-30 12:51
  • ·      snow.py4.4 KB2026-05-30 12:51
  • ·      training.py19.8 KB2026-06-04 01:46
  • ·      training_index.py6.0 KB2026-05-30 12:51
  • ·      train_joint.py70.4 KB2026-06-04 15:48
  • ·      train_joint_ddp_windows.py70.1 KB2026-06-04 15:48
  • ·      validate_checkpoint.py10.5 KB2026-06-04 14:13

Preprint预印本

Dong, N., et al. (2026). Enhancing National High-Resolution Distributed Hydrological Modeling with Physics-Encoded Differentiable Regionalization. ESS Open Archive preprint.

For questions, contact: Ningpeng Dong (ndong1993@163.com).