Title: LEARNING TO MAKE ANALOGIES BY CONTRASTING ABSTRACT RELATIONAL STRUCTURE

Abstract: Analogical reasoning has been a principal focus of various waves of AI research. Analogy is particularly challenging for machines because it requires relational structures to be represented such that they can be flexibly applied across diverse domains of experience. Here, we study how analogical reasoning can be induced in neural networks that learn to perceive and reason about raw visual data. We find that the critical factor for inducing such a capacity is not an elaborate architecture, but rather, careful attention to the choice of data and the manner in which it is presented to the model. The most robust capacity for analogical reasoning is induced when networks learn analogies by contrasting abstract relational structures in their input domains, a training method that uses only the input data to force models to learn about important abstract features. Using this technique we demonstrate capacities for complex, visual and symbolic analogy making and generalisation in even the simplest neural network architectures.

1. INTRODUCTION:

The ability to make analogies -that is, to flexibly map familiar relations from one domain of experience to another -is a fundamental ingredient of human intelligence and creativity (Gentner, 1983; Hofstadter, 1996; Hummel & Holyoak, 1997; Lovett & Forbus, 2017) . As noted, for instance, by Holyoak & Thagard (1995) , analogies gave Roman scientists a deeper understanding of sound when they leveraged knowledge of a familiar source domain (water waves in the sea) to better understand the structure of an unfamiliar target domain (acoustics). The Romans 'aligned' relational principles about water waves (periodicity, bending round corners, rebounding off solids) to phenomena observed in acoustics, in spite of the numerous perceptual and physical differences between water and sound. This flexible alignment, or mapping, of relational structure between source and target domains, independent of perceptual congruence, is a prototypical example of analogy making.

It has proven particularly challenging to replicate processes of analogical thought in machines. Many classical or symbolic AI models lack the flexibility to apply predicates or operations across diverse domains, particularly those that may have never previously been observed. It is natural to consider, however, whether the strengths of modern neural network-based models can be exploited to solve difficult analogical problems, given their capacity to represent stimuli at different levels of abstraction and to enable flexible, context-dependent computation over noisy and ambiguous inputs.

In this work we demonstrate that well-known neural network architectures can indeed learn to make analogies with remarkable generality and flexibility. This ability, however, is critically dependent on a method of training we call learning analogies by contrasting abstract relational structure (LABC). We show that simple architectures can be trained using this approach to apply abstract relations to never-before-seen source-target domain mappings, and even to entirely unfamiliar target domains.

Our work differs from previous computational models of analogy in two important ways. First, unlike previous neural network models of analogy, we optimize a single model to perform both stimulus representation and cross-domain mapping jointly. This allows us to explore the potential benefit of interactions between perception, representation and inter-domain alignment, a question of some debate in the analogy literature (Forbus et al., 1998) . Second, we do not instantiate an ? Source Domain (shape quantity)

Target Domain (line colour) Candidates:

Relation: Progression Types of Domains shape quantity shape colour shape type shape size shape position line type line colour

Types of Relations:

Progression XOR OR AND 1) Inspect the source panels:

The number of shapes increases as you go along the panels 2) Apply the relation from (1) to the target panels:

The darkness of the lines increases as you go along the panels One shape Two shapes Three shapes

Light grey Medium grey Dark grey:

Figure 1 : A visual analogy problem. In this example, the model must (1) identify a relation (Progression ) on a particular domain ( shape quantity ) in the source sequence (top), and (2) apply it to a different domain ( line color ) in order to find the candidate answer panel that correctly completes target sequence (bottom). There are seven possible domains and four possible relations in the dataset.

explicit cognitive theory or analogy-like computation in our model architecture, but instead use this theoretical insight to inform the way in which the model is trained.

2. ANALOGIES AS HIGH-LEVEL PERCEPTION AND STRUCTURE MAPPING:

Perhaps the best-known explanation of human analogical reasoning is Structure Mapping Theory (SMT) (Gentner, 1983) . SMT emphasizes the distinction between two means of comparing domains of experience; analogy and similarity. According to SMT, two domains are similar if they share many attributes (i.e. properties that can be expressed with a one-place predicate like BLUE(sea) ), whereas they are analogous if they share few attributes but many relations (i.e. properties expressed by many-place predicates like BENDS-AROUND(sea, solid-objects) ). SMT assumes that our perceptions can be represented as collections of attributes and structured relations, and that these representations do not necessarily depend on the subsequent mappings that use them.

The High-Level Perception (HLP) theory of analogy (Chalmers et al., 1992; Mitchell, 1993) instead construes analogy as a function of tightly-interacting perceptual and reasoning processes, positing that the creation of stimulus representations and the alignment of those representations are mutually dependent. For example, when making an analogy between the sea and acoustics, we might represent certain perceptual features (the fact that the sea appears to be moving) and ignore others (the fact that the sea looks blue), because the particular comparison that we make depends on location and direction, and not on colour.

