arXiv · 2312.12000
Diffusing More Objects for Semi-Supervised Domain Adaptation with Less Labeling
Abstract
For object detection, it is possible to view the prediction of bounding boxes as a reverse diffusion process. Using a diffusion model, the random bounding boxes are iteratively refined in a denoising step, conditioned on the image. We propose a stochastic accumulator function that starts each run with random bounding boxes and combines the slightly different predictions. We empirically verify that this improves detection performance. The improved detections are leveraged on unlabelled images as weighted pseudo-labels for semi-supervised learning. We evaluate the method on a challenging out-of-domain test set. Our method brings significant improvements and is on par with human-selected pseudo-labels, while not requiring any human involvement.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Leander van den Heuvel, Gertjan Burghouts, David W. Zhang, Gwenn Englebienne, Sabina B. van Rooij. 2023-12-19. Diffusing More Objects for Semi-Supervised Domain Adaptation with Less Labeling. https://arxiv.org/abs/2312.12000
Cite the original work for its findings. Save a collection to share your selection of sources.