用不确定性引导生成新上下文,提升分割模型对罕见目标的识别能力。
Preserve the Hard, Regenerate the Rest: Uncertainty-Guided Synthetic Training Data Augmentation with Diffusion Models

- 只对不确定区域的背景进行可控生成,保留标签准确性
- 在Cityscapes等数据集上显著提升小物体和稀有类别的分割精度
- 无需额外模型,适合复杂场景下的遥感与自动驾驶数据增强
语义分割模型在数据稀疏或视觉多样的区域(如航拍或自动驾驶数据中的密集区域、小物体)表现不佳。虽然合成数据增强是可行方案,但直接生成带标签的新数据易导致标签与像素错位。现有方法常依赖外部模型或粗略规则(如全图增广),浪费计算资源于无信息像素。本文提出一种基于不确定性的合成上下文增强策略:利用基础分割器的预测熵识别不确定语义区域,仅对互补视觉上下文进行修复生成;微调时仅在原始像素上计算损失,排除生成区域。该方法聚焦于未修改的不确定区域,并赋予其新上下文。在Cityscapes、UAVID和BDD100K数据集上均取得显著的mIoU提升,尤其在公交车、火车等罕见类别上效果突出。结果表明,该方法是一种高效提升复杂数据集分割性能的关键技术,代码已开源。
原文摘要 · Abstract (English)
Semantic segmentation models struggle with data sparsity and rare or visually diverse regions, e.g., dense regions or small objects in aerial or autonomous mobility data. While synthetic augmentation is an appealing solution, directly generating new labeled data risks misalignment of labels and generated pixels. Existing solutions to this problem often rely on external models, or employ coarse heuristics such as indiscriminately augmenting all foreground objects or entire backgrounds, which wastes capacity on uninformative pixels. To address this, we propose an uncertainty-guided synthetic context augmentation strategy that strictly preserves label validity and efficiently maximizes pixel informativeness per synthetic sample - no external guardrails required. Using a baseline segmenter's predictive entropy, we identify uncertain semantic regions and inpaint only the complementary visual context. When fine-tuning the segmenter on this synthetic data, we compute the loss only over the original pixels, excluding inpainted regions. This focuses learning on the unmodified, uncertain regions while presenting them in novel contexts. We demonstrate substantial mIoU gains on Cityscapes, UAVID, and BDD100K with the largest gains on rare and difficult classes such as buses, trains, or (from the aerial perspective) cars. Our results demonstrate that uncertainty-guided context augmentation is a highly effective lever to improve segmentation performance on complex datasets, with code provided at https://github.com/XITASO/Preserve-the-Hard-Regenerate-the-Rest.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。