当前位置: 首页 > 文章 > 基于自注意力机制和改进的K-BiLSTM的水产养殖水体溶解氧含量预测模型 江苏农业学报 2024,40 (3) 490-499
Position: Home > Articles > Prediction model of dissolved oxygen content in aquaculture water based on self-attention mechanism and improved K-BiLSTM Jiangsu Journal of Agricultural Sciences 2024,40 (3) 490-499

基于自注意力机制和改进的K-BiLSTM的水产养殖水体溶解氧含量预测模型

作  者:
冯国富;卢胜涛;陈明;王耀辉
单  位:
;上海海洋大学信息学院
关键词:
水产养殖;溶解氧预测;K-means聚类;双向长短期记忆网络(BiLSTM);自注意力机制
摘  要:
为精确预测水产养殖水体溶解氧含量,本研究提出一种基于自注意力机制(ATTN)和改进的K-means聚类-基于残差和批标准化(BN)的双向长短期记忆网络(BiLSTM)的水产养殖水体溶解氧含量预测模型.首先,根据环境数据的相似性,使用改进的K-means算法将数据划分成若干个类别;然后,在BiLSTM基础上构建残差连接和加入BN完成高层次特征提取,利用BiLSTM的长期记忆能力保存特征信息;最后,引入自注意力机制突出不同时间节点数据特征的重要性,进一步提升模型的性能.试验结果表明,本研究提出的基于自注意力机制和改进的K-BiLSTM模型的平均绝对误差为 0.238、均方根误差为 0.322、平均绝对百分比误差为 0.035,与单一的BP 模型、CNN-LSTM模型、传统的K-means-基于残差和BN的BiLSTM-ATTN等模型相比具有更优的预测性能和泛化能力.
译  名:
Prediction model of dissolved oxygen content in aquaculture water based on self-attention mechanism and improved K-BiLSTM
关键词:
aquaculture%dissolved oxygen prediction%K-means clustering%bidirectional long-term and short-term memory network(BiLSTM)%self-attention mechanism
摘  要:
In order to accurately predict the content of dissolved oxygen(DO)in aquaculture water,a prediction mod-el of dissolved oxygen content in aquaculture water based on self-attention mechanism(ATTN)and improved K-means cluste-ring-bidirectional long-term and short-term memory network(BiLSTM)was proposed.Firstly,according to the similarity of environmental data,the improved K-means algorithm was used to divide environmental data into several categories.Then,based on BiLSTM,residual connection was constructed and batch normalization(BN)was added to complete high-level fea-ture extraction,and the feature information was saved by the long-term memory ability of BiLSTM.Finally,the self-attention mechanism was introduced to highlight the importance of data characteristics at different time nodes,which further improved the performance of the model.The experimental results showed that the mean absolute error(MAE),root mean square error(RMSE)and average absolute percentage error(MAPE)of the hybrid model based on self-attention mechanism and improved K-BiLSTM were 0.238,0.322 and 0.035,re-spectively.Compared with single BP model,CNN-LSTM model and traditional K-means-BiLSTM-ATTN model based on residual and BN,the model constructed in this study had better prediction performance and generalization ability.

相似文章

计量
文章访问数: 44
HTML全文浏览量: 0
PDF下载量: 0

所属期刊

推荐期刊