用压缩感知提升时空动态重建,抗噪抗缺测更强。
CS-SHRED: Enhancing SHRED for Robust Recovery of Spatiotemporal Dynamics
- 将压缩感知融入浅层循环解码器,实现批量前向重建。
- 低信噪比区抑制噪声,高信噪比区保留精细结构。
- 适合气象、流体等稀疏传感场景下的数据恢复。
我们提出CS-SHRED,一种将压缩感知(CS)融入浅层循环解码器(SHRED)的深度学习架构,用于从不完整、压缩或损坏的数据中重建时空动态。方法引入两项创新:首先,通过在SHRED架构中融合CS技术,采用基于批次的前向框架与ℓ₁正则化,可在传感器稀疏、测量噪声大、采集不全等条件下稳健恢复信号;其次,设计自适应损失函数,动态结合均方误差(MSE)与平均绝对误差(MAE),并加入分段信噪比(SNR)正则项,有效抑制低信噪比区域的噪声和异常值,同时保留高信噪比区域的细粒度特征。我们在粘弹性流体、最大比湿场、海表温度分布及旋转湍流等挑战性问题上验证了该方法。相比传统SHRED,CS-SHRED显著提升重建保真度:峰值信噪比(PSNR)与结构相似性(SSIM)更高,归一化误差更低,感知图像质量(LPIPS)更优,对小尺度结构的保留更佳,且对噪声和异常值更具鲁棒性。结果表明,联合训练的CS与SHRED架构(含基于LSTM的时间演化建模与浅层解码网络(SDN)的高维状态空间建模)具有明显优势。基于信噪比引导的自适应损失函数,使CS-SHRED成为环境、气候及科学数据分析中极具潜力的工具。
原文摘要 · Abstract (English)
We present CS-SHRED, a novel deep learning architecture that integrates Compressed Sensing (CS) into a Shallow Recurrent Decoder (SHRED) to reconstruct spatiotemporal dynamics from incomplete, compressed, or corrupted data. Our approach introduces two key innovations. First, by incorporating CS techniques into the SHRED architecture, our method leverages a batch-based forward framework with $\ell_1$ regularization to robustly recover signals even in scenarios with sparse sensor placements, noisy measurements, and incomplete sensor acquisitions. Second, an adaptive loss function dynamically combines Mean Squared Error (MSE) and Mean Absolute Error (MAE) terms with a piecewise Signal-to-Noise Ratio (SNR) regularization, which suppresses noise and outliers in low-SNR regions while preserving fine-scale features in high-SNR regions. We validate CS-SHRED on challenging problems including viscoelastic fluid flows, maximum specific humidity fields, sea surface temperature distributions, and rotating turbulent flows. Compared to the traditional SHRED approach, CS-SHRED achieves significantly higher reconstruction fidelity -- as demonstrated by improved SSIM and PSNR values, lower normalized errors, and enhanced LPIPS scores-thereby providing superior preservation of small-scale structures and increased robustness against noise and outliers. Our results underscore the advantages of the jointly trained CS and SHRED design architecture which includes an LSTM sequence model for characterizing the temporal evolution with a shallow decoder network (SDN) for modeling the high-dimensional state space. The SNR-guided adaptive loss function for the spatiotemporal data recovery establishes CS-SHRED as a promising tool for a wide range of applications in environmental, climatic, and scientific data analyses.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。