English

Deep Set Prediction Networks

Machine Learning 2020-04-28 v6 Machine Learning

Abstract

Current approaches for predicting sets from feature vectors ignore the unordered nature of sets and suffer from discontinuity issues as a result. We propose a general model for predicting sets that properly respects the structure of sets and avoids this problem. With a single feature vector as input, we show that our model is able to auto-encode point sets, predict the set of bounding boxes of objects in an image, and predict the set of attributes of these objects.

Keywords

Cite

@article{arxiv.1906.06565,
  title  = {Deep Set Prediction Networks},
  author = {Yan Zhang and Jonathon Hare and Adam Prügel-Bennett},
  journal= {arXiv preprint arXiv:1906.06565},
  year   = {2020}
}

Comments

Appendix C contains an erratum

R2 v1 2026-06-23T09:54:36.377Z