Deep Set Prediction Networks
Authors: Yan Zhang, Jonathon Hare, Adam Prugel-Bennett
NeurIPS 2019 | Conference PDF | Archive PDF | Plain Text | LLM Run Details
| Reproducibility Variable | Result | LLM Response |
|---|---|---|
| Research Type | Experimental | We evaluate our model on several set prediction datasets (section 5). First, we demonstrate that the auto-encoder version of our model is sound on a set version of MNIST. Next, we use the CLEVR dataset to show that this works for general set prediction tasks. We predict the set of bounding boxes of objects in an image and we predict the set of object attributes in an image, both from a single feature vector. |
| Researcher Affiliation | Academia | Yan Zhang University of Southampton Southampton, UK yz5n12@ecs.soton.ac.uk Jonathon Hare University of Southampton Southampton, UK jsh2@ecs.soton.ac.uk Adam Prügel-Bennett University of Southampton Southampton, UK apb@ecs.soton.ac.uk |
| Pseudocode | Yes | Algorithm 1 One forward pass of the set prediction algorithm within the training loop. |
| Open Source Code | Yes | We provide the PyTorch [18] source code to reproduce all experiments at https://github.com/Cyanogenoid/dspn. |
| Open Datasets | Yes | We begin with the task of auto-encoding a set version of MNIST. Next, we turn to the task of object detection on the CLEVR dataset [11] |
| Dataset Splits | Yes | Next, we turn to the task of object detection on the CLEVR dataset [11], which contains 70,000 training and 15,000 validation images. |
| Hardware Specification | No | No specific hardware details (e.g., GPU models, CPU types, memory) are provided for the experimental setup. |
| Software Dependencies | No | The paper mentions 'PyTorch [18]' but does not specify a version number for it or any other software dependencies. |
| Experiment Setup | Yes | In all experiments, we fix the hyperparameters of our model to T = 10, η = 800, λ = 0.1. Further details about the model architectures, training settings, and hyperparameters are given in Appendix B. |