In this work we aim to induce flexible analogy making in neural networks by drawing inspiration from both SMT and HLP. The perceptual and domain-comparison components of our models are connected and jointly optimised end-to-end, which, as posited by HLP, reflects a high degree of interaction between perception and domain comparison. On the other hand, the key insight of this paper, LABC, is directly motivated by SMT. We find that LABC greatly enhances the ability of our networks to resolve analogies in a generalisable way by encouraging them to compare inputs at the more abstract level of relations rather than the less abstract level of attributes. LABC organizes the training data such that the inference and mapping of relational structure is essential for good performance. This means that problems cannot be resolved by considering mere similarity of attributes, or even less appropriately, via spurious surface-level statistics or memorization.

3. VISUAL ANALOGY PROBLEMS:

Our first experiments involve greyscale visual scenes similar to those previously applied to test both human reasoning ability (Raven, 1983; Geary et al., 2000) and reasoning in machine learning models (Bongard, 1967; Fleuret et al., 2011; Barrett et al., 2018) . Each scene is composed of a source sequence, consisting of three panels (distinct images), a target sequence, consisting of two panels, and four candidate answer panels (Fig. 1 ). In the source sequence a relation r is instantiated, where r is one of four possible relations from the set R = { XOR , OR , AND , Progression }. Models must then consider the two panels in the target sequence, together with the four candidate answer panels, to determine which answer panel best completes the target sequence -by analogy with the source sequence -so that r is also instantiated (Fig. 2 ). 

LABC raining:

Figure 2 : In LABC, the multiple choice candidates are all semantically plausible, in that they are all consistent completions of the target domain using some relation. Only the correct answer uses the same relation as the source domain, so, the only way to solve the problem is to use analogical reasoning. In contrast, perceptually plausible incorrect candidates are consistent with the target domain attributes but not the relations and all other possible candidates are inconsistent with the target domain relation and attributes.

The notion of domain is critical to analogy.

In our visual analogy task, a relation can be instantiated in one of seven different domains: line type , line colour , shape type , shape colour , shape size , shape quantity and shape position (see Fig. 1 and Appendix Fig. 7 for examples). Within a panel of a given domain, the attributes present in the scene (such as the colour of the shapes or the positions of the lines) can take one of 10 possible values. A question in the dataset is therefore defined by a relation r, a domain d s on which r is instantiated in the source sequence, a set of values for the source-domain

v s 1 • • • v s 3 , a target domain d t , values for the target-domain v t 1 • • • v t 3 , the po- sition k ∈ {1 • • • 4}

of the correct answer among the answer candidate panels and whatever is instantiated in the three incorrect candidate answer panels c i , i = k. Note, however, that the values of certain domain attributes that are not relevant to a given question (such as the colour of shapes in the shape quantity domain) still have to be selected, and can vary freely. Thus, despite the small number of latent factors involved, the space of possible questions is of the order of ten million.

The interplay between relations, domains and values makes it possible to construct questions that require increasing degrees of abstraction and analogymaking. The simplest case involves a relation r, a domain d s == d t , and values v s i == v t i that are common to both source and target sequences (Fig 3a) . To solve such a question a model must identify a candidate answer panel that results in a copy of the source sequence in the target sequence. This does not seem to require any meaningful understanding of r, nor any particular analogy-making. Somewhat greater abstraction and analogymaking is required for questions involving a single domain (d s == d t ), but different values in the source and target sequence v s i = v t i (Fig 3b ). In this case the model must learn that, in a given domain, the relation r can apply to a range of different values. Finally, in the full analogy questions considered in this study, the relation r can be instantiated on different domains in the source and target sequences (i.e. d t = d s ; Fig 3c ). These questions require a sensitivity to the idea that a single relation r can be applied in different (but related) ways to different domains of experience.1 

3.1. METHODS:

Our model consisted of a simple perceptual front-end -a convolutional neural network (CNN)which provided input for a recurrent neural network (RNN) by producing embeddings for each image panel independently. The RNN processed the source sequence embeddings, the target sequence embeddings, and a single candidate embedding, to produce a scalar score. Four such passes (one for each source-target-candidate set) produced four scalar scores, quantifying how the model evaluated the suitability of the particular candidate. Finally, a softmax was computed across the scores to select the model's 'answer'. Further model details are in appendix 7. contrasting, each answer choice is consistent with a relational structure in the target sequence. Only the correct answer choice is consistent with relations in both the source and target domains. This forces the network to consider the source sequence to infer the correct structure.

Learning Analogies By Contrasting (LABC) In the default setting of our data generator -the normal training regime -for a question involving source domain d s , target domain d t and relation r, the candidate answers can contain any (incorrect) values chosen at random from d t (Fig. 2 ). By selecting incorrect candidate answers from the same domain d t as the correct answer, we ensure that they are perceptually plausible, so that the problem cannot be solved trivially by matching the domain of the question to one of the answers. Even so, the baseline training regime may allow models to find perceptual correlations that allow it to arrive at the correct answer consistently over the training data. We can make this less likely by instead training the model to contrast abstract relational structurethe LABC regime -simply by ensuring that incorrect answers are both perceptually and semantically plausible (Fig. 2 ). More specifically, incorrect answers are selected from d t such that each answer Note that in all experiments reported below, we generated 600,000 training questions, 10,000 validation questions and test sets of 100,000 questions. These data will be published with the paper.

