用可控生成提升海洋障碍物分割数据多样性与质量
Quality-Driven and Diversity-Aware Sample Expansion for Robust Marine Obstacle Segmentation
- 基于风格库和自适应采样器,动态生成高多样性训练样本
- 在多个海洋障碍物数据集上显著提升分割精度
- 适合需要增强数据多样性的视觉感知研究者
海洋障碍物检测在日光闪烁、雾气和快速变化的波浪等挑战性条件下要求鲁棒的分割能力。这些因素降低图像质量,而海洋数据集稀少且结构重复,限制了训练数据的多样性。尽管掩码条件扩散模型可合成布局对齐样本,但当条件为低熵掩码和提示时,常产生低多样性输出,难以提升鲁棒性。本文提出一种无需重训练扩散模型的质量驱动、多样性感知的样本扩展流程,完全在推理阶段生成训练数据。框架包含两个关键组件:(i) 类感知风格库,构建高熵、语义相关的提示;(ii) 自适应退火采样器,在早期条件中引入扰动,同时采用COD引导的比例控制器调节扰动强度,以提升多样性而不损失布局保真度。在多个海洋障碍物基准上,使用这些受控合成样本增强训练数据,可一致提升多种骨干网络的分割性能,并显著增加罕见及纹理敏感类别的视觉变异。
原文摘要 · Abstract (English)
Marine obstacle detection demands robust segmentation under challenging conditions, such as sun glitter, fog, and rapidly changing wave patterns. These factors degrade image quality, while the scarcity and structural repetition of marine datasets limit the diversity of available training data. Although mask-conditioned diffusion models can synthesize layout-aligned samples, they often produce low-diversity outputs when conditioned on low-entropy masks and prompts, limiting their utility for improving robustness. In this paper, we propose a quality-driven and diversity-aware sample expansion pipeline that generates training data entirely at inference time, without retraining the diffusion model. The framework combines two key components:(i) a class-aware style bank that constructs high-entropy, semantically grounded prompts, and (ii) an adaptive annealing sampler that perturbs early conditioning, while a COD-guided proportional controller regulates this perturbation to boost diversity without compromising layout fidelity. Across marine obstacle benchmarks, augmenting training data with these controlled synthetic samples consistently improves segmentation performance across multiple backbones and increases visual variation in rare and texture-sensitive classes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。