A new few-shot learning model for runoff prediction: Demonstration in two data scarce regions

被引:16
|
作者
Yang, Minghong [1 ]
Yang, Qinli [1 ,2 ]
Shao, Junming [2 ,3 ]
Wang, Guoqing [4 ]
Zhang, Wei [5 ]
机构
[1] Univ Elect Sci & Technol China, Sch Resources & Environm, 2006 Xiyuan Ave, Chengdu 611731, Peoples R China
[2] Univ Elect Sci & Technol China, Yangtze Delta Reg Inst Huzhou, Huzhou 313001, Peoples R China
[3] Univ Elect Sci & Technol China, Sch Comp Sci & Engn, 2006 Xiyuan Ave, Chengdu 611731, Peoples R China
[4] Nanjing Hydraul Res Inst, State Key Lab Hydrol Water Resources & Hydraul Eng, Nanjing 210029, Peoples R China
[5] Sci & Technol Elect Informat Control Lab, Chengdu, Peoples R China
基金
中国国家自然科学基金;
关键词
Runoff prediction; Few -shot learning; LSTM; Prototypical network; Sparsely -gauged basins; UNGAUGED BASINS; RAINFALL; FLOOD; INDEX;
D O I
10.1016/j.envsoft.2023.105659
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Most existing hydrologic models and machine learning models failed to perform well on runoff prediction in data scarce regions. As an alternative to this, the Long Short-Term Memory (LSTM)-prototypical network fusion model based on few-shot learning is proposed, where the strong learning ability of LSTM and the low data dependence of prototypical network are combined. The proposed model was calibrated and implemented on monthly runoff prediction in the Lancang River basin (LRB) and the source region of the Yellow River basin (SRYRB). Compared with eight state-of-the-art data driven models (LSTM, SVR, ANN, ARMA, Random Forest, SimpleRNN, GRU, and BiLSTM), the proposed model outperformed especially when less training data were used. Results in the LRB indicate NSE of the proposed model achieved 0.802 and 0.832 when the proportion of training data (K) was 20% and 45%, improved by 0.527 and 0.222 relative to the mean NSE of other models, respectively. In the SRYRB, NSE reached 0.830 and improved by 0.354 when K was 40%. The findings imply that the new few-shot learning model provides a promising tool for runoff prediction in the two investigated basins and possibly other data -scarce basins where precipitation dominates runoff change, which will benefit regional water resources man-agement and water security.
引用
收藏
页数:12
相关论文
共 50 条
  • [31] Few-Shot Learning for Low-Data Drug Discovery
    Vella, Daniel
    Ebejer, Jean-Paul
    JOURNAL OF CHEMICAL INFORMATION AND MODELING, 2022, 63 (01) : 27 - 42
  • [32] Combat data shift in few-shot learning with knowledge graph
    Zhu, Yongchun
    Zhuang, Fuzhen
    Zhang, Xiangliang
    Qi, Zhiyuan
    Shi, Zhiping
    Cao, Juan
    He, Qing
    FRONTIERS OF COMPUTER SCIENCE, 2023, 17 (01)
  • [33] Explaining Siamese Networks in Few-Shot Learning for Audio Data
    Fedele, Andrea
    Guidotti, Riccardo
    Pedreschi, Dino
    DISCOVERY SCIENCE (DS 2022), 2022, 13601 : 509 - 524
  • [34] Few-shot learning on batch process modeling with imbalanced data
    Gu, Shaowu
    Chen, Junghui
    Xie, Lei
    CHEMICAL ENGINEERING SCIENCE, 2024, 285
  • [35] Combat data shift in few-shot learning with knowledge graph
    Yongchun Zhu
    Fuzhen Zhuang
    Xiangliang Zhang
    Zhiyuan Qi
    Zhiping Shi
    Juan Cao
    Qing He
    Frontiers of Computer Science, 2023, 17
  • [36] Few-Shot Learning With Enhancements to Data Augmentation and Feature Extraction
    Zhang, Yourun
    Gong, Maoguo
    Li, Jianzhao
    Feng, Kaiyuan
    Zhang, Mingyang
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, : 1 - 14
  • [37] Robust Few-Shot Learning for User-Provided Data
    Lu, Jiang
    Jin, Sheng
    Liang, Jian
    Zhang, Changshui
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2021, 32 (04) : 1433 - 1447
  • [38] FlipDA: Effective and Robust Data Augmentation for Few-Shot Learning
    Zhou, Jing
    Zheng, Yanan
    Tang, Jie
    Li, Jian
    Yang, Zhilin
    PROCEEDINGS OF THE 60TH ANNUAL MEETING OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS (ACL 2022), VOL 1: (LONG PAPERS), 2022, : 8646 - 8665
  • [39] Few-Shot Learning Method for Continuous Prediction of Rock Mechanical Parameters Based on Logging Data
    Zhao, Weiguang
    Sang, Shuxun
    Han, Sijie
    Cheng, Deqiang
    Zhou, Xiaozhi
    Zhang, Jinchao
    Zhao, Fuping
    ACS OMEGA, 2024, 9 (47): : 47234 - 47247
  • [40] Mask Mixup Model: Enhanced Contrastive Learning for Few-Shot Learning
    Xie, Kai
    Gao, Yuxuan
    Chen, Yadang
    Che, Xun
    APPLIED SCIENCES-BASEL, 2024, 14 (14):