观潮 · 模式预报与气象 AI 观察
城市内涝 AI 预报模型要多少样本:随机森林 50 条就收敛,BP 神经网络要 120 条
> 城市内涝 AI 预报的样本是拿水文水动力模型一条一条跑出来的,样本量直接换算成机时。这篇把「到底要多少条」当成一个可量化的问题:对 Ridge 回归、K 近邻、随机森林和 BP 神经网络四种算法,识别各自性能曲线的突变点、饱和点与最优样本量。判据是 ENS≥0.85 且 ERMS≤0.02。四种模型的最优样本量分别是 100、90、50、120 条,占总样本的 59.5%、53.6%、29.8%、71.4%;对应的数据集构建耗时比用全样本减少 5.00、5.71、8.61、3.39 小时。随机森林的样本优化潜力最大——只用不到三成样本就收敛。作者同时给出一个反直觉的现象:性能曲线不是单调上升的,中途会出现回落,原因是样本库涵盖 7 类差异显著的雨型,固定步长加样本时会随机把尚未学透的复杂工况带进来。
缺的是哪一块
物理数值模型能算准水深和流速,但网格一细、网格数一涨,物理方程求解就变重;城市下垫面还要在每格上叠加河网、管网等多个过程。GPU 并行加速缓解了速率,按文中的判断仍满足不了洪涝实时预警的时效性。
机器学习直接建降雨到内涝的非线性关系,绕开方程求解,效率提升明显,但训练样本依然要靠物理模型生成。样本少了过拟合、预测不稳定;样本多了则训练时间和数据生成成本往上走。已有研究大多在提升精度,对不同模型对样本规模的依赖程度和最优样本量的讨论相对有限。
研究区与样本库怎么搭
项 · 设置
研究区 · 广东省佛山市顺德区陈村镇南顺联安围,总面积 48.51 平方公里,西北略高、东南略低,河网密布,属典型水网平原
降水背景 · 多年平均降水量 1517.1 毫米,集中在 4—9 月(汛期),约占全年 81%
地形与下垫面 · DEM 水平分辨率 10 米 × 10 米,由 1355 × 718 个网格构成;最大似然法划分为道路、建筑用地、绿地、塘、水面及其他 6 类土地利用
管网 · 概化为 6815 个雨水节点、8017 个污水节点、16006 条管线
水动力求解 · 单元界面处采用 HLLC 近似黎曼求解器计算质量通量与动量通量,处理急变流与不连续
雨型设计 · 样本库涵盖 7 类差异显著的降雨雨型
流程 · 先用水文水动力模型模拟构建样本库,再分层抽样生成不同规模的总样本集,训练并评估四种模型,最后定量识别性能曲线的特征点
对比的四种算法是 Ridge 回归、K 近邻(KNN)、随机森林(RF)和 BP 神经网络,覆盖从线性到集成树到浅层神经网络的复杂度梯度。
结果数字
模型 · 最优样本量 · 占总样本比例 · 数据集构建耗时减少
Ridge 回归 · 100 · 59.5% · 5.00 小时
K 近邻(KNN) · 90 · 53.6% · 5.71 小时
随机森林(RF) · 50 · 29.8% · 8.61 小时
BP 神经网络 · 120 · 71.4% · 3.39 小时
精度门槛统一为 ENS≥0.85 且 ERMS≤0.02。随机森林只用 29.8% 的样本就达到门槛,且省下的时间最多(8.61 小时),是四种里样本优化潜力最大的。
BP 神经网络需要 71.4% 的样本,文中给了解释:模型寻优确定的(150,100)双隐藏层结构,在构建从 7 维静态输入到内涝峰值时刻 5 个易涝点积水深度的映射时,产生了极其庞大的自由权重参数。
一个反直觉的现象:曲线不是单调上升
按固定步长递增样本量时,性能曲线中途会出现回落。作者把它归因到样本库的设计上:7 类雨型差异显著,新增批次会随机引入极端或尚未被充分学习的复杂降雨—内涝工况。
机理层面的解释是:城市内涝本质上受管网排水能力、地表积水等多种物理阈值控制,是强非线性过程;对尚未达到全局收敛的模型,这些突增的复杂特征子集会暂时打破它已经初步建立的决策边界。
随机森林是个例外。单棵决策树在样本不足时确实可能对特定的复杂降雨—内涝响应过拟合,但多树集成平均的机制让它能在较小样本下就实现高鲁棒性——这是它 50 条就够的直接原因。
原文图

图1 作为训练标签来源的水文水动力模型验证:一场降雨过程中模拟水深过程线(黑线)与实测点(红点)吻合,峰值约 0.43 米。

图2 AI 快速预报水深与水文水动力模型的对照:五个点位(彩色圆点)基本落在 20% 误差带内,星标为实测降雨情景,1:1 线为黑实线。
作者承认的限制
限制 · 原文口径
GPU 加速仍不够 · 作者判断,即使近年 GPU 并行加速提升了传统物理模型速率,仍无法满足洪涝灾害实时预警的时效性需求
结论绑定本文样本库 · 曲线回落现象的解释依赖「样本库涵盖 7 类差异显著雨型」这一具体设计,换一套雨型设计是否重现未知
精度门槛是人为设定的 · ENS≥0.85 与 ERMS≤0.02 是本研究的判据,最优样本量的数值随门槛而变
主动学习未做 · 作者把「引入主动学习等前沿算法、发展自适应样本筛选与增量学习策略」列为未来研究
一句话判断
对城市内涝快速预报这种「样本靠物理模型跑」的场景,样本量不是越多越好:随机森林用不到三成样本就能达到精度门槛,省下 8.61 小时机时;BP 神经网络要七成以上。选型时把样本生成成本算进去,结论可能和单纯比精度不一样。