基于UNet-KAN-SR模型的城市积涝高效智能预报OA
Efficient and intelligent forecasting of urban waterlogging based on UNet-KAN-SR modeling
伴随着全球气候变化,在经历高速城镇化进程之后,城市积涝灾害趋多趋强,使得快速积涝预测成为研究热点问题.相较于传统数值模拟,基于深度学习的人工智能模拟方法能有效提升计算效率,但因GPU计算显存有限而易遭遇模型训练瓶颈.鉴于此,本研究提出了一种基于UNet-KAN-SR模型的城市积涝高效智能预报方法,先利用UNet-KAN模块学习低分辨率积涝分布的时空演变规律,再利用超分辨率SR模块与高分辨率下垫面信息,逐步将低分辨率积涝分布映射为高分辨率积涝分布,从而利用时空解耦在保障时空模拟精度的同时有效降低智能模型训练的计算资源需求.测试降雨情景的验证结果表明,UNet-KAN-SR模型能在3 min内完成未来3 h积涝分布预报,均方根误差为9 cm、命中率达0.86,具备较高的计算精度与效率.相较于已有智能模型常用的卷积层,UNet-KAN-SR模型引入的 KAN层可显著提升模型对积涝时空演变过程的非线性建模能力,使均方根误差降低10%.此外,本研究发现高分辨率下垫面信息能显著提升智能模型对积涝分布的预测能力且不同类型的下垫面信息所能取得的提升幅度相近,由此说明在构建智能模型时,可通过优选输入下垫面特征组合,有效提升模型训练速度并控制建模成本,实现高效智能预报.
With global climate change and accelerating urbanization,urban waterlogging disasters have become increasingly frequent and severe,making rapid waterlogging forecasting a key research focus.Compared with tradi-tional numerical simulation methods,deep-learning-based artificial intelligence(AI)models can significantly improve computational efficiency.However,they often encounter training bottlenecks due to limited GPU memory.To address this,this study proposes an efficient AI urban waterlogging forecasting model named as UNet-KAN-SR.This model first employs the UNet-KAN module to efficiently simulate the spatio-temporal evolution of waterlogging over low-resolution grids,and then leverages the SR(super-resolution)module,along with high-resolution surface infor-mation,to progressively map the low-resolution waterlogging distribution to high-resolution distribution.This spatio-temporal decoupling strategy can ensure simulation accuracy while substantially reducing the computational resources required for training AI models.Experimental results demonstrate that the UNet-KAN-SR model can simulate a 3-hour waterlogging distribution within 3 minutes,achieving a root mean square error(RMSE)of 9 cm and a probabil-ity of detection(POD)of 0.84,demonstrating high accuracy and computational efficiency.Further analysis reveals that the integration of the KAN module can significantly enhance the model's capability to capture nonlinear flood dynamics when compared with common CNN modules,reducing RMSE by 10%.Furthermore,this study finds that incorporating high-resolution features,such as surface topography,building coverage ratio,and land use,can sig-nificantly improve the simulation performance but performance improvement is similar under different feature combi-nations.This indicates that by optimizing the combination of input features during AI model construction,training speed can be enhanced,modeling costs controlled,and efficient intelligent forecasting achieved.
陈耀明;李瑞栋;陈骥;倪广恒
清华大学 水圈科学与水利工程全国重点实验室,北京 100084||香港大学 土木工程系,香港 999077清华大学 水圈科学与水利工程全国重点实验室,北京 100084香港大学 土木工程系,香港 999077清华大学 水圈科学与水利工程全国重点实验室,北京 100084
天文与地球科学
城市内涝预测深度学习时空预测超分辨率模型北京城市副中心
urban waterlogging forecastingdeep learningspatio-temporal predictionsuper-resolution modelBei-jing municipal administrative center
《水利学报》 2026 (5)
704-715,12
水圈科学与水利工程全国重点实验室项目(sklhse-TD-2024-C01)国家重点研发计划课题(2022YFC3090604)
评论