Skip to content


Switch branches/tags

Latest commit


Git stats


Failed to load latest commit information.
Latest commit message
Commit time

Deep Set Prediction Networks

Overview of set optimisation on the example of CLEVR bounding box prediction

This is the official implementation of our NeurIPS 2019 paper Deep Set Prediction Networks. We propose a new way of predicting sets with a neural network that doesn't suffer from discontinuity issues. This is done by backpropagating through a set encoder to act as a set decoder. You can take a look at the poster for NeurIPS 2019 or the poster for the NeurIPS 2019 workshop on Sets & Partitions.

To use the decoder, you only need You can see how it is used in with build_net and the Net class. For details on the exact steps to reproduce the experiments, check out the README in the dspn directory. You can download pre-trained models and the predictions thereof from the Resources page.

BibTeX entry

    author        = {Yan Zhang and Jonathon Hare and Adam Pr\"ugel-Bennett},
    title         = {{Deep Set Prediction Networks}},
    booktitle     = {Advances in Neural Information Processing Systems 32},
    year          = {2019},
    eprint        = {1906.06565},
    url           = {},