3.2. EXPERIMENT 1: NOVEL DOMAIN TRANSFER:

A key aspect of analogy-making is the process of comparing or aligning two domains. We can measure how well models acquire this ability by testing them on analogies involving unfamiliar source domain → target domain transfers. For each of the seven possible target domains d t we randomly selected a source domain d s = d t , yielding a test set of seven domain transfer pairs [d s → d t ]. Our models were then trained on questions involving one of the remaining 7 × 7 -7 = 42 domain transfer pairs. For a test question involving domains d s and d t , each model was therefore familiar with d s and d t but had not been trained to make an analogy from d s to d t .

We found that a network can indeed learn to apply relations by analogy involving novel domain transfers, but that this ability crucially relies on learning by contrasting. The effect is strong; for the most focused test questions involving semantically plausible (contrasting) candidate answers the model trained by contrasting achieves 83% accuracy (depending on the held-out domain), versus 58% for a model trained with randomly-chosen candidate answers.

3.3. EXPERIMENT 2: NOVEL TARGET DOMAIN:

Humans can use analogies to better understand comparatively unfamiliar domains, as in the Roman explanation of acoustics by analogy with the sea. To capture this scenario, we held out two domains ( line type and shape colour , chosen at random) from the model's training data, and ensured In the case of interpolation, we found that a model trained with random candidates performs very poorly on the more challenging contrasting test questions (Fig 4c 45% vs 93% for LABC), which suggests that models trained in the normal regime overfit to a strategy that bears no resemblance to human-like analogical reasoning. We verified this hypothesis by running an analysis where we presented only the target domain sequence and candidate answers to the model. After a long period of training in the normal regime, this 'source-blind' model achieved 97% accuracy, which confirms that it indeed finds short-cut solutions that do not require analogical mapping. In contrast, the accuracy of the source-blind model in the LBAC condition converged at 32%.

We also found, somewhat surprisingly, that LBAC results in a (modest) improvement in how well models can extrapolate to novel input values (Fig 4c ); a model trained on questions with both contrasting and random candidate answers performs significantly better than the normal model on the test questions with contrasting candidate answers (62% vs. 43%), and mantains comparable performance on test questions with random candidate answers (45% vs. 44%).

3.3.2. EXPERIMENT 4: MODEL COMPARISON:

Finally, we explore the extent to which LABC improves generalistion for different various different network architectures, taking as a guide the models considered in Barrett et al. (2018) . We run the Novel Domain Transfer experiment with each of these models, trained using both LABC and in the normal training regime. We measure generalisation on a mixed set comprised equally of test questions with semantically-plausible incorrect answer candidates (matching LABC training) and those with merely perceptually-plausible incorrect answers (matching normal training). All models perform better at novel domain transfer when trained with LABC (Table 1 ). This confirms that our method does not depend on the use of a specific architecture. Further, the fact that LABC yields better performance than normal training on a balanced, mixed test set shows that it is the most effective way to train models in problem instances where the exact details of test questions may be unknown.

3.4. CONCLUSION:

These results demonstrate that LABC increases the ability of models to generalize beyond the distribution of their training data. This effect is observed for the prototypical analogical processes involving novel domain mappings and unfamiliar target domains (Experiments 1 & 2). Interestingly, it also results in moderate improvements to how well models extrapolate to perceptual input outside the range of their training experience (Experiment 3).

4. EMERGENT RELATIONAL REPRESENTATIONS:

Cluster LABC Normal Training inter-relation dist. 5.2 (± 2.2) 4.4 (± 2.0) inter-domain dist.

1.8 (± 1.8) 2.0 (± 1.8)

Table 2 : Mean (+SD) distance between clusters of RNN hidden state representations. LABC seems to promote greater distinctions between different abstract relations in the representation space of models. Since the maximum euclidean distance between two points in a 64 dimensional unit cube is 64 i 1 2 = 8, the distances that we observe between relational representations are close to the maximum possible.

To understand the mechanisms that support this generalisation we analysed neural activity in models trained with LABC compared to models that were not.

First, we took the RNN hidden state activity just prior to the input of the candidate panel. For a model trained via LABC, we found that these activities clustered according to relation type (e.g. progression ) more-so than domain (e.g., shape colour ) (Fig 5a , Table 2 ). In contrast, for models trained normally the relationbased clusters overlapped to a greater extent (Fig 5b , Table 2 ). Thus, LABC seems to encourage the model to represent relations more explicitly, which could in turn explain its capacity to generalise by analogy to novel domains. 

5. SYMBOLIC ANALOGY PROBLEMS:

