同时提升模型对新类别和新场景的适应能力。
Generalize or Detect? Towards Robust Semantic Segmentation Under Multiple Distribution Shifts
- 设计生成增强方法,模拟图像与物体级分布偏移。
- 在多个基准上实现最佳的异常检测与域泛化性能。
- 适合需要鲁棒分割的开放世界应用开发者。
在开放世界场景中,模型需同时应对未知类别和新领域。现有方法常混淆领域级与语义级分布偏移,导致泛化或异常检测效果差。本文提出一种新型生成增强方法,生成融合异常物体与多层级协变量偏移的连贯图像;设计针对性训练策略,重校准语义偏移下的不确定性,并优化特征提取器以对齐领域偏移特征。在包含语义与领域偏移的多个基准上验证,本方法在所有评测中均达到当前最优的异常检测与域泛化性能。代码已公开于 https://github.com/gaozhitong/MultiShiftSeg。
原文摘要 · Abstract (English)
In open-world scenarios, where both novel classes and domains may exist, an ideal segmentation model should detect anomaly classes for safety and generalize to new domains. However, existing methods often struggle to distinguish between domain-level and semantic-level distribution shifts, leading to poor out-of-distribution (OOD) detection or domain generalization performance. In this work, we aim to equip the model to generalize effectively to covariate-shift regions while precisely identifying semantic-shift regions. To achieve this, we design a novel generative augmentation method to produce coherent images that incorporate both anomaly (or novel) objects and various covariate shifts at both image and object levels. Furthermore, we introduce a training strategy that recalibrates uncertainty specifically for semantic shifts and enhances the feature extractor to align features associated with domain shifts. We validate the effectiveness of our method across benchmarks featuring both semantic and domain shifts. Our method achieves state-of-the-art performance across all benchmarks for both OOD detection and domain generalization. Code is available at https://github.com/gaozhitong/MultiShiftSeg.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。