The DOI matches its DataCite record; 37 of the 37 references it cites were found in a registry or an index.
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.
Overview
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
Verification
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?
The source backs part of the claim, not all of it — or backs it for a different population, condition or comparison.
In the citing paper
“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]
Experimental — accuracy not yet measured
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Citations
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
Very Deep Convolutional Networks for Large-Scale Image Recognition
Simonyan, K., & Zisserman, A. (2014) · arXiv · doi:10.48550/arxiv.1409.1556 · Cited by 114,148
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
Adam: A Method for Stochastic Optimization
Kingma, D. P., & Ba, J. (2014) · arXiv · doi:10.48550/arxiv.1412.6980 · Cited by 98,946
Layer NormalizationPreprint
Ba, J., Kiros, J., & Hinton, G. E. (2016) · arXiv.org · doi:10.48550/arxiv.1607.06450 · Cited by 13,762
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 10,029
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
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
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 6,245
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
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
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,234
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 3,005
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
Lin, T., Wang, Y., Liu, X., & Qiu, X. (2022) · AI Open · doi:10.1016/j.aiopen.2022.10.001 · Cited by 1,422
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
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
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
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
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 824
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 557
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 505
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 495
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
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 464
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
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
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 378
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 641
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 375
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 365
Engel, N., Belagiannis, V., & Dietmayer, K. (2021) · IEEE Access · doi:10.1109/access.2021.3116304 · Cited by 304
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
Sun, L., Lian, Z., Liu, B., & Tao, J. (2023) · IEEE Transactions on Affective Computing · doi:10.1109/taffc.2023.3274829 · Cited by 288
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 346
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 351
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 257
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 298
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 252
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 230
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
Horiguchi, S., Fujita, Y., Watanabe, S., Xue, Y., & Nagamatsu, K. (2020) · Interspeech (USB) · doi:10.21437/interspeech.2020-1022 · Cited by 182
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
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
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
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 247
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
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 161
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 161
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
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
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
Buterez, D., Janet, J. P., Kiddle, S. J., Oglić, D., & Lió, P. (2024) · Nature Communications · doi:10.1038/s41467-024-45566-8 · Cited by 153
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 163
Münchmeyer, J., Bindi, D., Leser, U., & Tilmann, F. J. (2021) · Geophysical Journal International · doi:10.1093/gji/ggab139 · Cited by 135
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 146
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 125
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 126
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
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 130
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
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
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 116
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 128
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 108
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 111
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
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 105
Hyperbolic Neural Networks++Preprint
Shimizu, R., Mukuta, Y., & Harada, T. (2020) · arXiv (Cornell University) · doi:10.48550/arxiv.2006.08210 · Cited by 92
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
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
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
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 92
Fast Transformers with Clustered AttentionPreprint
Vyas, A., Katharopoulos, A., & Fleuret, F. (2020) · arXiv (Cornell University) · doi:10.48550/arxiv.2007.04825 · Cited by 86
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 111
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 123
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
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 114
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
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 84
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
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
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
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
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 77
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 185
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
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
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 77
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 79
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
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
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 75
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 74
Luna: Linear Unified Nested Attention
Ma, X., Kong, X., Wang, S., Zhou, C., May, J., Ma, H., & Zettlemoyer, L. (2021) · arXiv · doi:10.48550/arxiv.2106.01540
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 73
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
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
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 83
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
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 77
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
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
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 69
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 78
Adaptive Kalman-Informed Transformer
Cohen, N., & Klein, I. (2025) · Engineering Applications of Artificial Intelligence · doi:10.1016/j.engappai.2025.110221 · Cited by 60
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
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
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
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 59
Metadata
- 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