Many of the most important studies of analogy in AI involve symbolic or numerical patterns (Hofstadter, 1996) , and various neural models of semantic cognition more generally also operate on discrete, feature-based representations of input concepts (Rogers & McClelland, 2004; Devereux et al., 2018) . To verify our findings in this setting, we implemented a symbolic analogy task based on feature-based stimuli. This more controlled domain allows to show that the construction of appropriate incorrect answer candidates can be learned in a proposal model that is trained jointly with Published as a conference paper at ICLR 2019 a model that learns to contrast the candidates, widening the potential applications of LABC to task settings where we lack a clear understanding of the underlying abstract relational structure (Sec 5.3).

In our task, inputs are D-dimensional vectors v of discrete, integer-valued features (analogous to semantic properties such as furry or carnivorous in previous work). Across any set V : v ∈ V of stimuli, each feature dimension then corresponds to a domain (the domains of skin-type or dietary habits in the present example). To simulate a space of abstract relational structures on these domains, we simply take a set F whose elements f are common mathematical functions operating on sets: We study generalization in this task by restricting the particular ordered domain mappings d s → d t that are observed in the training and test set; for example, aligning structure from domain 3 to domain 1 may be required in the test set, but may have never been seen before in the training set. While we withhold particular alignments (e.g., 3 → 1), we ensure that all dimensions are aligned 'out-of' (3 →) and 'into' (→ 1) at least once in the training set. Note that this setup is directly analogous to the 'Novel Domain Transfer' experiment in the visual analogy problems (Sec. 3.2).

The high level model structure was similar to that of the previous experiments: candidates and their context were processed independently to produce scores, which we put through a softmax and trained with a cross-entropy loss. See appendix 7.2 for further details.Producing appropriate candidate answers C to train by contrasting abstract relational structures (LABC) is straightforward.

For a problem involving function f , we sample functions f ∼ F \ {f } at random and populate C with the c f , where c f = f (V t ). In other words, c f adheres to some relational structure, but just not the structure apparent in the source set. Thus, to determine which candidate is correct, the model necessarily has to first infer the particular relational structure of the source set. Because the implementation of this training regime requires knowledge of the underlying structure of the data (i.e. the space of all functions), we refer to this condition as LABC-explicit SMT.

We replicated the main findings of the visual analogy task (Experiment 1 S3.2): models trained without LABC performed at just above chance, whereas models trained with LABC achieved accuracies of just under 90%. Note that we tested models with candidates generated using the functions in F not equal to the function used to define the relation in the source set. If instead we were to generate random vectors as candidates, the models could simply learn and embed all possible f ∈ F into their weights, and choose the correct answer by process of elimination; any candidate that does not satisfy f (V t ) for any f is necessarily an incorrect candidate. This method does not require any analogical reasoning, since the model can outright ignore the relation instantiated in the source set, which is precisely a type of back-door solution characteristic of neural network approaches. These results are intuitive at first glance -a model that cannot use back-door solutions, and instead is required to be more discerning at training time will perform better at test time. However, this intuition is less obvious when testing demands novel domain transfer. In other words, it is not obvious that a model Figure 6 : Structure alignment and mapping on symbolic, numeric data. In this task a particular structure is implemented as a set-function, which in the depicted example is f = min. For example, the "answer" a 1 could denote the minimum size from the sizes in the source symbolic vector set.

The model must then evaluate the minimum for the "aligned" domain, which in the depicted case is intensity. In this depicted example the candidate does not adhere to this structure, so it would be an incorrect candidate. The correct candidate would look like the answer vector to the right of the image.

that has learned to discern the various functions in F would necessarily be able to flexibly apply the functions in ways never-before-seen, as is demanded in the test set.

5.3. EXPERIMENT 2: NOVEL DOMAIN TRANSFER WITH AUTOMATIC LABC METHODS:

We explored ways to replicate the results of Experiment 1 without hand-crafting the candidate answers. For our first method, (LABC-topk) we uniformly sampled integer values for the c ∈ C from within some range (0, 64) rather than computing them as f (V t ) for some f ∈ F . As mentioned previously, such a method should encourage back-door memorization based solutions, since for most candidates, c = f (V t ) for any f ∈ F . To counter this, we randomly generated a set of candidates, performed a forward pass with each, selected the top-k scalar scores produced by the model, and backpropagated gradients only through these k candidates. Intuitively, this forces the model to train on only those candidates that are maximally confusing. Thus, we rely on random generation to chose our contrasting candidates, and rely on the model itself to sub-select them from within the randomly generated pool. This method improves performance from chance (25%) to approximately 77%. It is possible that this top-k method simply exploited random sampling to stumble on the candidates that would have otherwise been hand-crafted. Indeed, for more difficult problems (involving real images, for example) a random generator may not produce anything resembling data from the underlying distribution, making this method less suitable. We thus replicated the top-k experiment, but actively excluded randomly generated candidates that satisfied c = f (V t ) for some f ∈ F . Performance fell to 43%, confirming this intuition, but interestingly, still greatling improving baseline performance.

Finally, we considered a method for generating candidates that did not depend on random generation (LABC-adversarial), but instead exploited a generator model. The generator was identical to the model used to solve the task except its input consisted only of the target set V t , its output was a proposed candidate vector c. This candidate vector was then passed to the original model, which solved the analogy problem as before. The generator model was trained to maximize the score given by the analogy model; i.e. it was trained to produce maximally confusing candidates. The overall objective therefore resembled a two-player minimax game (Goodfellow et al., 2014) :

