用注意力机制融合多通道数据,提升海浪高度估算精度
SCAWaveNet: A Spatial-Channel Attention-Based Network for Global Significant Wave Height Retrieval
- 设计空间-通道注意力网络,让各通道特征自主交互
- 在ERA5和浮标数据上分别达0.438米和0.432米的均方根误差
- 相比顶尖模型误差降低超3.5%,适合遥感海洋监测研究者
近年来,星载GNSS任务产生了大规模全球数据集,为基于深度学习的显著波高(SWH)反演提供了坚实基础。现有深度学习模型主要使用四通道的CYGNSS数据,但大多采用单通道输入或简单拼接,未充分利用通道间信息交互。为此,提出一种基于空间-通道注意力的新型网络SCAWaveNet用于SWH反演。具体地,将每个通道的延迟多普勒图(DDMs)特征建模为独立注意力头,实现空间与通道信息融合;同时设计轻量级注意力机制,在空间与通道维度分配权重,最终整合多维特征。模型使用四通道CYGNSS数据进行评估。以ERA5为参考时,平均均方根误差(RMSE)为0.438米;以美国国家数据浮标中心(NDBC)浮标数据为参考时,平均RMSE为0.432米。相比当前最优模型,分别降低平均误差至少3.52%和5.68%。代码已开源:https://github.com/Clifx9908/SCAWaveNet。
原文摘要 · Abstract (English)
Recent advancements in spaceborne GNSS missions have produced extensive global datasets, providing a robust basis for deep learning-based significant wave height (SWH) retrieval. While existing deep learning models predominantly utilize CYGNSS data with four-channel information, they often adopt single-channel inputs or simple channel concatenation without leveraging the benefits of cross-channel information interaction during training. To address this limitation, a novel spatial-channel attention-based network, namely SCAWaveNet, is proposed for SWH retrieval. Specifically, features from each channel of the DDMs are modeled as independent attention heads, enabling the fusion of spatial and channel-wise information. For auxiliary parameters, a lightweight attention mechanism is designed to assign weights along the spatial and channel dimensions. The final feature integrates both spatial and channel-level characteristics. Model performance is evaluated using four-channel CYGNSS data. When ERA5 is used as a reference, SCAWaveNet achieves an average RMSE of 0.438 m. When using buoy data from NDBC, the average RMSE reaches 0.432 m. Compared to state-of-the-art models, SCAWaveNet reduces the average RMSE by at least 3.52% on the ERA5 dataset and by 5.68% on the NDBC buoy observations. The code is available at https://github.com/Clifx9908/SCAWaveNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。