Set transformer: A framework for attention-based permutation-invariant neural networks J Lee, Y Lee, J Kim, A Kosiorek, S Choi, YW Teh International conference on machine learning, 3744-3753, 2019 | 1232 | 2019 |
Stacked Capsule Autoencoders AR Kosiorek, S Sabour, YW Teh, GE Hinton Advances in Neural Information Processing Systems, 2019 | 324 | 2019 |
Genesis: Generative scene inference and sampling with object-centric latent representations M Engelcke, AR Kosiorek, OP Jones, I Posner arXiv preprint arXiv:1907.13052, 2019 | 278 | 2019 |
Sequential attend, infer, repeat: Generative modelling of moving objects A Kosiorek, H Kim, YW Teh, I Posner Advances in Neural Information Processing Systems 31, 2018 | 264 | 2018 |
Tighter variational bounds are not necessarily better T Rainforth, AR Kosiorek, TA Le, CJ Maddison, M Igl, F Wood, YW Teh International Conference on Machine Learning, 2018 | 222 | 2018 |
Nerf-vae: A geometry aware 3d scene generative model AR Kosiorek, H Strathmann, D Zoran, P Moreno, R Schneider, S Mokrá, ... International Conference on Machine Learning, 5742-5752, 2021 | 131 | 2021 |
Decomposing 3d scenes into objects via unsupervised volume segmentation K Stelzner, K Kersting, AR Kosiorek arXiv preprint arXiv:2104.01148, 2021 | 93 | 2021 |
Hierarchical attentive recurrent tracking A Kosiorek, A Bewley, I Posner Advances in neural information processing systems 30, 2017 | 81 | 2017 |
Adversarial masking for self-supervised learning Y Shi, N Siddharth, P Torr, AR Kosiorek International Conference on Machine Learning, 20026-20040, 2022 | 75 | 2022 |
Revisiting reweighted wake-sleep for models with stochastic control flow TA Le, AR Kosiorek, N Siddharth, YW Teh, F Wood Uncertainty in Artificial Intelligence, 1039-1049, 2020 | 74* | 2020 |
Metafun: Meta-learning with iterative functional updates J Xu, JF Ton, H Kim, A Kosiorek, YW Teh International Conference on Machine Learning, 10617-10627, 2020 | 73 | 2020 |
Conditional set generation with transformers AR Kosiorek, H Kim, DJ Rezende arXiv preprint arXiv:2006.16841, 2020 | 42 | 2020 |
Set transformer J Lee, Y Lee, J Kim, AR Kosiorek, S Choi, YW Teh International Conference on Machine Learning 4 (8), 2019 | 28 | 2019 |
End-to-end recurrent multi-object tracking and trajectory prediction with relational reasoning FB Fuchs, AR Kosiorek, L Sun, OP Jones, I Posner arXiv preprint arXiv:1907.12887, 2019 | 15 | 2019 |
Generative adversarial set transformers K Stelzner, K Kersting, AR Kosiorek Workshop on Object-Oriented Learning at ICML 3, 1, 2020 | 14 | 2020 |
Guiding Physical Intuition with Neural Stethoscopes F Fuchs, O Groth, A Kosiorek, A Bewley, M Wulfmeier, A Vedaldi, I Posner | 13* | |
Bayesian delay embeddings for dynamical systems N Dhir, AR Kosiorek, I Posner NIPS Timeseries Workshop, 2017 | 6 | 2017 |
Laser: Latent set representations for 3d generative modeling P Moreno, AR Kosiorek, H Strathmann, D Zoran, RG Schneider, ... arXiv preprint arXiv:2301.05747, 2023 | 3 | 2023 |
Rendering new images of scenes using geometry-aware neural networks conditioned on latent variables AR Kosiorek, H Strathmann, DJ Rezende, D Zoran, PM Comellas US Patent App. 18/275,332, 2024 | 1 | 2024 |
Object discovery in images through categorizing object parts AR Kosiorek, GE Hinton, SSR Aghdam, YW Teh US Patent App. 17/613,767, 2022 | | 2022 |