min θ max φ L(f θ (S 1 , a 1 , S 2 , a 2 , g φ (S 2 )))

where f θ is the analogy model and g φ is the candidate proposal model. Using this method to propose candidates improved the model's test performance from chance (25%) to approximately 62%.

These latter experiments show interesting links between LABC, GANs (Goodfellow et al., 2014) , and possibly self-play (Silver et al., 2017) . Indeed, the latter two approaches may be seen as automated methods that can approximate LABC. For example, in self-play, agents continuously challenge their opponents by proposing maximally challenging data. In the case of analogy, maximally challenging candidates may be those that are semantically-plausible rather than simply perceptually plausible.

To our knowledge no prior work has demonstrated the effects of adversarial training regimes on out-of-distribution generalization in a controlled setting like the present context.

6. DISCUSSION:

Our experiments show that simple neural networks can learn to make analogies with visual and symbolic inputs, but this is critically contingent on the way in which they are trained; during training, the correct answers should be contrasted with alternative incorrect answers that are plausible at the level of relations rather than simple perceptual attributes. This is consistent with the SMT of human analogy-making, which highlights the importance of inter-domain comparison at the level of abstract relational structures. At the same time, in the visual analogy domain, our model reflects the idea of analogy as closely intertwined with perception itself. We find that models that are trained by LABC to reason better by analogy are, perhaps surprisingly, also better able to extrapolate to a wider range of input values. Thus, making better analogies seems connected to the ability of models to perceive and represent their raw experience.

Recent literature has questioned whether neural networks can generalise in systematic ways to data drawn from outside the training distribution (Lake & Baroni, 2018) . Our results show that neural networks are not fundamentally limited in this respect. Rather, the capacity needs to be coaxed out through careful learning. The data with which these networks learn, and the manner in which they learn it, are of paramount importance. Such a lesson is not new; indeed, the task of one-shot learning was thought to be difficult, if not impossible to perform using neural networks, but was nonetheless "solved" using appropriate training objectives, models, and optimization innovations (e.g., (Santoro et al., 2016; Finn et al., 2017) ). The insights presented here may guide promising, general purpose approaches to obtain similar successes in flexible, generalisable abstract reasoning. 2015) learn an analogy model on top of pre-trained visual embeddings of geometric figures and rendered graphics, while Mikolov et al. (2013) show how analogies can be made via non-parametric operations on vector-spaces of text-based word representations. While the input to our visual analogy model is less naturalistic than these latter cases, this permits clear control over the semantics of training or test data when designing and evaluating hypotheses. Our study is nonetheless the only that we are aware to demonstrates such flexible, generalisable analogy making in neural networks learning end-to-end from raw perception. It is therefore a proof of principle that even very basic neural networks have the potential for strong analogical reasoning and generalization.

As discussed in Sec. 5.3, in many machine-learning contexts it may not be possible to know exactly what a 'good quality' negative example looks like. The experiments there show that, in such cases, we might still achieve notable improvements in generalization via methods that learn to play the role of teacher by presenting alternatives to the main (student) model, as per Shafto et al. (2014) . This underlines the fact that, for established learning algorithms involving negative examples such as (noise) contrastive estimation (Smith & Eisner, 2005; Gutmann & Hyvärinen, 2010) or negative sampling (Mikolov et al., 2013) , the way in which negative examples are selected can be critical3 . It may also help to explain the power of methods like self-play (Silver et al., 2016) , in which a model is encouraged to continually challenge itself by posing increasingly difficult learning challenges.

Analogies as the functions of the mind To check whether a plate is on a table we can look at the space above the table, but to find out whether a picture is on a wall or a person is on a train, the equivalent check would fail. A single on function operating in the same way on all input domains could not explain these entirely divergent outcomes of function evaluation. On the other hand, it seems implausible that our cognitive system encodes the knowledge underpinning these apparently distinct applications of the on relation in entirely independent representations. The findings of this work argue instead for a different perspective; that a single concept of on is indeed exploited in each of the three cases, but that its meaning and representation is sufficiently abstract to permit flexible interaction with, and context-dependent adaptation to, each particular domain of application. If we equate this process with analogy-making, then analogies are something like the functions of the mind. We believe that greater focus on analogy may be critical for replicating human-like cognitive processes, and ultimately human-like intelligent behaviour, in machines. It may now be time to revisit the insights from past waves of AI research on analogy, while bringing to bear the tools, perspectives and computing power of the present day.

The CNN was 4-layers deep, with 32 kernels per layer, each of size 3×3 with a stride of 2. Thus, each layer downsampled the image by half. Each panel in a question was 80 × 80 pixels, and greyscale.

