轻量级框架提升弱监督语义分割边界精度与小物体召回率。
Lightweight Transformer Framework for Weakly Supervised Semantic Segmentation
- 引入边界分支、不确定度引导优化和动态多尺度融合,增强弱监督效果。
- 在相同训练种子下,边界F值、小物体召回率和mIoU均优于基线模型。
- 无需复杂后处理,适合集成到现有弱监督分割流程中。
弱监督语义分割(WSSS)需从噪声大、不完整的图像级标签中学习稠密掩码。本文重新审视SegFormer解码器,提出仅通过三个轻量级且协同的改进即可显著提升弱监督性能——不改变MiT主干网络,也无需复杂后处理。所提方法CrispFormer在解码器中加入:(1) 边界分支,使用轻量边缘头和边界感知损失监督细长物体轮廓;(2) 不确定度引导修正模块,预测像素级可变性不确定度,并用于加权损失及控制分割逻辑值的残差修正;(3) 动态多尺度融合层,以空间softmax门控替代静态拼接,可选地受不确定度调制。该模型实现单次前向传播,保持清晰边界,按位置自适应选择尺度,有效抵抗弱标签噪声。集成至标准WSSS流程(初始化、学生模型与EMA重标注),在相同种子下,始终优于基准的SegFormer,在边界F-score、小物体召回率和mIoU上均有提升,同时计算开销极低。该解码器为中心的设计简单易实现,兼容多种SegFormer变体,为图像级监督生成高保真掩码提供了可复现路径。
原文摘要 · Abstract (English)
Weakly supervised semantic segmentation (WSSS) must learn dense masks from noisy, under-specified cues. We revisit the SegFormer decoder and show that three small, synergistic changes make weak supervision markedly more effective-without altering the MiT backbone or relying on heavy post-processing. Our method, CrispFormer, augments the decoder with: (1) a boundary branch that supervises thin object contours using a lightweight edge head and a boundary-aware loss; (2) an uncertainty-guided refiner that predicts per-pixel aleatoric uncertainty and uses it to weight losses and gate a residual correction of the segmentation logits; and (3) a dynamic multi-scale fusion layer that replaces static concatenation with spatial softmax gating over multi-resolution features, optionally modulated by uncertainty. The result is a single-pass model that preserves crisp boundaries, selects appropriate scales per location, and resists label noise from weak cues. Integrated into a standard WSSS pipeline (seed, student, and EMA relabeling), CrispFormer consistently improves boundary F-score, small-object recall, and mIoU over SegFormer baselines trained on the same seeds, while adding minimal compute. Our decoder-centric formulation is simple to implement, broadly compatible with existing SegFormer variants, and offers a reproducible path to higher-fidelity masks from image-level supervision.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。