TRACE: Your Diffusion Model is Secretly an Instance Edge Detector
On the map: Diffusion models → Medical imaging and segmentation
Abstract (arXiv)
High-quality instance and panoptic segmentation has traditionally relied on dense instance-level annotations such as masks, boxes, or points, which are costly, inconsistent, and difficult to scale. Unsupervised and weakly-supervised approaches reduce this burden but remain constrained by semantic backbone constraints and human bias, often producing merged or fragmented outputs. We present TRACE (TRAnsforming diffusion Cues to instance Edges), showing that text-to-image diffusion models secretly function as instance edge annotators. TRACE identifies the Instance Emergence Point (IEP) where object boundaries first appear in self-attention maps, extracts boundaries through Attention Boundary Divergence (ABDiv), and distills them into a lightweight one-step edge decoder. This design removes the need for per-image diffusion inversion, achieving 81x faster inference while producing sharper and more connected boundaries. On the COCO benchmark, TRACE improves unsupervised instance segmentation by +5.1 AP, and in tag-supervised panoptic segmentation it outperforms point-supervised baselines by +1.7 PQ without using any instance-level labels. These results reveal that diffusion models encode hidden instance boundary priors, and that decoding these signals offers a practical and scalable alternative to costly manual annotation. Project Page: https://shjo-april.github.io/TRACE/
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in Medical imaging and segmentation, one of 14 directions in Diffusion models. Builds on 81 of its 112 references that are in the map. First posted on arXiv 11 Mar 2025.
Builds on
81 of 112 references.
Closest ancestors · 4
Same direction, newest first
- 2024EmerDiff: Emerging Pixel-level Semantic Knowledge in Diffusion Models22 Jan
- 2023MosaicFusion: Diffusion Models as Data Augmenters for Large Vocabulary Instance Segmentation22 Sep
- 2023Diffuse, Attend, and Segment: Unsupervised Zero-Shot Segmentation using Stable Diffusion23 Aug
- 2023Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Models8 Mar
Other references · 77
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2023Segment Anythingcited by 8,234
- 2023Visual Instruction Tuningcited by 7,869
- 2017Mask R-CNNcited by 7,277
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2025Qwen2.5-VL Technical Reportcited by 5,587
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,444
38 more
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,356
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
- 2023Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detectioncited by 3,332
- 2015Learning Deep Features for Discriminative Localizationcited by 2,436
- 2021Masked-attention Mask Transformer for Universal Image Segmentationcited by 2,351
- 2024Depth Anything: Unleashing the Power of Large-Scale Unlabeled Datacited by 1,360
- 2018Unified Perceptual Parsing for Scene Understandingcited by 1,271
- 2019LVIS: A Dataset for Large Vocabulary Instance Segmentationcited by 1,217
- 2021Per-Pixel Classification is Not All You Need for Semantic Segmentationcited by 1,152
- 2016Semantic Understanding of Scenes through the ADE20K Datasetcited by 1,069
- 2016COCO-Stuff: Thing and Stuff Classes in Contextcited by 864
- 2023PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesiscited by 864
- 2019Panoptic Feature Pyramid Networkscited by 645
- 2015Holistically-Nested Edge Detectioncited by 592
- 2021Extract Free Dense Labels from CLIPcited by 583
- 2020Dense Contrastive Learning for Self-Supervised Visual Pre-Trainingcited by 512
- 2023Ferret: Refer and Ground Anything Anywhere at Any Granularitycited by 509
- 2022All are Worth Words: A ViT Backbone for Diffusion Modelscited by 502
- 2023EVA-02: A Visual Representation for Neon Genesiscited by 381
- 2023Repurposing Diffusion-Based Image Generators for Monocular Depth Estimationcited by 327
- 2018Learning Pixel-level Semantic Affinity with Image-level Supervision for Weakly Supervised Semantic Segmentationcited by 321
- 2020clDice -- A Novel Topology-Preserving Loss Function for Tubular Structure Segmentationcited by 294
- 2019Weakly Supervised Learning of Instance Segmentation with Inter-pixel Relationscited by 247
- 2022Unsupervised Semantic Segmentation by Distilling Feature Correspondencescited by 241
- 2021K-Net: Towards Unified Image Segmentationcited by 212
- 2019iSAID: A Large-scale Dataset for Instance Segmentation in Aerial Imagescited by 197
- 2023Cut and Learn for Unsupervised Object Detection and Instance Segmentationcited by 187
- 2022Multi-class Token Transformer for Weakly Supervised Semantic Segmentationcited by 138
- 2022Learning to Generate Text-grounded Mask for Open-world Semantic Segmentation from Only Image-Text Pairscited by 130
- 2023DatasetDM: Synthesizing Data with Perception Annotations Using Diffusion Modelscited by 123
- 2022CLIP is Also an Efficient Segmenter: A Text-Driven Approach for Weakly Supervised Semantic Segmentationcited by 118
- 2022TokenCut: Segmenting Objects in Images and Videos with Self-supervised Transformer and Normalized Cutcited by 113
- 2023Dataset Diffusion: Diffusion-based Synthetic Dataset Generation for Pixel-Level Semantic Segmentationcited by 112
- 2021Panoptic SegFormer: Delving Deeper into Panoptic Segmentation with Transformerscited by 109
- 2022FreeSOLO: Learning to Segment Objects without Annotationscited by 102
- 2020Fully Convolutional Networks for Panoptic Segmentationcited by 92
- 2018Weakly Supervised Instance Segmentation using Class Peak Responsecited by 75
- 2022Class Re-Activation Maps for Weakly-Supervised Semantic Segmentationcited by 75
50 most cited of 77 All 77 (JSON)
Built on by
Citing papers per year: 2025: 1; 2026: 3.
4 papers.
- 2025ISAC: Training-Free Instance-to-Semantic Attention Control for Multi-Instance Generationcited by 0
- 2026StreamKL: Fast and Memory-Efficient KL Divergence for Boosting Attention Distillationcited by 0
- 2026Diffuse2Seg: Diffusion Models Can Segment Anything Without Supervisioncited by 0
- 2026TAG: Calibrating SAM3 for Universal Image Segmentation without Dense Supervisioncited by 0
- 2026TAG: Calibrating SAM3 for Universal Image Segmentation without Dense Supervision10 Oct 2026
- 2026Diffuse2Seg: Diffusion Models Can Segment Anything Without Supervision6 Sep 2026
- 2026StreamKL: Fast and Memory-Efficient KL Divergence for Boosting Attention Distillation18 Jun 2026
- 2025ISAC: Training-Free Instance-to-Semantic Attention Control for Multi-Instance Generation27 May 2025
Similar, not cited
Close in wording, no citation either way
- 2024InstanceDiffusion: Instance-level Control for Image Generation
- 2022DiffusionInst: Diffusion Model for Instance Segmentation
- 2025Conditional Latent Diffusion Models for Zero-Shot Instance Segmentation
- 2024A Simple Latent Diffusion Approach for Panoptic Segmentation and Mask Inpainting
- 2026Making Training-Free Diffusion Segmentors Scale with the Generative Power
- 2022A Generalist Framework for Panoptic Segmentation of Images and Videos
- 2023Catch Me If You Can Describe Me: Open-Vocabulary Camouflaged Instance Segmentation with Diffusion
- 2024High-Precision Dichotomous Image Segmentation via Probing Diffusion Capacity