Feature Resemblance: Towards a Theoretical Understanding of Analogical Reasoning in Transformers
Abstract
Lay Summary
AI models sometimes seem able to make a useful guess by comparing two similar things. For example, if a model knows that two objects share some features, and it knows another fact about one of them, it may guess that the same fact also applies to the other. This paper studies how this kind of “learning by comparison” can appear during training. The main idea is that training can make the model represent similar things in similar ways. When this happens, a fact learned about one thing can more easily carry over to another related thing. This gives a simple explanation for how a model may make analogies. We find that the order of training matters. A model is more likely to make good analogies if it first learns which things are similar and then learns facts about them. If it learns separate facts first, it may memorize the examples without learning how to connect them. We also show that some multi-step reasoning needs explicit examples that teach the model how to connect the middle step. Experiments with both simple models and larger language models show similar patterns. Overall, the paper helps explain how AI models may learn to transfer knowledge from one thing to another.