从真实遮挡中学习物理动态,避免模型死区和生成崩溃。
Observation-Aligned Mask Priors for Learning Physical Dynamics from Authentic Occlusions

- 用贝叶斯流网络学习真实遮挡分布,生成与观测对齐的掩码。
- 在256×256分辨率下,相比扩散基线,MSE降低18.3%,PSNR提升2.1dB。
- 适用于海洋观测等存在真实遮挡的物理系统建模,适合数据不完整场景。
直接从不完整观测中学习物理动态极具挑战性,因为真实遮挡具有结构性、样本依赖性且常非随机缺失。现有方法多依赖启发式掩码规则或预设掩码分布。本文提出观察对齐掩码先验(Observation-Aligned Mask Priors),通过在二值观测掩码上预训练贝叶斯流网络(BFN)以捕捉真实遮挡拓扑,再通过全局归一化交叉熵目标引导采样,生成与稀疏观测对齐的样本特异性掩码。掩码与观测的交集作为上下文,剩余观测点作为扩散重建模型的查询目标。该交集划分确保每个有效观测维度均有正概率被查询,避免零查询死区和局部生成崩溃。在三个含真实卫星遮挡的海洋学数据集上,分辨率最高达256×256,实验显示该方法在均方误差(MSE)和峰值信噪比(PSNR)上持续优于强扩散基线。
原文摘要 · Abstract (English)
Learning physical dynamics directly from incomplete observations is challenging because authentic occlusions are structured, sample-dependent, and often missing not at random, whereas existing methods typically rely on heuristic masking rules or predefined mask distributions. We propose Observation-Aligned Mask Priors, a framework that learns the distribution of authentic observation masks and uses it to construct context-query partitions for training from incomplete data. Specifically, we pretrain a Bayesian Flow Network (BFN) on binary observation masks to capture real occlusion topologies, then guide BFN sampling with a globally normalized cross-entropy objective to generate sample-specific masks aligned with each sparse observation. The intersection between the guided mask and the observed mask defines the context, and the remaining observed entries become query targets for a diffusion-based reconstruction model. We show that this intersection-based partitioning gives every valid observed dimension a strictly positive probability of being queried, preventing zero-query dead zones and local generative collapse. Experiments on three real-world oceanographic datasets with authentic satellite occlusions, across resolutions up to 256$\times$256, show consistent improvements over strong diffusion baselines in MSE and PSNR. These results demonstrate that learning mask priors from authentic occlusions is an effective alternative to heuristic masking for learning from incomplete physical observations without access to fully observed fields.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。