用可控编辑生成挑战样本,测试分割模型在外观与几何变化下的鲁棒性。
Benchmarking Semantic Segmentation Models via Appearance and Geometry Attribute Editing

- 基于扩散模型精准编辑真实图像的外观和几何属性,保留原标签。
- 发现开放词汇模型在几何变化下不如封闭集模型鲁棒。
- 可作为数据增强工具提升模型对分布内外样本的性能。
语义分割在自动驾驶、医学图像分析等场景中至关重要。部署前需预先测试模型在复杂多变场景中的表现。本文构建自动数据生成流水线 Gen4Seg,通过引入对象级与图像级的外观和几何属性变化(如物体颜色、材质、尺寸、位置及天气、风格)来压力测试分割模型。利用扩散模型实现对真实图像的可控属性编辑,同时保持结构信息完整,使原有分割标签可复用,显著降低标注成本。基于此,我们构建了两个新基准 Pascal-EA 与 COCO-EA,评估了从封闭集到开放词汇的大规模分割模型。关键发现包括:1)先进开放词汇模型在几何变化下并未表现出比封闭集方法更强的鲁棒性;2)CutOut、CutMix 等数据增强技术对提升外观变化鲁棒性作用有限;3)该管道亦可作为数据增强手段,有效提升模型在分布内与分布外样本上的性能。本工作展示了生成模型在自动分析分割模型方面的潜力,为构建更鲁棒可靠的分割系统提供参考。
原文摘要 · Abstract (English)
Semantic segmentation takes pivotal roles in various applications such as autonomous driving and medical image analysis. When deploying segmentation models in practice, it is critical to test their behaviors in varied and complex scenes in advance. In this paper, we construct an automatic data generation pipeline Gen4Seg to stress-test semantic segmentation models by generating various challenging samples with different attribute changes. Beyond previous evaluation paradigms focusing solely on global weather and style transfer, we investigate variations in both appearance and geometry attributes at the object and image level. These include object color, material, size, position, as well as image-level variations such as weather and style. To achieve this, we propose to edit visual attributes of existing real images with precise control of structural information, empowered by diffusion models. In this way, the existing segmentation labels can be reused for the edited images, which greatly reduces the labor costs. Using our pipeline, we construct two new benchmarks, Pascal-EA and COCO-EA. We benchmark a wide variety of semantic segmentation models, spanning from closed-set models to open-vocabulary large models. We have several key findings: 1) advanced open-vocabulary models do not exhibit greater robustness compared to closed-set methods under geometric variations; 2) data augmentation techniques, such as CutOut and CutMix, are limited in enhancing robustness against appearance variations; 3) our pipeline can also be employed as a data augmentation tool and improve both in-distribution and out-of-distribution performances. Our work suggests the potential of generative models as effective tools for automatically analyzing segmentation models, and we hope our findings will assist practitioners and researchers in developing more robust and reliable segmentation models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。