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 条
  • [1] RUNOFF PREDICTION IN A DATA SCARCE REGION BASED ON FEW-SHOT LEARNING
    Yang, Minghong
    Yang, Qinli
    Shao, Junming
    Wang, Guoqing
    2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 6304 - 6307
  • [2] Enhanced landslide susceptibility mapping in data-scarce regions via unsupervised few-shot learning
    Kong, Linghao
    Feng, Wenkai
    Yi, Xiaoyu
    Xue, Zhenghai
    Bai, Luyao
    GONDWANA RESEARCH, 2025, 138 : 31 - 46
  • [3] Domain Transfer for Surface Defect Detection using Few-Shot Learning on Scarce Data
    Gerschner, Felix
    Paul, Jonas
    Schmid, Lukas
    Barthel, Nico
    Gouromichos, Victor
    Schmid, Florian
    Atzmueller, Martin
    Theissler, Andreas
    2023 IEEE 21ST INTERNATIONAL CONFERENCE ON INDUSTRIAL INFORMATICS, INDIN, 2023,
  • [4] Demonstration-Conditioned Reinforcement Learning for Few-Shot Imitation
    Cachet, Theo
    Perez, Julien
    Dance, Christopher R.
    INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 139, 2021, 139
  • [5] DemoGrasp: Few-Shot Learning for Robotic Grasping with Human Demonstration
    Wang, Pengyuan
    Manhardt, Fabian
    Minciullo, Luca
    Garattoni, Lorenzo
    Meier, Sven
    Navab, Nassir
    Busam, Benjamin
    2021 IEEE/RSJ INTERNATIONAL CONFERENCE ON INTELLIGENT ROBOTS AND SYSTEMS (IROS), 2021, : 5733 - 5740
  • [6] Learning a Few-shot Embedding Model with Contrastive Learning
    Liu, Chen
    Fu, Yanwei
    Xu, Chengming
    Yang, Siqian
    Li, Jilin
    Wang, Chengjie
    Zhang, Li
    THIRTY-FIFTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THIRTY-THIRD CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE AND THE ELEVENTH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2021, 35 : 8635 - 8643
  • [7] Few-Shot Model Agnostic Federated Learning
    Huang, Wenke
    Ye, Mang
    Du, Bo
    Gao, Xiang
    PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2022, 2022, : 7309 - 7316
  • [8] Relative Performance Prediction using Few-Shot Learning
    Dey, Arunavo
    Dhakal, Aakash
    Islam, Tanzima Z.
    Yeom, Jae-Seung
    Patki, Tapasya
    Nichols, Daniel
    Movsesyan, Alexander
    Bhatele, Abhinav
    2024 IEEE 48TH ANNUAL COMPUTERS, SOFTWARE, AND APPLICATIONS CONFERENCE, COMPSAC 2024, 2024, : 1764 - 1769
  • [9] Few-shot Learning for New Environment Adaptation
    Wang, Ouya
    Zhou, Shenglong
    Li, Geoffrey Ye
    IEEE CONFERENCE ON GLOBAL COMMUNICATIONS, GLOBECOM, 2023, : 351 - 356
  • [10] Epilepsy seizure prediction with few-shot learning method
    Nazari, Jamal
    Nasrabadi, Ali Motie
    Menhaj, Mohammad Bagher
    Raiesdana, Somayeh
    BRAIN INFORMATICS, 2022, 9 (01)