The panels were presented one at a time to the CNN to produce 9 total embeddings (3 for the source sequence, 2 for the target sequence, and 4 for each candidate). We then used these embeddings to compile 4 distinct inputs for the RNN. Each input was comprised of the source sequence embeddings, the target sequence embeddings, and a single candidate embedding, for a total of 6 embeddings per RNN-input sequence. We passed these independently to the RNN (with 64 hidden units), whose final output was then passed through a linear layer to produce a single scalar. 4 such passes (one for each source-target-candidate sequence) produced 4 scalar scores, denoting the model's evaluation of the suitability of the particular candidate for the analogy problem. Finally, a softmax was computed across the scores to select the model's "answer". We used a cross entropy loss function and the Adam optimizer with a learning rate of 1e -4 .

A given input consisted of a set of vectors 16-dimensional vectors. This set included 8 vectors comprising S 1 , one vector d 1 , 8 vectors comprising S 2 , and 8 vectors comprising the set of candidate vectors C. Vectors were given a single digit binary variable tag to denote whether they were members of the source or target set (augmenting their size to 17-dimensions).

We note that the entity vectors have 0's in their unused dimensions. While this may make the problem easier, this experiment was designed to explicitly test domain-transfer generalization, moreso than an ability to discern the domains that need to be considered, by stripping away any difficulties in perception (i.e., in identifying the relevant domains), and seeing if the effect of LABC persists. Thus, at test time the model should have an easy time identifying the relevant dimensions, but it will never have seen the particular transfer from, say, dimension i to dimension j. So, even though it may have an easy time identifying and processing each dimension i and j, it may be incapable (without LABC) of integrating the information processed from each of these dimensions.

We employed a parallel processing architecture, similar to the visual analogy experiments, with a Relation Network (128 unit, 3 layer MLP with ReLU non-linearities for the g θ function and a similar 2-layer MLP for the f φ function) replacing the RNN core. Thus, a single model processed (S 1 , d 1 , S 2 , c n ), with c n being a different candidate vector from C for each parallel pass. The model's output was a single scalar denoting the score assigned to the particular candidate c n -these scores were then passed through a softmax, and training proceeded using a cross entropy loss function. We used batch sizes of 32 and the Adam optimizer with a learning rate of 3e -4 . 

8. SUPPLEMENTARY RESULTS:



LABC:



9. MODEL COMPARISON DETAILS:

Our application of a ResNet-50 processes all nine panels simultaneously (five analogy question panels along with the four multiple choice candidates) as a set of input channels. The Parallel ResNet-50 processes six panels simultaneously as input channels (five analogy question panels along with one multiple choice candidate) to produce a score. Then, similar to the RNN model described above, the candidate with the highest score is chosen by the model. The parallel relation network model also processes six panels simultaneously, using a convnet to obtain panel embeddings and using a relation network (Santoro et al., 2017) for computing a score. For full model architecture details, see the appendix.

Interestingly, the model with strongest generalisation is the parallel relation network, with a particularly high accuracy of 95% on the held out domain-transfer test set. This model was tested on a mixture of multiple choice candidates (that included semantically plausible and perceptually plausible candidates), indicating that models trained with LABC do not over-specialize to problem settings where only semantically plausible candidates are available. We also observe that during normal training, test set performance can oscillate between good solutions and poor solutions, indicated by the high standard deviation in the test set accuracy. These results imply that there are multiple model configurations that have good performance on the training set, but that only some of these configurations have the desired generalisation behaviour on the test set. LABC encourages a model to learn the configurations that generalise at the most abstract semantically-meaningful level, as desired.

We also note that the fact that a model trained in the normal regime performs marinally better than one trained using (normal+)LABC data on test questions involving perceptually-plausible candidates.

We believe this may be understood as a symptom of the strong ability of deep learning models to specialize to the exact nature of the problems on which they are trained. The model comparison experiments demonstrate that this negligible but undesirable specialization effect is outweighed by the greater benefits of training with LABC on test questions with semantically-plausible candidates (i.e. those that require a higher-level semantic interpretation of the problem). Training with LABC will therefore yield a much higher expected performance, for instance, in cases where the exact details of the test questions is not known.

10. FURTHER DISCUSSION AND RELATED WORK:

It is interesting to consider to what extent the effects reported in this work can transfer to a wider class of learning and reasoning problems beyond classical analogies. The importance of teaching concepts (to humans or models) by contrasting with negative examples is relatively established in both cognitive science (Shafto et al., 2014; Smith & Gentner, 2014) and educational research (Silver, 2010; Ali, 1981) . Our results underline the importance of this principle when training modern neural networks to replicate human-like cognitive processes and reasoning from raw perceptual input. In cases where expert understanding of potential data exists, for instance in the case of active learning with human interaction, it provides a recipe for achieving more robust representations leading to far greater powers of generalization. We should aspire to select as negative examples those examples that are plausible considering the most abstract principles that describe the data.

A further notable property of our trained networks is the fact they can resolve analogies (even those involving with unfamiliar input domains) in a single rollout (forward pass) of a recurrent network. This propensity for fast reasoning has an interesting parallel with the fast and instinctive way in which humans can execute visual analogical reasoning (Morrison et al., 2001; Qiu et al., 2008) .

10.1. DISTANCE METRIC APPROACHES:

