Limit Analysis for Symbolic Multi-step Reasoning Tasks with Information Propagation Rules Based on Transformers
Organizations: School of Mathematical Sciences, Shanghai Jiao Tong University · School of Mathematics and Statistics, Wuhan University · Institute of Natural Sciences, MOE-LSC, Shanghai Jiao Tong University
Abstract
Transformers are able to perform reasoning tasks, however the intrinsic mechanism remains widely open. In this paper we propose a set of information propagation rules based on Transformers and utilize symbolic reasoning tasks to theoretically analyze the limit reasoning steps. We show that the limit number of reasoning steps is between and for a model with attention layers in a single-pass. In addition, we find that, for problems with a fixed number of reasoning steps, Transformers exhibit a preference for generating sequences that satisfy certain ordering relations. Ultimately, we construct a Transformer and prove that the limiting reasoning depth is achievable.
Figures & tables
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| the number of reasoning steps | 3 | 4 | 5 |
| the size of datasets | 1200000 | 6000000 | 30000000 |
| 64 | 128 | 256 | 512 | 1024 | |
|---|---|---|---|---|---|
| Test accuracy (%) | 9.6 | 11.9 | 72.2 | 99.6 | 99.6 |
| the number of reasoning steps | 1 | 2 |
|---|---|---|
| test accuracy(%) | 100 | 8.4 |
Transformer
Transformers are neural architectures that use attention to model relationships among sequence elements in language, vision, and scientific data.
See all 558 papers