Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces a discrete diffusion contour refinement pipeline for boundary detection, specifically designed for low-data regimes common in medical imaging and environmental monitoring. The method employs a CNN with self-attention to iteratively denoise sparse contour representations conditioned on segmentation masks. By simplifying the diffusion process, customizing the model architecture, and minimizing post-processing, the approach achieves state-of-the-art or competitive performance on KVASIR, HAM10K, and a custom Smoke dataset, while also improving inference speed.
Achieve robust boundary detection with limited data by iteratively denoising sparse contours using a lightweight discrete diffusion process, outperforming SOTA methods and boosting inference speed by 3.5x.
Boundary detection of irregular and translucent objects is an important problem with applications in medical imaging, environmental monitoring and manufacturing, where many of these applications are plagued with scarce labeled data and low in situ computational resources. While recent image segmentation studies focus on segmentation mask alignment with ground-truth, the task of boundary detection remains understudied, especially in the low data regime. In this work, we present a lightweight discrete diffusion contour refinement pipeline for robust boundary detection in the low data regime. We use a Convolutional Neural Network(CNN) architecture with self-attention layers as the core of our pipeline, and condition on a segmentation mask, iteratively denoising a sparse contour representation. We introduce multiple novel adaptations for improved low-data efficacy and inference efficiency, including using a simplified diffusion process, a customized model architecture, and minimal post processing to produce a dense, isolated contour given a dataset of size<500 training images. Our method outperforms several SOTA baselines on the medical imaging dataset KVASIR, is competitive on HAM10K and our custom wildfire dataset, Smoke, while improving inference framerate by 3.5X.