LBC shares similarities with distance metric approaches such as the large-margin nearest neighbor classifier (LMNN) (Weinberger & Saul, 2009) , the triplet loss (Schroff et al., 2015) , and others. In these approaches the goal is to transform inputs such that the distance between input embeddings from the same class is small, while the distance between input embeddings from different classes is large. Given these improved embeddings, classification can proceed using off-the-shelf classification algorithms, such as k-nearest neighbors. We note that these approaches emphasize the form of the loss function and the quality of the resultant input embeddings on subsequent classification. However, the goal of LBC is not to induce better classification per se, as it is in these methods. Instead, the goal is to induce out-of-distribution generalisation by virtue of improved abstract understanding of the underlying problem. It is unclear, for example, whether the embeddings produced by LMNN or the triplet loss are naturally amenable to this kind of generalisation, and as far as we are aware, it has not beed tested. Nonetheless, LBC places a critical focus on the nature, or quality of the data comprising the incorrect classes, and is agnostic to the exact nature of the loss function. Thus, it is possible to use previous approaches (e.g., LMNN or triplet loss, etc.) in conjunction with LBC, which we do not explore. LBC also shares similarities to recent generative adversarial active learning approaches (Zhu & Bento, 2017) . However, these approaches do not explicitly point to the effects of the quality of incorrect samples to out-of-distribution generalisation, nor are we aware of any experiments that test abstract generalisation using networks trained with generative samples.

11. VISUAL ANALOGY EXAMPLES:



:

The visual analogy dataset can be downloaded from https://github.com/deepmind/ abstract-reasoning-matrices

It is important to note that that LABC as described here relies on our understanding of the underlying data-generating process; we demonstrate its application that does not require such understanding in Sec. 5.3.

See Lazaridou et al. (2015) for a further interesting example of this

Figures and Tables:

Figure 3: (a), (b), (c) Three types of visual reasoning questions. Each question requires a different degree of analogy making, with the question on the right demanding the most fluid and abstract application of the underlying relation. (d) Learning analogies by contrasting. When learning by contrasting, each answer choice is consistent with a relational structure in the target sequence.Only the correct answer choice is consistent with relations in both the source and target domains. This forces the network to consider the source sequence to infer the correct structure.

Figure4: Results of the three experiments in the visual analogy domain for a network that learns from random candidate answers, by contrasting abstract structures or both types of question interleaved. Bar heights depict the means across eight seeds in each condition; standard errors were < 0.01 for each condition (not shown -see the appendix Table4for the values)

Figure 5: LABC supports emergent relational representations. Principle component analysis (PCA) (right) and t-SNE analysis (left) of RNN hidden states. Each dot represents a (64-dimensional) state coloured according to the relation type and domain of the corresponding question.

