让流模型自己识别异常输入,避免在危险场景下盲目输出
Native Extrapolation Awareness in Flow-Based Conditional Generation
- 通过结构设计使异常输入传输效率低下,实现原生外推检测
- 在合成数据、风格迁移和气象预测中均有效检测外推,且不降低预测精度
- 适合医疗、机器人、气候等高风险领域部署
流匹配(Flow Matching, FM)在建模复杂条件分布方面表现优异,已成为机器人、天气预报等预测任务的领先方法。然而,在安全关键场景中,其存在严重的外推风险:由于平滑性偏差,流模型会对离域输入生成看似合理但错误的输出,导致无声失败。本文提出Diverging Flows,一种新方法,使单一模型能同时完成条件生成与原生外推检测,通过结构化设计强制对离域输入进行低效传输。我们在合成流形、跨域风格迁移和气温预报任务上验证了该方法,结果表明其可在不牺牲预测保真度或推理延迟的前提下,有效检测外推输入。这些成果确立了Diverging Flows作为可信流模型的稳健方案,为医学、机器人、气候科学等领域的可靠部署铺平道路。
原文摘要 · Abstract (English)
The ability of Flow Matching (FM) to model complex conditional distributions has established it as the state-of-the-art for prediction tasks (e.g., robotics, weather forecasting). However, deployment in safety-critical settings is hindered by a critical extrapolation hazard: driven by smoothness biases, flow models yield plausible outputs even for off-manifold conditions, resulting in silent failures indistinguishable from valid predictions. In this work, we introduce Diverging Flows, a novel approach that enables a single model to simultaneously perform conditional generation and native extrapolation detection by structurally enforcing inefficient transport for off-manifold inputs. We evaluate our method on synthetic manifolds, cross-domain style transfer, and weather temperature forecasting, demonstrating that it achieves effective detection of extrapolations without compromising predictive fidelity or inference latency. These results establish Diverging Flows as a robust solution for trustworthy flow models, paving the way for reliable deployment in domains such as medicine, robotics, and climate science.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。