Skip to yearly menu bar Skip to main content


Poster
in
Workshop: NeurIPS 2023 Workshop on Diffusion Models

Diffusing More Objects for Semi-Supervised Domain Adaptation with Less Labeling

Leander van den Heuvel · Gertjan Burghouts · David Zhang · Gwenn Englebienne · Sabina van Rooij


Abstract:

For object detection, it is possible to view the prediction of bounding boxes as a reverse diffusion process, where the random bounding boxes are iteratively refined as a denoising step, conditioned on the image using a diffusion model. 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.

Chat is not available.