Earlier work on analogical reasoning in AI and cognitive science employed constructed symbolic stimuli or pre-processed perceptual input (Carbonell (1981); Hummel & Holyoak (1997); Hofstadter (1996); Larkey & Love (2003) inter alia; see Gentner & Forbus (2011) for a full review). More recenty, Reed et al. (

Test performance of a selection of visual reasoning network architectures trained in the normal regime and with LABC on the Novel Domain Transfer experiment. The test questions include those with both semantically-plausible and merely perceptually-plausible incorrect answers. Mean (+SD) for 10 models in each condition with different random initialisations.that each test question involved one of these domains. To make sense of the test questions, a model must therefore (presumably) learn to represent the relations in the dataset in a sufficiently general way that this knowledge can be applied to completely novel domains. Any model that resolves such problems successfully must therefore exploit any (rudimentary) perceptual similarity between the test domain and the domains that were observed during training. For instance, the process of applying a relation in the shape colour domain may recruit similar feature detectors to those required when applying it to the line colour domain.

Table content: <table><tr><td/><td colspan="2">LABC training Normal training</td></tr><tr><td>RNN (this paper)</td><td>0.90 (0.01)</td><td>0.79 (0.09)</td></tr><tr><td>ResNet-50</td><td>0.82 (0.02)</td><td>0.77 (0.10)</td></tr><tr><td>Parallel ResNet-50</td><td>0.89 (0.01)</td><td>0.75 (0.14)</td></tr><tr><td>Parallel Relation Net (WReN)</td><td>0.95 (0.017)</td><td>0.72 (0.21)</td></tr></table>

MAX , ARGMIN , ARGMAX , SUM and RANGE . Abstract relations in this context can be stated, for instance, asSUM(1, 2, 3; 6)  . Given a such a function f , a set V of stimulus vectors, and a random choice of domain d, we can compute a (D-dimensional) answer vector a[V,d,f ]  as the result of applying f to d on V (i.e. executing f on the d-th dimension of each v ∈ V ). It is now simple to randomly construct an analogy question in this setting. We select a function f , source d s and target d t domains at random, randomly generate source V s and target V t stimuli. We then apply f to V s on d s and to V t on d t to generate the source and target domain solutions, a [Vs,ds,f ] and a [Vt,dt,f ] , respectively. The inputs V s , d s , a [Vs,ds,f ] , V t and v t are passed to the model, together with a d × k matrix of alternative answer choices that includes a [Vt,dt,f ] (k is a fixed number of choices, four in the visual analogies presented previously). We then require the model to select which of these alternatives is the true completion of the analogy a[Vt,dt,f ]  . As in the visual analogy tasks, to resolve such a question the model must detect an abstract relationship in the input domain d s , that explains the connection between the source stimuli V s and the answer vector a[Vs,ds,f ]  . Once this achieved, the model must evaluate the function f that describes this relationship on the source domain d

Table content: <table/>

Test performance on our symbolic analogy in the four different training regimes.

Table content: <table><tr><td>Training method</td><td>Test accuracy</td></tr><tr><td>LABC; explicit SMT</td><td>0.89</td></tr><tr><td>LABC; top-k</td><td>0.77</td></tr><tr><td>LABC; adversarial</td><td>0.62</td></tr><tr><td>Random candidate answers</td><td>0.25</td></tr></table>

Results for the visual analogy task (RNN Model).

Table content: <table><tr><td/><td/><td/><td>Normal</td><td>Mix</td><td>LABC</td><td>Normal</td><td>Mix</td><td>LABC</td><td>Normal</td><td>Mix</td></tr><tr><td/><td/><td>(Train)</td><td>(Train)</td><td>(Train)</td><td>(Contrasting)</td><td>(Contrasting)</td><td>(Contrasting)</td><td>(Normal)</td><td>(Normal)</td><td>(Normal)</td></tr><tr><td>Extrapolation</td><td>Mean</td><td>0.94</td><td>0.95</td><td>0.94</td><td>0.62</td><td>0.43</td><td>0.56</td><td>0.45</td><td>0.44</td><td>0.39</td></tr><tr><td/><td>Std</td><td>0.005</td><td>0.007</td><td>0.005</td><td>0.02</td><td>0.009</td><td>0.012</td><td>0.01</td><td>0.01</td><td>0.04</td></tr><tr><td>Interpolation</td><td>Mean</td><td>0.94</td><td>0.97</td><td>0.94</td><td>0.93</td><td>0.45</td><td>0.89</td><td>0.65</td><td>0.89</td><td>0.87</td></tr><tr><td/><td>Std</td><td>0.003</td><td>0.003</td><td>0.003</td><td>0.004</td><td>0.004</td><td>0.008</td><td>0.01</td><td>0.006</td><td>0.01</td></tr><tr><td>Novel Domain Transfer</td><td>Mean</td><td>0.88</td><td>0.83</td><td>0.85</td><td>0.87</td><td>0.48</td><td>0.88</td><td>0.7</td><td>0.82</td><td>0.79</td></tr><tr><td/><td>Std</td><td>0.015</td><td>0.01</td><td>0.015</td><td>0.005</td><td>0.02</td><td>0.009</td><td>0.01</td><td>0.01</td><td>0.02</td></tr><tr><td colspan="2">Novel Domain (shape colour) Mean</td><td>0.87</td><td>0.84</td><td>0.85</td><td>0.78</td><td>0.50</td><td>0.80</td><td>0.51</td><td>0.61</td><td>0.58</td></tr><tr><td/><td>Std</td><td>0.007</td><td>0.008</td><td>0.007</td><td>0.004</td><td>0.02</td><td>0.006</td><td>0.02</td><td>0.01</td><td>0.02</td></tr><tr><td>Novel Domain (line type)</td><td>Mean</td><td>0.87</td><td>0.85</td><td>0.86</td><td>0.76</td><td>0.45</td><td>0.75</td><td>0.5</td><td>0.57</td><td>0.54</td></tr><tr><td/><td>Std</td><td>0.006</td><td>0.004</td><td>0.006</td><td>0.02</td><td>0.01</td><td>0.02</td><td>0.02</td><td>0.02</td><td>0.01</td></tr></table>

Peer review (iclr2019):
 The paper describes an approach to train neural networks for analogical reasoning tasks. General analogical reasoning is quite a significant milestone in Machine learning. Therefore, the paper tackles an extremely challenging problem. The paper does a good job of constructing various tasks to show that analogies can be learned in different scenarios which are complex analogy tasks. Specifically, visual analogy and symbolic analogies are considered. The main idea is to choose training examples such that the model is forced to learn the relational structure rather than simply learn superficial features. One weakness is that we need to hand code the training examples to force it to have contrasting relational structure for different tasks. Is this realistic in different problems? That is maybe a limiting factor of this work. An automated method for generating such examples is given, but there is not too much detail on this (5.3). Maybe this needs to be expanded. Also, is the idea of LABC different from SMT. The novelty may be a bit weak in this aspect. If LABC can be described in a more general manner, it would help a reader not familiar with the other related work. Since the baseline comparison is with a very weak method (randomly chosen examples), it is hard to judge the impact of the proposed approach. In summary, I think the paper has nice ideas, particularly, if we can automatically generate examples using LABC. but maybe there is a need to work on better organizing the ideas, more general formulation of LABC and a more convincing experimental evaluation that includes a state-of-the-art method if available