Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks
Open accessAbstract
Many machine learning tasks such as multiple instance learning, 3D shape recognition, and few-shot image classification are defined on sets of instances. Since solutions to such problems do not depend on the order of elements of the set, models used to address them should be permutation invariant. We present an attention-based neural network module, the Set Transformer, specifically designed to model interactions among elements in the input set. The model consists of an encoder and a decoder, both of which rely on attention mechanisms. In an effort to reduce computational complexity, we introduce an attention scheme inspired by inducing point methods from sparse Gaussian process literature. It reduces the computation time of self-attention from quadratic to linear in the number of elements in the set. We show that our model is theoretically attractive and we evaluate it on a range of tasks, demonstrating the state-of-the-art performance compared to recent methods for set-structured data.
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
Self-attention is an equivariant transformation capable of performing pooling operations like max-pooling.
“This is not surprising, given that self-attention is an equivariant transformation capable of performing pooling operations like max-pooling [12].”
From Exploring Length Generalization in Large Language Models (2022) · cites this paper as [12]
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Cites
Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A. N., Kaiser, L., & Polosukhin, I. (2017) · arXiv · doi:10.48550/arxiv.1706.03762 · Cited by 120,404 · see citing works
ImageNet classification with deep convolutional neural networks
Krizhevsky, A., Sutskever, I., & Hinton, G. E. (2017) · Communications of the ACM · doi:10.1145/3065386 · Cited by 112,155 · see citing works
Adam: A Method for Stochastic Optimization
Kingma, D. P., & Ba, J. (2014) · arXiv · doi:10.48550/arxiv.1412.6980 · Cited by 98,946 · see citing works
Very Deep Convolutional Networks for Large-Scale Image Recognition
Simonyan, K., & Zisserman, A. (2014) · arXiv · doi:10.48550/arxiv.1409.1556 · Cited by 74,248 · see citing works
Layer NormalizationPreprint
Ba, J., Kiros, J., & Hinton, G. E. (2016) · arXiv.org · doi:10.48550/arxiv.1607.06450 · Cited by 13,745 · see citing works
PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
Charles, R., Su, H., Mo, K., & Guibas, L. J. (2017) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr.2017.16 · Cited by 9,767 · see citing works
Speech recognition with deep recurrent neural networks
Graves, A., Mohamed, A.-R., & Hinton, G. (2013) · 2013 IEEE International Conference on Acoustics, Speech and Signal Processing · doi:10.1109/icassp.2013.6638947 · Cited by 8,893 · see citing works
Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks
Finn, C., Abbeel, P., & Levine, S. (2017) · arXiv · doi:10.48550/arxiv.1703.03400 · Cited by 8,512 · see citing works
Prototypical Networks for Few-shot LearningPreprint
Snell, J., Swersky, K., & Zemel, R. S. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1703.05175 · Cited by 5,863 · see citing works
3D ShapeNets: A deep representation for volumetric shapes
Wu, Z., Song, S., Khosla, A., Yu, F., Zhang, L., Tang, X., & Xiao, J. (2015) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr.2015.7298801 · Cited by 5,413 · see citing works
Multi-view Convolutional Neural Networks for 3D Shape Recognition
Su, H., Maji, S., Kalogerakis, E., & Miller, E. G. (2015) · IEEE/CVF International Conference on Computer Vision (ICCV) · doi:10.1109/iccv.2015.114 · Cited by 3,515 · see citing works
Solving the multiple instance problem with axis-parallel rectangles
Dietterich, T. G., Lathrop, R. H., & Lozano-Pérez, T. (1997) · Artificial Intelligence · doi:10.1016/s0004-3702(96)00034-3 · Cited by 2,866 · see citing works
Human-level concept learning through probabilistic program induction
Lake, B. M., Salakhutdinov, R., & Tenenbaum, J. B. (2015) · Science · doi:10.1126/science.aab3050 · Cited by 2,817 · see citing works
ShapeNet: An Information-Rich 3D Model Repository
Chang, A. X., Funkhouser, T., Guibas, L., Hanrahan, P., Huang, Q., Li, Z., Savarese, S., Savva, M., Song, S., Su, H., et al. (2015) · arXiv · doi:10.48550/arxiv.1512.03012 · Cited by 2,330 · see citing works
A Unifying View of Sparse Approximate Gaussian Process Regression
Quiñonero-Candela, J., & Rasmussen, C. E. (2005) · doi:10.5555/1046920.1194909 · Cited by 1,739 · see citing works
ShapeNet: An Information-Rich 3D Model Repository
Chang, A. X., Funkhouser, T. A., Guibas, L. J., Hanrahan, P., Qi-xing Huang, A., Li, Z., Savarese, S., Savva, M., Song, S., Su, H., et al. (2023) · doi:10.5281/zenodo.10089705 · Cited by 1,624 · see citing works
Sparse Gaussian Processes using Pseudo-inputs
Snelson, E., & Ghahramani, Z. (2005) · Cited by 1,415 · see citing works
Spectral grouping using the nystrom method
Fowlkes, C. C., Belongie, S., Chung, F., & Malik, J. (2004) · IEEE Transactions on Pattern Analysis and Machine Intelligence · doi:10.1109/tpami.2004.1262185 · Cited by 1,310 · see citing works
Thrun, S., & Pratt, L. (1998) · Springer eBooks · doi:10.1007/978-1-4615-5529-2 · Cited by 1,291 · see citing works
A simple neural network module for relational reasoning
Santoro, A., Raposo, D., Barrett, D. G. T., Malinowski, M., Pascanu, R., Battaglia, P., & Lillicrap, T. (2017) · arXiv · doi:10.48550/arxiv.1706.01427 · Cited by 1,281 · see citing works
A Framework for Multiple-Instance Learning
Maron, O., & Lozano‐Pérez, T. (1997) · Cited by 1,232 · see citing works
A Simple Neural Attentive Meta-LearnerPreprint
Mishra, N., Rohaninejad, M., Chen, X., & Abbeel, P. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1707.03141 · Cited by 961 · see citing works
Order Matters: Sequence to sequence for setsPreprint
Vinyals, O., Samy Bengio, A., & Kudlur, M. (2015) · arXiv (Cornell University) · doi:10.48550/arxiv.1511.06391 · Cited by 696 · see citing works
Towards a Neural StatisticianPreprint
Edwards, H., & Storkey, A. (2016) · International Conference on Learning Representations · doi:10.48550/arxiv.1606.02185 · Cited by 450 · see citing works
DeepPano: Deep Panoramic Representation for 3-D Shape Recognition
Shi, B., Bai, S., Zhou, Z., & Bai, X. (2015) · IEEE Signal Processing Letters · doi:10.1109/lsp.2015.2480802 · Cited by 413 · see citing works
Zaheer, M., Kottur, S., Ravanbakhsh, S., Póczos, B., Salakhutdinov, R. R., & Smola, A. J. (2017) · neural information processing systems · Cited by 279 · see citing works
Attentive Neural ProcessesPreprint
Kim, H., Mnih, A., Schwarz, J., Garnelo, M., Eslami, A., Rosenbaum, D., Vinyals, O., & Teh, Y. W. (2019) · arXiv (Cornell University) · doi:10.48550/arxiv.1901.05761 · Cited by 257 · see citing works
Self-citationNeural Episodic ControlPreprint
Pritzel, A., Uría, B., Srinivasan, S., Puigdomènech, A., Vinyals, O., Hassabis, D., Wierstra, D., & Blundell, C. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1703.01988 · Cited by 235 · see citing works
Discovering Causal Signals in Images
Lopez-Paz, D., Nishihara, R., Chintala, S., Scholkopf, B., & Bottou, L. (2017) · 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr.2017.14 · Cited by 209 · see citing works
Attention-based Deep Multiple Instance Learning
Ilse, M., Tomczak, J. M., & Welling, M. (2018) · UvA-DARE (University of Amsterdam) · Cited by 155 · see citing works
Robust Attentional Aggregation of Deep Feature Sets for Multi-view 3D Reconstruction
Yang, B., Wang, S., Markham, A., & Trigoni, N. (2019) · International Journal of Computer Vision · doi:10.1007/s11263-019-01217-w · Cited by 147 · see citing works
Attend and Interact: Higher-Order Object Interactions for Video Understanding
Ma, C., Kadav, A., Melvin, I., Kira, Z., AlRegib, G., & Graf, H. P. (2018) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr.2018.00710 · Cited by 137 · see citing works
On the Limitations of Representing Functions on Sets
Wagstaff, E., Fuchs, F. B., Engelcke, M., Posner, I., & Osborne, M. A. (2019) · Oxford University Research Archive (ORA) (University of Oxford) · doi:10.48550/arxiv.1901.09006 · Cited by 113 · see citing works
Learning from Distributions via Support Measure MachinesPreprint
Muandet, K., Fukumizu, K., Dinuzzo, F., & Schölkopf, B. (2012) · arXiv (Cornell University) · doi:10.48550/arxiv.1202.6504 · Cited by 91 · see citing works
Probabilistic symmetries and invariant neural networksPreprint
Bloem-Reddy, B., & Teh, Y. W. (2019) · arXiv (Cornell University) · doi:10.48550/arxiv.1901.06082 · Cited by 66 · see citing works
Self-citationDistribution to Distribution Regression
Oliva, J. B., Póczos, B., & Schneider, J. (2018) · Figshare · doi:10.1184/r1/6475571 · Cited by 53 · see citing works
Robust Attentional Aggregation of Deep Feature Sets for Multi-view 3D Reconstruction
Yang, B., Wang, S., Markham, A., & Trigoni, N. (2018) · arXiv · doi:10.48550/arxiv.1808.00758
Flagged first, then by citations
Research works printed but not linked: 5
- Garnelo, M., Rosenbaum, D., Maddison, C. J., Ramalho, T., Saxton, D., Shanahan, M., Teh, Y. W., Rezende, D. J., and Eslami, S. M. A. Conditional neural processes. In Proceedings of the International Conference on Machine Learning (ICML), 2018.
- Lee, Y. and Choi, S. Gradient-based meta-learning with learned layerwise metric and subspace. In Proceedings of the International Conference on Machine Learning (ICML), 2018.
- Oliva, J., Póczos, B., and Schneider, J. Distribution to dis- tribution regression. In Proceedings of the International Conference on Machine Learning (ICML), 2013.
- Schmidhuber, J. Evolutionary Principles in Self-Referential Learning. PhD thesis, Technical University of Munich, 1987.
- Zaheer, M., Kottur, S., Ravanbakhsh, S., Poczos, B., Salakhutdinov, R. R., and Smola, A. J. Deep sets. In Advances in Neural Information Processing Systems (NeurIPS), 2017.
Cited by
Sorted by citations
Transformers in Vision: A Survey
Khan, S., Naseer, M., Hayat, M., Zamir, S. W., Khan, F. S., & Shah, M. (2022) · ACM Computing Surveys · doi:10.1145/3505244 · Cited by 3,595 · see citing works
SuperGlue: Learning Feature Matching With Graph Neural Networks
Sarlin, P.-E., DeTone, D., Malisiewicz, T., & Rabinovich, A. (2020) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr42600.2020.00499 · Cited by 3,010 · see citing works
Zhao, H., Jiang, L., Jia, J., Torr, P. H. S., & Koltun, V. (2021) · IEEE/CVF International Conference on Computer Vision (ICCV) · doi:10.1109/iccv48922.2021.01595 · Cited by 2,847 · see citing works
Attention mechanisms in computer vision: A survey
Guo, M.-H., Xu, T.-X., Liu, J.-J., Liu, Z.-N., Jiang, P.-T., Mu, T.-J., Zhang, S.-H., Martin, R. R., Cheng, M.-M., & Hu, S.-M. (2022) · Computational Visual Media · doi:10.1007/s41095-022-0271-y · Cited by 2,566 · see citing works
Lin, T., Wang, Y., Liu, X., & Qiu, X. (2022) · AI Open · doi:10.1016/j.aiopen.2022.10.001 · Cited by 1,422 · see citing works
Pre-trained models: Past, present and future
Han, X., Zhang, Z., Ding, N., Gu, Y., Liu, X., Huo, Y., Qiu, J., Yao, Y., Zhang, A., Zhang, L., et al. (2021) · AI Open · doi:10.1016/j.aiopen.2021.08.002 · Cited by 994 · see citing works
Interpretable machine learning: Fundamental principles and 10 grand challenges
Rudin, C., Chen, C., Chen, Z., Huang, H., Semenova, L., & Zhong, C. (2022) · Statistics Surveys · doi:10.1214/21-ss133 · Cited by 958 · see citing works
Zhai, X., Kolesnikov, A., Houlsby, N., & Beyer, L. (2022) · 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr52688.2022.01179 · Cited by 891 · see citing works
Meta-Learning in Neural Networks: A Survey
Hospedales, T. M., Antoniou, A., Micaelli, P., & Storkey, A. J. (2021) · IEEE Transactions on Pattern Analysis and Machine Intelligence · doi:10.1109/tpami.2021.3079209 · Cited by 886 · see citing works
Reinforcement learning for combinatorial optimization: A survey
Mazyavkina, N., Sviridov, S., Ivanov, S., & Burnaev, E. V. (2021) · Computers & Operations Research · doi:10.1016/j.cor.2021.105400 · Cited by 762 · see citing works
Advances in medical image analysis with vision Transformers: A comprehensive review
Azad, R., Kazerouni, A., Heidari, M., Aghdam, E. K., Molaei, A., Jia, Y., Jose, A., Roy, R., & Merhof, D. (2023) · Medical Image Analysis · doi:10.1016/j.media.2023.103000 · Cited by 523 · see citing works
3D Object Detection with Pointformer
Pan, X., Xia, Z., Song, S., Li, L. E., & Huang, G. M. (2021) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr46437.2021.00738 · Cited by 493 · see citing works
TransferTransfo: A Transfer Learning Approach for Neural Network Based Conversational AgentsPreprint
Wolf, T., Sanh, V., Chaumond, J., & Delangue, C. (2019) · arXiv (Cornell University) · doi:10.48550/arxiv.1901.08149 · Cited by 435 · see citing works
Perceiver: General Perception with Iterative Attention
Jaegle, A., Gimeno, F., Brock, A., Zisserman, A., Vinyals, O., & Carreira, J. (2021) · arXiv · doi:10.48550/arxiv.2103.03206 · Cited by 414 · see citing works
Object-Centric Learning with Slot AttentionPreprint
Locatello, F., Weissenborn, D., Unterthiner, T., Mahendran, A., Heigold, G., Uszkoreit, J., Dosovitskiy, A., & Kipf, T. (2020) · arXiv (Cornell University) · doi:10.48550/arxiv.2006.15055 · Cited by 399 · see citing works
Multimodal Co-Attention Transformer for Survival Prediction in Gigapixel Whole Slide Images
Chen, R. J., Lu, M. Y., Weng, W.-H., Chen, T. Y., Williamson, D. F., Manz, T., Shady, M., & Mahmood, F. (2021) · 2021 IEEE/CVF International Conference on Computer Vision (ICCV) · doi:10.1109/iccv48922.2021.00398 · Cited by 398 · see citing works
Nyströmformer: A Nyström-based Algorithm for Approximating Self-Attention
Xiong, Y., Zeng, Z., Chakraborty, R., Tan, M., Fung, G., Li, Y., & Singh, V. (2021) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v35i16.17664 · Cited by 376 · see citing works
Multi-Scale Vision Longformer: A New Vision Transformer for High-Resolution Image Encoding
Zhang, P., Dai, X., Yang, J., Xiao, B., Yuan, L., Zhang, L., & Gao, J. (2021) · IEEE/CVF International Conference on Computer Vision (ICCV) · doi:10.1109/iccv48922.2021.00299 · Cited by 358 · see citing works
Perceiver IO: A General Architecture for Structured Inputs & OutputsPreprint
Jaegle, A., Borgeaud, S., Alayrac, J.-B., Doersch, C., Ionescu, C., Ding, X. D., Koppula, S., Zoran, D., Brock, A., Shelhamer, E., et al. (2021) · arXiv (Cornell University) · doi:10.48550/arxiv.2107.14795 · Cited by 357 · see citing works
MultiPath++: Efficient Information Fusion and Trajectory Aggregation for Behavior Prediction
Varadarajan, B., Hefny, A., Srivastava, A., Refaat, K. S., Nayakanti, N., Cornman, A., Chen, K., Douillard, B., Lam, C. P., Anguelov, D., et al. (2022) · Proceedings - IEEE International Conference on Robotics and Automation/Proceedings · doi:10.1109/icra46639.2022.9812107 · Cited by 338 · see citing works
Attention, please! A survey of neural attention models in deep learning
Correia, A. D. S., & Colombini, E. L. (2022) · Artificial Intelligence Review · doi:10.1007/s10462-022-10148-x · Cited by 330 · see citing works
Engel, N., Belagiannis, V., & Dietmayer, K. (2021) · IEEE Access · doi:10.1109/access.2021.3116304 · Cited by 304 · see citing works
SoftGroup for 3D Instance Segmentation on Point Clouds
Vu, T., Kim, K., Luu, T. M., Nguyen, T. M., & Yoo, C. D. (2022) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr52688.2022.00273 · Cited by 283 · see citing works
Sun, L., Lian, Z., Liu, B., & Tao, J. (2023) · IEEE Transactions on Affective Computing · doi:10.1109/taffc.2023.3274829 · Cited by 283 · see citing works
SE(3)-Transformers: 3D Roto-Translation Equivariant Attention Networks
Fuchs, F. B., Worrall, D. E., Fischer, V., & Welling, M. (2020) · UvA-DARE (University of Amsterdam) · doi:10.48550/arxiv.2006.10503 · Cited by 280 · see citing works
Wayformer: Motion Forecasting via Simple & Efficient Attention Networks
Nayakanti, N., Al‐Rfou, R., Zhou, A., Goel, K., Refaat, K. S., & Sapp, B. (2023) · Proceedings - IEEE International Conference on Robotics and Automation/Proceedings · doi:10.1109/icra48891.2023.10160609 · Cited by 278 · see citing works
Voxel Set Transformer: A Set-to-Set Approach to 3D Object Detection from Point Clouds
He, C., Li, R., Li, S., & Zhang, L. (2022) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr52688.2022.00823 · Cited by 249 · see citing works
A Gentle Introduction to Graph Neural Networks
Sánchez-Lengeling, B., Reif, E., Pearce, A., & Wiltschko, A. B. (2021) · Distill · doi:10.23915/distill.00033 · Cited by 247 · see citing works
Hyperbolic Deep Neural Networks: A Survey
Peng, W., Varanka, T., Mostafa, A., Shi, H., & Zhao, G. (2021) · IEEE Transactions on Pattern Analysis and Machine Intelligence · doi:10.1109/tpami.2021.3136921 · Cited by 220 · see citing works
Medical Image Segmentation using Squeeze-and-Expansion Transformers
Li, S., Sui, X., Luo, X., Xu, X., Liu, Y., & Goh, R. S. M. (2021) · International Joint Conference on Artificial Intelligence · doi:10.24963/ijcai.2021/112 · Cited by 216 · see citing works
Scene Memory Transformer for Embodied Agents in Long-Horizon Tasks
Fang, K., Toshev, A., Fei-Fei, L., & Savarese, S. (2019) · 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr.2019.00063 · Cited by 178 · see citing works
Horiguchi, S., Fujita, Y., Watanabe, S., Xue, Y., & Nagamatsu, K. (2020) · Interspeech (USB) · doi:10.21437/interspeech.2020-1022 · Cited by 182 · see citing works
Transformer Dissection: An Unified Understanding for Transformer’s Attention via the Lens of Kernel
Tsai, Y.-H. H., Bai, S., Yamada, M., Morency, L.-P., & Salakhutdinov, R. (2019) · Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP) · doi:10.18653/v1/d19-1443 · Cited by 173 · see citing works
How to Represent Part-Whole Hierarchies in a Neural Network
Hinton, G. E. (2022) · Neural Computation · doi:10.1162/neco_a_01557 · Cited by 173 · see citing works
Semi-Supervised 3D Hand-Object Poses Estimation with Interactions in Time
Liu, S., Jiang, H., Xu, J., Liu, S., & Wang, X. (2021) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr46437.2021.01445 · Cited by 169 · see citing works
Random Feature AttentionPreprint
Peng, H., Pappas, N., Yogatama, D., Schwartz, R., Smith, N. A., & Kong, L. (2021) · arXiv (Cornell University) · doi:10.48550/arxiv.2103.02143 · Cited by 169 · see citing works
Learning to Match Features with Seeded Graph Matching Network
Chen, H., Luo, Z., Zhang, J., Zhou, L., Bai, X., Hu, Z., Chiew-lan Tai, B., & Quan, L. (2021) · IEEE/CVF International Conference on Computer Vision (ICCV) · doi:10.1109/iccv48922.2021.00624 · Cited by 161 · see citing works
RayMVSNet++: Learning Ray-Based 1D Implicit Fields for Accurate Multi-View Stereo
Shi, Y., Xi, J., Hu, D., Cai, Z., & Xu, K. (2023) · IEEE Transactions on Pattern Analysis and Machine Intelligence · doi:10.1109/tpami.2023.3296163 · Cited by 160 · see citing works
Enable Deep Learning on Mobile Devices: Methods, Systems, and Applications
Cai, H., Lin, J., Lin, Y., Liu, Z., Tang, H., Wang, H., Zhu, L., & Han, S. (2022) · ACM Transactions on Design Automation of Electronic Systems · doi:10.1145/3486618 · Cited by 152 · see citing works
Efficient Attentions for Long Document Summarization
Huang, L., Cao, S., Parulian, N., Ji, H., & Wang, L. (2021) · Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies · doi:10.18653/v1/2021.naacl-main.112 · Cited by 146 · see citing works
Combining Reinforcement Learning and Constraint Programming for Combinatorial Optimization
Cappart, Q., Moisan, T., Rousseau, L.-M., Prémont-Schwarz, I., & Ciré, A. A. (2021) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v35i5.16484 · Cited by 146 · see citing works
Self-Positioning Point-Based Transformer for Point Cloud Understanding
Park, J., Lee, S., Kim, S., Xiong, Y., & Kim, H. J. (2023) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr52729.2023.02089 · Cited by 139 · see citing works
Buterez, D., Janet, J. P., Kiddle, S. J., Oglić, D., & Lió, P. (2024) · Nature Communications · doi:10.1038/s41467-024-45566-8 · Cited by 139 · see citing works
Accurate Learning of Graph Representations with Graph Multiset PoolingPreprint
Baek, J., Kang, M., & Hwang, S. J. (2021) · arXiv (Cornell University) · doi:10.48550/arxiv.2102.11533 · Cited by 130 · see citing works
Münchmeyer, J., Bindi, D., Leser, U., & Tilmann, F. J. (2021) · Geophysical Journal International · doi:10.1093/gji/ggab139 · Cited by 129 · see citing works
Pang, L., Xu, J., Ai, Q., Lan, Y., Cheng, X., & Wen, J.-R. (2020) · International ACM SIGIR Conference on Research and Development in Information Retrieval (SIGIR) · doi:10.1145/3397271.3401104 · Cited by 127 · see citing works
Delving Deeper into Anti-Aliasing in ConvNets
Zou, X., Xiao, F., Yu, Z., Li, Y., & Lee, Y. J. (2022) · International Journal of Computer Vision · doi:10.1007/s11263-022-01672-y · Cited by 117 · see citing works
Self-citationVcT: Visual Change Transformer for Remote Sensing Image Change Detection
Jiang, B., Wang, Z., Wang, X., Zhang, Z., Chen, L., Wang, X., & Luo, B. (2023) · IEEE Transactions on Geoscience and Remote Sensing · doi:10.1109/tgrs.2023.3327139 · Cited by 115 · see citing works
On the Limitations of Representing Functions on Sets
Wagstaff, E., Fuchs, F. B., Engelcke, M., Posner, I., & Osborne, M. A. (2019) · Oxford University Research Archive (ORA) (University of Oxford) · doi:10.48550/arxiv.1901.09006 · Cited by 113 · see citing works
Learning disentangled representations in the imaging domain
Liu, X., Sanchez, P., Thermos, S., O’Neil, A. Q., & Tsaftaris, S. A. (2022) · Medical Image Analysis · doi:10.1016/j.media.2022.102516 · Cited by 112 · see citing works
Chasing Sparsity in Vision Transformers: An End-to-End Exploration
Chen, T., Cheng, Y., Gan, Z., Yuan, L., Zhang, L., & Wang, Z. (2021) · arXiv · doi:10.48550/arxiv.2106.04533 · Cited by 111 · see citing works
PointMixer: MLP-Mixer for Point Cloud Understanding
Choe, J., Park, C., Rameau, F., Park, J., & Kweon, I. S. (2022) · Lecture notes in computer science · doi:10.1007/978-3-031-19812-0_36 · Cited by 111 · see citing works
Empirical Bayes Transductive Meta-Learning with Synthetic GradientsPreprint
Hu, S. X., Moreno, P. G., Xiao, Y., Shen, X., Obozinski, G., Lawrence, N. D., & Damianou, A. (2020) · arXiv (Cornell University) · doi:10.48550/arxiv.2004.12696 · Cited by 110 · see citing works
MotionLM: Multi-Agent Motion Forecasting as Language Modeling
Seff, A., Cera, B., Chen, D., Ng, M., Zhou, A., Nayakanti, N., Refaat, K. S., Al‐Rfou, R., & Sapp, B. (2023) · IEEE/CVF International Conference on Computer Vision (ICCV) · doi:10.1109/iccv51070.2023.00788 · Cited by 108 · see citing works
Conditional Text Image Generation with Diffusion Models
Zhu, Y., Li, Z., Wang, T., He, M., & Yao, C. (2023) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr52729.2023.01368 · Cited by 102 · see citing works
Modern Hopfield Networks and Attention for Immune Repertoire ClassificationPreprint
Widrich, M., Schäfl, B., Pavlović, M., Ramsauer, H., Gruber, L., Holzleitner, M., Brandstetter, J., Sandve, G. K., Greiff, V., Hochreiter, S., et al. (2020) · bioRxiv (Cold Spring Harbor Laboratory) · doi:10.1101/2020.04.12.038158 · Cited by 102 · see citing works
Predicting cellular responses to perturbation across diverse contexts with State
Adduri, A. K., Gautam, D., Bevilacqua, B., Naghipourfar, M., Imran, A., Shah, R., Teyssier, N., Verma, R., Carpenter, C., Eraslan, B., et al. (2026) · Cell · doi:10.1016/j.cell.2026.07.052 · Cited by 100 · see citing works
Learning urban region representations with POIs and hierarchical graph infomax
Huang, W., Zhang, D., Mai, G., Guo, X., & Cui, L. (2023) · ISPRS Journal of Photogrammetry and Remote Sensing · doi:10.1016/j.isprsjprs.2022.11.021 · Cited by 100 · see citing works
Hyperbolic Neural Networks++Preprint
Shimizu, R., Mukuta, Y., & Harada, T. (2020) · arXiv (Cornell University) · doi:10.48550/arxiv.2006.08210 · Cited by 92 · see citing works
OcTr: Octree-Based Transformer for 3D Object Detection
Zhou, C., Zhang, Y., Chen, J., & Huang, D. (2023) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr52729.2023.00500 · Cited by 88 · see citing works
Annotating metabolite mass spectra with domain-inspired chemical formula transformers
Goldman, S., Wohlwend, J., Stražar, M., Haroush, G., Xavier, R. J., & Coley, C. W. (2023) · Nature Machine Intelligence · doi:10.1038/s42256-023-00708-3 · Cited by 87 · see citing works
VILA: Learning Image Aesthetics from User Comments with Vision-Language Pretraining
Ke, J., Ye, K., Yu, J., Wu, Y., Milanfar, P., & Yang, F. (2023) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr52729.2023.00968 · Cited by 87 · see citing works
Pan, X., Lin, Q. M., Ye, S., Li, L., Guo, L., & Harmon, B. A. (2024) · Heritage Science · doi:10.1186/s40494-024-01179-4 · Cited by 87 · see citing works
Fast Transformers with Clustered AttentionPreprint
Vyas, A., Katharopoulos, A., & Fleuret, F. (2020) · arXiv (Cornell University) · doi:10.48550/arxiv.2007.04825 · Cited by 86 · see citing works
CaloClouds: fast geometry-independent highly-granular calorimeter simulation
Buhmann, E., Diefenbacher, S., Eren, E., Gaede, F., Kasicezka, G., Korol, A., Korcari, W., Krüger, K., & McKeown, P. (2023) · Journal of Instrumentation · doi:10.1088/1748-0221/18/11/p11025 · Cited by 85 · see citing works
Convolutional Conditional Neural ProcessesPreprint
Gordon, J. A. R., Bruinsma, W. P., Foong, A. Y. K., Requeima, J., Dubois, Y., & Turner, R. E. (2019) · arXiv (Cornell University) · doi:10.17863/cam.48067 · Cited by 84 · see citing works
Morphological Prototyping for Unsupervised Slide Representation Learning in Computational Pathology
Song, A. H., Chen, R. J., Ding, T., Williamson, D. F. K., Jaume, G., & Mahmood, F. (2024) · 2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr52733.2024.01099 · Cited by 79 · see citing works
You are AllSet: A Multiset Function Framework for Hypergraph Neural NetworksPreprint
Chien, E., Pan, C., Peng, J., & Milenković, O. (2021) · arXiv (Cornell University) · doi:10.48550/arxiv.2106.13264 · Cited by 79 · see citing works
Towards Bridging Event Captioner and Sentence Localizer for Weakly Supervised Dense Event Captioning
Chen, S., & Jiang, Y. (2021) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr46437.2021.00832 · Cited by 78 · see citing works
Reinforcement Learning in Game Industry—Review, Prospects and Challenges
Souchleris, K., Sidiropoulos, G. K., & Papakostas, G. A. (2023) · Applied Sciences · doi:10.3390/app13042443 · Cited by 78 · see citing works
Deep Learning Methods for Calibrated Photometric Stereo and Beyond
Ju, Y., Lam, K., Xie, W., Zhou, H., Dong, J., & Shi, B. (2024) · IEEE Transactions on Pattern Analysis and Machine Intelligence · doi:10.1109/tpami.2024.3388150 · Cited by 78 · see citing works
SA-Det3D: Self-Attention Based Context-Aware 3D Object Detection
Bhattacharyya, P., Huang, C. G., & Czarnecki, K. (2021) · IEEE International Conference on Computer Vision Workshops · doi:10.1109/iccvw54120.2021.00337 · Cited by 77 · see citing works
A Survey of 6DoF Object Pose Estimation Methods for Different Application Scenarios
Guan, J., Hao, Y., Wu, Q., Li, S., & Fang, Y. (2024) · Sensors · doi:10.3390/s24041076 · Cited by 76 · see citing works
Point TransformerPreprint
Zhao, H., Jiang, L., Jia, J., Torr, P. H. S., & Koltun, V. (2020) · arXiv (Cornell University) · doi:10.48550/arxiv.2012.09164 · Cited by 75 · see citing works
Combining Machine Learning with Physical Knowledge in Thermodynamic Modeling of Fluid Mixtures
Jirasek, F., & Hasse, H. (2023) · Annual Review of Chemical and Biomolecular Engineering · doi:10.1146/annurev-chembioeng-092220-025342 · Cited by 74 · see citing works
Language-Driven Representation Learning for Robotics
Karamcheti, S., Nair, S., Chen, A., Kollar, T., Finn, C., Sadigh, D., & Liang, P. (2023) · Robotics: Science and Systems · doi:10.15607/rss.2023.xix.032 · Cited by 74 · see citing works
Right for the Right Concept: Revising Neuro-Symbolic Concepts by Interacting with their Explanations
Stammer, W., Schramowski, P., & Kersting, K. (2021) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr46437.2021.00362 · Cited by 73 · see citing works
Multilevel Scattering Center and Deep Feature Fusion Learning Framework for SAR Target Recognition
Liu, Z., Wang, L., Wen, Z., Li, K., & Pan, Q. (2022) · IEEE Transactions on Geoscience and Remote Sensing · doi:10.1109/tgrs.2022.3174703 · Cited by 73 · see citing works
Machine learning for predicting the viscosity of binary liquid mixtures
Bilodeau, C., Kazakov, A. F., Mukhopadhyay, S., Emerson, J., Kalantar, T. H., Muzny, C. D., & Jensen, K. F. (2023) · Chemical Engineering Journal · doi:10.1016/j.cej.2023.142454 · Cited by 72 · see citing works
Capsules with Inverted Dot-Product Attention RoutingPreprint
Tsai, Y.-H. H., Srivastava, N., Goh, H., & Salakhutdinov, R. (2020) · arXiv (Cornell University) · doi:10.48550/arxiv.2002.04764 · Cited by 71 · see citing works
Encoder-Decoder Based Attractors for End-to-End Neural Diarization
Horiguchi, S., Fujita, Y., Watanabe, S., Xue, Y., & Garcia, L. P. (2022) · IEEE/ACM Transactions on Audio Speech and Language Processing · doi:10.1109/taslp.2022.3162080 · Cited by 69 · see citing works
SSTrans-Net: Smart Swin Transformer Network for medical image segmentation
Fu, L., Chen, Y., Ji, W., & Feng, Y. (2024) · Biomedical Signal Processing and Control · doi:10.1016/j.bspc.2024.106071 · Cited by 68 · see citing works
SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data
Kim, J.-W., Yoo, J., Lee, J., & Hong, S. (2021) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr46437.2021.01481 · Cited by 67 · see citing works
Self-citationLong-Short Transformer: Efficient Transformers for Language and Vision
Zhu, C., Ping, W., Xiao, C., Shoeybi, M., Goldstein, T., Anandkumar, A., & Catanzaro, B. (2021) · CaltechAUTHORS (California Institute of Technology) · doi:10.48550/arxiv.2107.02192 · Cited by 67 · see citing works
Luna: Linear Unified Nested AttentionPreprint
Ma, X., Kong, X., Wang, S., Zhou, C., May, J., Ma, H., & Zettlemoyer, L. (2021) · arXiv (Cornell University) · doi:10.48550/arxiv.2106.01540 · Cited by 67 · see citing works
Deep learning methods in metagenomics: a review
Roy, G., Prifti, E., Belda, E., & Zucker, J. (2024) · Microbial Genomics · doi:10.1099/mgen.0.001231 · Cited by 66 · see citing works
Probabilistic symmetries and invariant neural networksPreprint
Bloem-Reddy, B., & Teh, Y. W. (2019) · arXiv (Cornell University) · doi:10.48550/arxiv.1901.06082 · Cited by 66 · see citing works
Self-citationImproving GNSS Positioning Using Neural-Network-Based Corrections
Kanhere, A. V., Gupta, S., Shetty, A., & Gao, G. (2022) · NAVIGATION Journal of the Institute of Navigation · doi:10.33012/navi.548 · Cited by 64 · see citing works
EPiC-GAN: Equivariant point cloud generation for particle jets
Buhmann, E., Kasieczka, G., & Thaler, J. (2023) · SciPost Physics · doi:10.21468/scipostphys.15.4.130 · Cited by 64 · see citing works
Stacked Capsule Graph Autoencoders for geometry-aware 3D head pose estimation
Hong, C., Chen, L., Liang, Y., & Zeng, Z. (2021) · Computer Vision and Image Understanding · doi:10.1016/j.cviu.2021.103224 · Cited by 62 · see citing works
Self-Attention Between Datapoints: Going Beyond Individual Input-Output Pairs in Deep LearningPreprint
Kossen, J., Band, N., Lyle, C., Gomez, A. N., Rainforth, T., & Gal, Y. (2021) · arXiv (Cornell University) · doi:10.48550/arxiv.2106.02584 · Cited by 62 · see citing works
ATISS: Autoregressive Transformers for Indoor Scene Synthesis
Paschalidou, D., Kar, A., Shugrina, M., Kreis, K., Geiger, A., & Fidler, S. (2021) · arXiv · doi:10.48550/arxiv.2110.03675 · Cited by 62 · see citing works
MuRCL: Multi-Instance Reinforcement Contrastive Learning for Whole Slide Image Classification
Zhu, Z., Yu, L., Wu, W., Yu, R., Zhang, D., & Wang, L. (2022) · IEEE Transactions on Medical Imaging · doi:10.1109/tmi.2022.3227066 · Cited by 61 · see citing works
Set Functions for Time SeriesPreprint
Horn, M., Moor, M., Bock, C., Rieck, B., & Borgwardt, K. M. (2019) · arXiv (Cornell University) · doi:10.3929/ethz-b-000462048 · Cited by 61 · see citing works
Neural Symbolic Regression that ScalesPreprint
Biggio, L., Bendinelli, T., Neitz, A., Lucchi, A., & Parascandolo, G. (2021) · arXiv (Cornell University) · doi:10.48550/arxiv.2106.06427 · Cited by 61 · see citing works
Adaptive Kalman-Informed Transformer
Cohen, N., & Klein, I. (2025) · Engineering Applications of Artificial Intelligence · doi:10.1016/j.engappai.2025.110221 · Cited by 60 · see citing works
Coordination among neural modules through a shared global workspace
Goyal, A., Didolkar, A., Lamb, A., Badola, K., Ke, N. R., Rahaman, N., Binas, J., Blundell, C., Mozer, M., & Bengio, Y. (2021) · arXiv · doi:10.48550/arxiv.2103.01197 · Cited by 59 · see citing works
MemGPT: Towards LLMs as Operating Systems
Packer, C., Wooders, S., Lin, K., Fang, V., Patil, S. G., Stoica, I., & Gonzalez, J. E. (2023) · arXiv · doi:10.48550/arxiv.2310.08560 · Cited by 64 · see citing works
SymFormer: End-to-End Symbolic Regression Using Transformer-Based Architecture
Vastl, M., Kulhánek, J., Kubalı́k, J., Derner, E., & Babuška, R. (2024) · IEEE Access · doi:10.1109/access.2024.3374649 · Cited by 57 · see citing works
4SDrug: Symptom-based Set-to-set Small and Safe Drug Recommendation
Tan, Y., Kong, C., Yu, L., Li, P., Chen, C., Zheng, X., Hertzberg, V., & Yang, C. (2022) · ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD) · doi:10.1145/3534678.3539089 · Cited by 55 · see citing works
- Authors
- Lee, J., Lee, Y., Kim, J., Kosiorek, A. R., Choi, S., & Teh, Y. W.
- Affiliations
- University of Oxford (GB)
- Pohang University of Science and Technology (KR)
- Venue
- arXiv
- Publisher
- arXiv
- Published
- 2018
- ISSN
- 2331-8422 (ISSN-L)
- arXiv
- 1810.00825
- OpenAlex
- W2894384847
- Cited by
- 691
- References
- 37
- Citations by year
- 2026: 20 · 2025: 84 · 2024: 116 · 2023: 137 · 2022: 96 · 2021: 145 · 2020: 76 · 2019: 16 · 2018: 1
- Type
- posted content
- Language
- en
- Open access
- green
- arxiv.org
- Locations
- arXiv (Cornell University) (submittedVersion)
- Oxford University Research Archive (ORA) (University of Oxford) (submittedVersion, cc-by)
- arXiv (Cornell University)
- Funders
- European Commission (2007–2013, FP7 2007-2013, FP7/2007)
- Ministry of Science and ICT, South Korea (2017-0-01779)
- Samsung
- Engineering and Physical Sciences Research Council (EP/P026753/1)
- Institute for Information and Communications Technology Promotion (2017‐0‐01779, No.2017-0-01779)
- FP7 Ideas: European Research Council (617071)
- Topics
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Neural Network Applications
- Physical Sciences › Computer Science › Artificial Intelligence › Neural Networks and Applications
- Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
- Keywords
- Set Transformer, permutation invariance, attention mechanism, multiple instance learning, 3D shape recognition, few-shot image classification, inducing points, sparse Gaussian processes, self-attention complexity, encoder-decoder architecture
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review
