无需训练即可精准分割图像中任意实体,解决过切和欠切问题。
E-SAM: Training-Free Segment Every Entity Model

- 通过多层级掩码生成,分层处理SAM输出以保留细节。
- 在基准测试中性能提升30.1点,达到当前最优水平。
- 完全免训练,适合快速部署于开放世界场景。
实体分割(ES)旨在不依赖预定义类别标签的情况下识别并分割图像中的独立实体,适用于动态变化的开放世界应用。现有方法通常需大量标注数据或高昂训练成本,限制了可扩展性与适应性。尽管分割一切模型(SAM)在全自动掩码生成模式下展现出全图分割潜力,但仍存在过切和欠切问题,难以胜任实体分割任务。本文提出E-SAM,一种全新的免训练框架,具备出色的实体分割能力。首先引入多层级掩码生成(MMG),对SAM的自动掩码生成输出进行分层处理,生成可靠的物体级掩码同时保持其他层级的精细细节。随后,实体级掩码精修(EMR)将物体级掩码优化为精确的实体级掩码:通过分离重叠掩码消除冗余,并基于实体一致性评估合并相似掩码。最后,欠切精修(USR)生成高置信度额外掩码,融合到EMR输出中,解决欠切问题,得到最终的实体分割图。三个模块无缝协同,实现最优效果且无额外训练开销。大量实验表明,相比已有方法,E-SAM在基准指标上提升30.1点,达到当前最优水平。
原文摘要 · Abstract (English)
Entity Segmentation (ES) aims at identifying and segmenting distinct entities within an image without the need for predefined class labels. This characteristic makes ES well-suited to open-world applications with adaptation to diverse and dynamically changing environments, where new and previously unseen entities may appear frequently. Existing ES methods either require large annotated datasets or high training costs, limiting their scalability and adaptability. Recently, the Segment Anything Model (SAM), especially in its Automatic Mask Generation (AMG) mode, has shown potential for holistic image segmentation. However, it struggles with over-segmentation and under-segmentation, making it less effective for ES. In this paper, we introduce E-SAM, a novel training-free framework that exhibits exceptional ES capability. Specifically, we first propose Multi-level Mask Generation (MMG) that hierarchically processes SAM's AMG outputs to generate reliable object-level masks while preserving fine details at other levels. Entity-level Mask Refinement (EMR) then refines these object-level masks into accurate entity-level masks. That is, it separates overlapping masks to address the redundancy issues inherent in SAM's outputs and merges similar masks by evaluating entity-level consistency. Lastly, Under-Segmentation Refinement (USR) addresses under-segmentation by generating additional high-confidence masks fused with EMR outputs to produce the final ES map. These three modules are seamlessly optimized to achieve the best ES without additional training overhead. Extensive experiments demonstrate that E-SAM achieves state-of-the-art performance compared to prior ES methods, demonstrating a significant improvement by +30.1 on benchmark metrics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。