mlsanityData to 14 Oct 2026

Revitalizing Dense Material Segmentation: Stabilized Vision Transformers and the Generalization Paradox

arXiv 2605.23747 ↗First posted 22 May 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Material segmentation, the pixel-wise classification of physical surface properties, remains a challenging problem in computer vision, requiring physicochemical understanding distinct from object-centric parsing. Despite the introduction of the rigorous Apple Dense Material Segmentation (DMS) dataset, the benchmark has suffered from attrition and stagnation, increasingly overshadowed by geometry-biased foundation models. In this paper, we revive the Apple-DMS benchmark to establish a modern Vision Transformer baseline. We conduct an exhaustive evaluation of SegFormer and Mask2Former architectures, revealing that standard training paradigms fail on amorphous texture fields due to high-variance gradients. To address this, we introduce a stabilized training recipe featuring High-Fidelity Logit Projection, Query Entropy Regularization, and a domain-specific, physics-compliant augmentation pipeline. Our optimized SegFormer-B5 achieves a new State-of-the-Art (SOTA) of 0.4572 mIoU on the original dataset split, significantly surpassing the prior convolutional baseline. Furthermore, we identify a critical "Generalization Paradox": while re-partitioning the dataset into a data-rich 80/10/10 split inflates the metric to 0.5276 mIoU, expert qualitative analysis reveals this induces distributional homogenization, severely degrading real-world, out-of-distribution performance. By releasing our recovered dataset index and robust training framework, we demonstrate that material perception is far from solved and urge the community to leverage the rigorous original split to drive genuine progress in physically grounded artificial intelligence.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 12 of its 18 references that are in the map. First posted on arXiv 22 May 2026.

Builds on

12 of 18 references.

References · 12

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  2. 2023Segment Anythingcited by 8,234
  3. 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,859
  4. 2021SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformerscited by 2,940
  5. 2024SAM 2: Segment Anything in Images and Videoscited by 2,863
  6. 2021Masked-attention Mask Transformer for Universal Image Segmentationcited by 2,351
  7. 2013Describing Textures in the Wildcited by 2,125
  8. 2025SAM 3: Segment Anything with Conceptscited by 906
  9. 2016COCO-Stuff: Thing and Stuff Classes in Contextcited by 864
  10. 2023Segment Anything Is Not Always Perfect: An Investigation of SAM on Different Real-world Applicationscited by 121
  11. 2014Material Recognition in the Wild with the Materials in Context Databasecited by 79
  12. 2022A Dense Material Segmentation Dataset for Indoor and Outdoor Scene Parsingcited by 23

Built on by

Not cited yet.