Learning to Count Objects in Natural Images for Visual Question Answering. - Yan Zhang 067; Jonathon S. Hare; Adam Prügel-Bennett (ICLR (Poster), 2018)

Learning Representations of Sets through Optimized Permutations. - Yan Zhang 067; Jonathon S. Hare; Adam Prügel-Bennett (ICLR (Poster), 2019)

Deep Set Prediction Networks. - Yan Zhang 067; Jonathon S. Hare; Adam Prügel-Bennett (NeurIPS, 2019)

Learning to represent and predict sets with deep neural networks - Yan Zhang 067 (2019)

FSPool: Learning Set Representations with Featurewise Sort Pooling. - Yan Zhang 067; Jonathon S. Hare; Adam Prügel-Bennett (ICLR, 2020)

Equivariance with Learned Canonicalization Functions. - Sékou-Oumar Kaba; Arnab Kumar Mondal; Yan Zhang 067; Yoshua Bengio; Siamak Ravanbakhsh (ICML, 2023)

Object centric architectures enable efficient causal representation learning. - Amin Mansouri; Jason S. Hartford; Yan Zhang 067; Yoshua Bengio (ICLR, 2024)

Improving Equivariant Networks with Probabilistic Symmetry Breaking. - Hannah Lawrence; Vasco Portilheiro; Yan Zhang 067; Sékou-Oumar Kaba (ICLR, 2025)