Title: TOWARDS UNDERSTANDING THE EFFECT OF LOSS FUNCTION ON THE PERFORMANCE OF KNOWLEDGE GRAPH EMBEDDING Anonymous authors Paper under double-blind review

Abstract: Knowledge graphs (KGs) represent world's facts in structured forms. KG completion exploits the existing facts in a KG to discover new ones. Translation-based embedding model (TransE) is a prominent formulation to do KG completion. Despite the efficiency of TransE in memory and time, it is claimed that TransE suffers from several limitations in encoding relation patterns (such as symmetric, reflexive relations etc). To solve that, most of the attempts have circled around the revision of the TransE score function, resulting more complicated score functions (such as TransA/D/G/H/R etc). These attempts have totally disregarded effect of loss functions to this end. We show that loss functions are key factors in this regard and disregarding them results in conclusions which are inaccurate or even wrong. More concretely, we show that the claimed limitations are inaccurate, as the effect of the loss was ignored. In this regard, we pose theoretical investigations of the main limitations of TransE in the light of loss function. To the best of our knowledge, so far, this has not been comprehensively investigated. We show that by a proper selection of the loss function for training the TransE model, the main limitations are mitigated. That is achieved by setting upper-bound for the scores of positive samples, defining the region of truth (i.e. the region that a triple is considered positive by the model). Our theoretical proofs with experimental results fill the gap in understanding of the limitation of translation-based class of embedding models and confirm the importance of the selection of loss functions for training the models and on their performance.

1. INTRODUCTION:

Knowledge is considered as commonsense facts and other information accumulated from different sources. A Knowledge Graph (KG) is collection of facts and is usually represented as a set of triples (h, r, t) where h, t are entities and r is a relation, e.g. (iphone, hyponym, smartphone). Entities and relations are nodes and edges in the graph, respectively. As KGs are inherently incomplete, making prediction of missing links is a fundamental task in knowlege graph analyses. Among different approaches used for KG completion, KG Embedding (KGE) has recently received growing attentions. KGE embeds entities and relations as low dimensional vectors known as embeddings.

To measure the degree of plausibility of a triple, a scoring function is defined over the embeddings. TransE, Translation-based Embedding model, (Bordes et al., 2013) is one of the most widely used KGE models. The original assumption of TransE is to hold: h + r = t, for every positive triple (h, r, t) where h, r, t ∈ R d are embedding vectors of head (h), relation (r) and tail (t) respectively. TransE and its many variants like TransH (Wang et al., 2014) and TransR (Lin et al., 2015b) , underperform greatly compared to the current state-of-the-art models. That is reported to be due to the limitations of their scoring functions. For instance, (Wang et al., 2018) reports that TransE cannot encode a relation pattern which is neither reflexive nor irreflexive.

In most of these works the effect of the loss function is ignored and the provided proofs are based on the assumptions that are not fulfilled by the associated loss functions. For instance (Sun et al., 2019) proves that TransE is incapable of encoding symmetric relation. To this end the loss function must enforce the distance of h + rt to zero, but this is never fulfilled (or even approximated) by the employed loss function. Similarly, (Wang et al., 2018) reports that TransE cannot encode a relation pattern which is neither reflexive nor irreflexive and (Wang et al., 2014) adds that TransE cannot properly encode reflexive, one-to-many, many-to-one and many-to-many relations. However, as mentioned earlier, such reported limitations are not accurate and the problem is not fully investigated due to the effect of the loss function.

In this regards, although TransH, TransR and TransD (Wang et al., 2014; Lin et al., 2015b; Ji et al., 2015) addressed the reported problem of TransE in one-to-many, many-to-one, many-to-many and reflexive etc, they were misled by the assumption (enforcing h + rt to be zero) that was not fulfilled by the employed loss function. Considering the same assumption, (Kazemi & Poole, 2018) investigated three additional limitations of TransE, FTransE (Feng et al., 2016) , STransE (Nguyen et al., 2016), TransH and TransR models: (i) if the models encode a reflexive relation r, they automatically encode symmetric, (ii) if the models encode a reflexive relation r, they automatically encode transitive and, (iii) if entity e 1 has relation r with every entity in ∆ ∈ E and entity e 2 has relation r with one of entities in ∆, then e 2 must have the relation r with every entity in ∆.

Assuming that the loss function enforces the norm to be zero, the aforementioned works have investigated these limitations by focusing on the capability of scoring functions in encoding relation patterns. However, we prove that the selection of loss function affects the boundary of score functions; consequently, the selection of loss functions significantly affects the limitations. Therefore, the above mentioned theories corresponding to the limitations of translation-based embedding models in encoding relation patterns are inaccurate. We pose new theories about the limitations of TransX(X=H,D,R, etc) models considering the loss functions. To the best of our knowledge, it is the first time that the effect of loss function is investigated to prove theories corresponding to the limitations of translation-based models.

In a nutshell, the key contributions of this paper is as follows. (i) We show that different loss functions enforce different upper-bounds and lower-bounds for the scores of positive and negative samples respectively. This implies that existing theories corresponding the limitation of TransX models are inaccurate because the effect of loss function is ignored. We introduce new theories accordingly and prove that the proper selection of loss functions mitigates the main limitations. (ii) We reformulate the existing loss functions and their optimization problems as an standard constrained optimization problem. This makes perfectly clear that how each of the loss functions affect on the boundary of triples scores and consequently ability of relation pattern encoding. (iii) Using symmetric relation patterns, we obtain a proper upper-bound of positive triples score to enable encoding of symmetric patterns. (iv) We additionally investigate the theoretical capability of translation-based embedding model when translation is applied in the complex space (TransComplEx). We show that TransComplEx is a more powerful embedding model with fewer theoretical limitations in encoding different relation patterns such as symmetric while it is efficient in memory and time.

2. RELATED WORKS:

Most of the previous work have investigated the capability of translation-based class of embedding models considering solely the formulation of the score function. Accordingly, in this section, we review the score functions of TransE and some of its variants together with their capabilities. Then, in the next section the existing limitations of Translation-based embedding models emphasized in recent works are reviewed. These limitations will be reinvestigated in the light of score and loss functions in the section 4.

The score of TransE (Bordes et al., 2013) is defined as: f r (h, t) = h + rt . TransH (Wang et al., 2014) projects each entity (e) to the relation space (e ⊥ = ew r ew T r ). The score function is defined as f r (h, t) = h ⊥ + rt ⊥ . TransH can encode reflexive, one-to-many, many-to-one and many-to-many relations. However, recent theories (Kazemi & Poole, 2018) prove that encoding reflexive results in encoding the both symmetric and transitive which is undesired. TransR (Lin et al., 2015b) projects each entity (e) to the relation space by using a matrix provided for each relation (e ⊥ = eM r , M r ∈ R de×dr ). TransR uses the same scoring function as TransH. TransD (Ji et al., 2015) provides two vectors for each individual entities and relations (h, h p , r, r p , t, t p ). Head and tail entities are projected by using the following matrices: M rh = r T p h p + I m×n , M rt = r T p t p + I m×n . The score function of TransD is similar to the score function of TransH.

RotatE (Sun et al., 2019) rotates the head to the tail entity by using relation. RotatE embeds entities and relations in the Complex space. By inclusion of constraints on the norm of entity vectors, the model would be degenerated to TransE. The scoring function of RotatE is f r (h, t) = h • rt , where h, r, t ∈ C d , and • is element-wise product. RotatE obtains the state-of-the-art results using very big embedding dimension (1000) and a lot of negative samples (1000). TorusE (Ebisu & Ichise, 2018) fixes the problem of regularization in TransE by applying translation on a compact Lie group. The model has several variants including mapping from torus to Complex space. In this case, the model is regarded as a very special case of RotatE Sun et al. (2019) that applies rotation instead of translation in the target the Complex space. According to Sun et al. (2019) , TorusE is not defined on the entire Complex space. Therefore, it has less representation capacity. TorusE needs a very big embedding dimension (10000 as reported in Ebisu & Ichise (2018) ) which is a limitation.

3. THE MAIN LIMITATIONS OF TRANSLATION-BASED EMBEDDING MODELS:

We review six limitations of translation-based embedding models in encoding relation patterns (e.g. reflexive, symmetric) mentioned in the literature (Wang et al., 2014; Kazemi & Poole, 2018; Wang et al., 2018; Sun et al., 2019) .

Limitation L1. TransE cannot encode reflexive relations when the relation vector is non-zero (Wang et al., 2014) .

Limitation L2. TransE cannot encode a relation which is neither reflexive nor irreflexive. To see that, if TransE encodes a relation r, which is neither reflexive nor irreflexive we have h 1 + r = h 1 and h 2 + r = h 2 , resulting r = 0, r = 0 which is a contradiction (Wang et al., 2018) .

Limitation L3. TransE cannot properly encode symmetric relation when r = 0. To see that (Sun et al., 2019) , if r is symmetric, then we have: h + r = t and t + r = h. Therefore, r = 0 and so all entities appeared in head or tail parts of training triples will have the same embedding vectors.

The following limitations hold for TransE, FTransE, STransE, TransH and TransR (Feng et al., 2016; Nguyen et al., 2016; Kazemi & Poole, 2018) :

Limitation L4. If r is reflexive on ∆ ∈ E

, where E is the set of all entities in the KG, then r must also be symmetric.

Limitation L5. If r is reflexive on ∆ ∈ E, r must also be transitive.

Limitation L6.:

If entity e 1 has relation r with every entity in ∆ ∈ E and entity e 2 has relation r with one of entities in ∆, then e 2 must have the relation r with every entity in ∆.

4. OUR MODEL:

TransE and its variants underperform compared to other embedding models due to their limitations we iterated in Section 3. In this section, we reinvestigate the limitations. We show that the corresponding theoretical proofs are inaccurate because the effect of loss function is ignored. So we propose new theories and prove that each of the limitations of TransE are resolved by revising either the scoring function or the loss. In this regard, we consider several loss functions and their effects on the boundary of the TransE scoring function. For each of the loss functions, we pose theories corresponding to the limitations. We additionally investigate the limitations of TransE using each of the loss functions while translation is performed in the complex space and show that by this new approach the aforementioned limitations are lifted. Our new model, TransComplEx, with a proper selection of loss function addresses the above problems.

4.1. TRANSCOMPLEX: TRANSLATIONAL EMBEDDING MODEL IN THE COMPLEX SPACE:

TransComplEx translates head entity vector to the conjugate of the tail vector using relation in the complex space (Trouillon et al., 2016) . Assuming h, r, t ∈ C d be complex vectors of dimension d, the score function is defined as follows:

f r (h, t) = h + r -t

Advantages of TransComplEx: We highlight four advantages of using the above formulation. (i) Comparing to TransE and its variants, TransComplEx has less limitations in encoding different relation patterns. The theories and proofs are provided in the next part. (ii) Using conjugate of tail vector in the formulation enables the model to make difference between the role of an entity as subject or object. This cannot be properly captured by TransE and its variants. (iii) Given the example (A, Like, Juventus), (Juventus, hasP layer, C.Ronaldo), that C.Ronaldo plays for Juventus may affect the person A to like the team. This type of information cannot be properly captured by models such as CP decomposition (Hitchcock, 1927) where two independent vectors are provided (Kazemi & Poole, 2018) for Juventus (for subject and object). In contrast, our model uses same real and imaginary vectors for Juventus when it is used as subject or object. Therefore, TransCom-plEx can properly capture dependency between the two triples with the same entity used as subject and object. And finally, (iiii) ComplEx (Trouillon et al., 2016) has much more computational complexity comparing to TransComplEx because it needs to compute eight vector multiplications to obtain score of a triple while our model only needs to do four vector summation/subtractions. In the experiment section, we show that TransComplEx outperforms ComplEx on various dataset.

4.2. REINVESTIGATION OF THE LIMITATIONS OF TRANSLATION-BASED MODELS:

The aim of this part is to analyze the limitations of Translation-based embedding models (including TransE and TransComplEx) by considering the effect of both score and loss functions. Different loss functions provide different upper-bound and lower-bound for positive and negative triples scores, respectively. Therefore, the loss functions affect the limitations of the models to encode relation patterns. In this regard, the existing works consider a positive triple of (h, r, t) and a negative triple of (h , r, t ) to satisfy some assumptions in a score function. For instance in TransE where

f r (h, t) = h + r -t , it is expected that f r (h, t) = 0 and f r (h , t ) > 0.

Unfortunately, as we show later, this can not be fulfilled (or even approximated) by using the proposed loss functions (e.g. margin ranking loss and RotatE loss).

To investigate and address the limitations, we propose four conditions (Table -1 ) for taking a triple as positive or negative by the score function. This is done by defining upper-bound and lower-bound for the scores. We show that these conditions can be approximated by proper loss functions, and in this regards we propose four losses that each will handle one of the condition.

Table 1 : Region of truth and falsity.

Condition Positive Negative γ 1 , γ 2 ∈ R (a) f r (h, t) = γ 1 , f r (h , t ) ≥ γ 2 γ 1 = 0, γ 2 > 0 (b) f r (h, t) = γ 1 f r (h , t ) ≥ γ 2 γ 2 > γ 1 > 0 (c) f r (h, t) ≤ γ 1 f r (h , t ) ≥ γ 2 γ 2 > γ 1 > 0 (d) f r (h, t) ≤ γ 1(h,r,t) f r (h , t ) ≥ γ 2(h,r,t) γ 2(h,r,t) > γ 1(h,r,t) > 0

To better comprehend Table-1, we have visualized the conditions in Figure 1 . The condition (a) indicates a triple is positive if h + r = t holds. It means that the length of residual vector i.e. = h + rt, is zero. It is the most strict condition that expresses being positive. Authors in (Sun et al., 2019; Kazemi & Poole, 2018) consider this condition to prove their theories as well as limitation of TransE in encoding of symmetric relations. However, the employed loss cannot approximate (a), rather it fulfills (c), resulting the reported limitation to be void in that setting.

Condition (b) considers a triple to be positive if its residual vector lies on a hyper-sphere with radius γ 1 . It is less restrictive than (a) which considers a point to express being positive. The optimization problem that approximates the conditions (a) (γ 1 = 0) and (b) (γ 1 > 0) is as follows:

         min ξ h,t (h,r,t)∈S + ξ h,t 2 f r (h, t) = γ 1 , (h, r, t) ∈ S + f r (h , t ) ≥ γ 2 -ξ h,t , (h , r, t ) ∈ S - ξ h,t ≥ 0 (1)

where S + , S -are the set of positive and negative samples respectively. ξ h,t are slack variables to reduce the effect of noise in negative samples (Nayyeri et al., 2019) . 1 . The residual vector , (a) becomes 0, (b) lies on the border of a sphere with radius γ 1 , (c) lies inside of a sphere with radius γ 1 , and (d) (h1,r1,t1) lies inside of a sphere with radius γ (h1,r1,t1) . Bottom: The histogram of the scores of triples when TransE is trained on WordNet (WN18RR) using the losses of Equation 2(γ 1 = 0), 2 (γ 1 = 4), 4 (γ 1 = 4) and 5 (γ = 6) respectively. Each of the bottom figures is the approximation of the corresponding conditions (a) to (d). One loss function that approximates the conditions (a) and (b) is as follows. Please note that for case (a) we set γ 1 = 0 and for case (b) we set γ 1 > 0 in the formula.

L a|b = (h,r,t)∈S + λ 1 f r (h, t) -γ 1 + (h ,r,t )∈S - (h,r,t) λ 2 max(γ 2 -f r (h , t ), 0 .

(2)

Condition (c) considers a triple to be positive if its residual vector lies inside a hyper-sphere of radius γ 1 . The optimization problem that approximates the condition (c) is as follows (Nayyeri et al., 2019) :

         min ξ h,t (h,r,t)∈S + ξ h,t 2 f r (h, t) ≤ γ 1 , (h, r, t) ∈ S + f r (h , t ) ≥ γ 2 -ξ h,t , (h , r, t ) ∈ S - ξ h,t ≥ 0 (3)

The loss function that approximates the condition (c) is as follows Nayyeri et al. (2019) :

L c = (h,r,t)∈S + λ 1 max(f r (h, t) -γ 1 , 0) + (h ,r,t )∈S - (h,r,t) λ 2 max(γ 2 -f r (h , t ), 0) . (4) Remark:

The loss function which is defined in Zhou et al. (2017b) is slightly different from the loss in 4. The former slides the margin while the latter fixes the margin by inclusion of a lower-bound for the score of negative triples. Both losses put an upper-bound for scores of positive triples.

Apart from the loss 4, the RotatE loss Sun et al. (2019) also approximates the condition (c). The formulation of the RotatE loss is as follows:

L RotatE c = - (h,r,t)∈S + log σ(γ -f r (h, t)) + (h ,r,t )∈S - (h,r,t) log σ(f r (h , t ) -γ) .

Condition (d) is similar to (c), but provides different γ 1 , γ 2 for each triples. Using (d), there is not a unique region of truth for all positive triples, rather for each positive triple (h, r, t) and its corresponding negative of (h , r, t ) there are triple-specific region of truth and falsity. Margin ranking loss (Bordes et al., 2013) approximates (d). Defining [x] + = max(0, x), the loss is defined as:

EQUATION

To investigate the limitations, we must assume that the relation vectors is not null otherwise we will have the same embedding for head and tail which is undesirable. Considering the conditions (a) to (d), we investigate the limitations L1 to L6 and we prove that existing theories are just valid under (a), which is not fulfilled under the given loss. In this regard we have the following theorem. For complete proofs, please refer to the appendix of the paper. Proof: Proofs of (i) and (ii) are provided in the appendix. For (iii) we have:

Under (b), for TransE we have h + rt = γ 1 and t + rh = γ 1 . The necessity condition for encoding symmetric relation is h+r-t = t+r-h . This implies h cos(θ h,r ) = t cos(θ t,r ).

Let ht = u, by definition we have u + r = γ 1 , ur = γ 1 . Now let γ 1 = α r , we have:

EQUATION

Therefore we have:

u 2 +(1-α 2 ) r 2 = -( u 2 +(1-α 2 ) r 2

), which can be written as u 2 = (α 2 -1) r 2 . To avoid contradiction we must have α > 1. Once α > 1 we have cos(θ u,r ) = π/2. Therefore, TransE can encode symmetric relation with condition (b), when γ 1 = α r and α > 1. Symmetric: Assume that r is a symmetric relation. To encode it we should have f r (h, t) ≈ f r (t, h), therefore f r (h, t) -f r (t, h) = 0. According to the definition of score function of TransComplEx, we have the following algebraic formulae: R Sym := Re(h) -Re(t) = 0.

Using similar argument to symmetric, the following formulae are derived for transitive, composition, inverse and implication.

Equivalence: Let p, q be equivalence relations, therefore we should have f p (h, t) ≈ f q (h, t). We obtain, R Eq := pq = 0.

Implication: Let p → q, be the implication rule. We obtain R Imp := max(f p (h, t) -f q (h, t), 0) = 0.

Inverse: Let r ←→ r -1 be the inverse relation. We obtain R Inv := rr -1 .

Finally, the following optimization problem should be solved:

EQUATION

)

where θ is embedding parameters, L is one of the losses 2, 4 or 5 and R i is one of the derived formulae mentioned above.

5. EXPERIMENTS AND EVALUATIONS:

In this section, we evaluate performance of our model, TransComplEx, with different loss functions on the link prediction task. The aim of the task is to complete the triple (h, r, ?) or (?, r, t) by prediction of the missed entity h or t. Filtered Mean Rank (MR), Mean Reciprocal Rank (MRR) and Hit@10 are used for evaluations (Wang et al., 2017; Lin et al., 2015b) .

Dataset. We use two dataset extracted from Freebase (Bollacker et al., 2008) (i.e. FB15K (Bordes et al., 2013) and FB15K-237 (Toutanova & Chen, 2015) ) and two others extracted from Word-Net (Miller, 1995) (i.e. WN18 (Bordes et al., 2013) and WN18RR (Dettmers et al., 2018) ). FB15K and WN18 are earlier dataset which have been extensively used to compare performance of KGEs. FB15K-237 and WN18RR are two dataset which are supposed to be more challenging after removing inverse patterns from FB15K and WN18. Guo et al. (2018) --→ (a, N ationality, b). We drop the relation patterns with confidence level less than 0.8. Generally, we use 454 and 14 relation patterns for FB15K and WN18 respectively. We do grounding for symmetric and transitive relation patterns. Thanks to the formulation of score function, grounding is not needed for inverse, implication and equivalence.

Experimental Setup. We implement TransComplEx with the losses 2, 4 and 5 and TransE with the loss 4 in PyTorch. Adagrad is used as an optimizer. We generate 100 mini-batches in each iteration. The hyperparameter corresponding to the score function is embedding dimension d. We add slack variables to the losses 2 and 4 to have soft margin as in (Nayyeri et al., 2019) . The loss 4 is rewritten as follows Nayyeri et al. (2019) :

EQUATION

We set λ 1 and λ 2 to one and search for the hyperparameters γ 1 (γ 2 > γ 1 ) and λ 0 in the sets {0.1, 0.2, . . . , 2} and {0.01, 0.1, 1, 10, 100} respectively. Moreover, we generate α ∈ {1, 2, 5, 10} (Zhou et al., 2017a) , (Ebisu & Ichise, 2019) and (Akrami et al., 2018) respectively, while the rest are taken from original papers/code. Dashes: results could not be obtained.

Results. We trained relation pattern version of our model i.e. RPTransComplEx using the losses of 2, 4, 5. All variants of RPTransComplEx, except PRTransComplEx4, are generally performing better than models in CAT3 which also explicitly inject relation patterns. The exception, PRTransComplEx5, is due to the fact that it uses margin ranking loss which we already showed its restriction regarding condition (d). In this regard, please consider histogram (d) in Figure 1 .

As all limitations were mitigated by condition (c), we expected that RPTransComplEx4, which is associated to (c), should perform better than others. This is empirically approved as shown in 2. The question is that how our model behaves under the loss 4 while we additionally disregard injection of relation patterns. To this end, we considered TransComplEx4 which uses loss 4 but does not do any injection. Since the performances of RPTransComplEx4 and TransComplEx4 are very close, we conclude that the latter also learns the pattern over the existing triples quite efficiently. This is also confirmed by the convergence performance of them as shown in Figure 3 over the symmetric relation.

To have a fair comparison1 with the categories of models which disregard injection of relation patterns (CAT1/2) we used the TransComplEx4 version of our approach. As shown in Table-2, we can observe that TransComplEx4 performs better than other models in CAT1/2 considering MR and Hits@10 and performing very closely to the best models on MRR.

As discussed earlier, FB15K-237 and WN18RR are two more challenging dataset provided recently. Table 3 presents the comparisons of our models with those other models that their performance results were available on these two datasets. Similar to our previous discussion on Table-2, we observe that PRTransComplEx4 and TransComplEx4 are performing closely which means that the latter has learned the relation pattern over triples very well without any injection. On WN18RR, TorusE was performing better than TransComplEx4 due to big embedding dimension of 10,000. On FB15k-237, RPTransComplEx4 performed better that others on MRR and Hits@10.

In order to investigate the effect of grounding, we train RPTransComplEx4 in two settings: 1) RP-TransComplEx4(w grounding) is trained when the grounded patterns are injected, 2) RPTransCom-plEx4(w/o grounding) is trained when the relation patterns which are not grounded used. According to the Table 4 , the grounding does not affect the performance significantly. We conclude that the model properly learns the relation patterns even without injection.

Boosting techniques: There are several ways to improve the performance of embedding models: 1) designing a more sophisticated scoring function, 2) proper selection of loss function, 3) using more negative samples 4) using negative sampling techniques, 5) enriching dataset (e.g. adding reverse triples). Among the mentioned techniques, we focus on the first and second ones and avoid using other techniques. We keep the setting used in (Trouillon et al., 2016) to have a fair comparison.

Using other techniques can further improve the performance of every models including ours. For example, TransComplEx with embedding dimension 200 and 50 negative samples gets 52.2 for Hits@10. Further analyses of our models in a big setting (bigger embedding dimension and more negative samples) are provided in appendix. Still, loss 4 is performing better and we conclude that our theoretical framework is approved in empirical experiments.

6. CONCLUSION:

In this paper, we reinvestigated the main limitations of Translation-based embedding models from two aspects: score and loss. We showed that existing theories corresponding to the limitations of the models are inaccurate because the effect of loss functions has been ignored. Accordingly, we presented new theories about the limitations by consideration of the effect of score and loss functions.

We proposed TransComplEx, a new variant of TransE which is proven to be less limited comparing to the TransE. The model is trained by using various loss functions on standard dataset including FB15K, FB15K-237, WN18 and WN18RR. According to the experiments, TransComplEx with proper loss function significantly outperformed translation-based embedding models. Moreover, TransComplEx got competitive performance comparing to the state-of-the-art embedding models while it is more efficient in time and memory. The experimental results conformed the presented theories corresponding to the limitations.

A FURTHER EXPERIMENTS WITH A BIGGER SETTING:

In this section we compare TransE, TransComplEx and RotatE trained by using the losses 2 (condition (a),(b)) , 4 (condition (c)) and the RotatE loss (condition (c)). In contrast to our previous experiments, we use a bigger setting: For FB15K-237, we set the embedding dimension to 300 and the number of the negative samples to 256, and for WN18RR, we set the embedding dimension and the number of negative samples to 300 and 250 respectively. We additionally use adversarial negative sampling technique used in (Sun et al., 2019) for all models.

FB15K-237 WN18RR:

MR MRR Hits@10 MR MRR Hits@10 respectively. The similar notations are considered for TransComplEx and RotatE when they are trained by using different loss functions. The loss 2 with γ 1 = 0 approximates the condition (a). The loss 2 with γ 1 > 0 approximates the condition (b). The condition (c) can be approximated by using the loss 4 and the RotatE loss (i.e., L RotatE c ). However, the loss 4 provides a better separation for positive and the negative samples than the RotatE loss. According to the table 5, the loss 4 obtains a better performance than the other losses in each class of the studied models. It is consistent with our theories indicating that the condition (c) is less restrictive. Although we only investigated the main limitations of the translation-based class of embedding models, the theories can be generalized to different models including the RotatE model. From the table, we can see that the loss 4 improves the performance of RotatE. Regarding the table 5, the loss 2 (γ 1 = 0) gets the worst results. It confirms our theories that with the condition (a), most of the limitations are held. However, with the condition (c), the limitations no longer exist. There have not been any losses that approximate the condition (a). However, most of the theories corresponding to the main limitations of the translation-based class of embedding models have been proven using the condition (a) while the used loss didn't approximate the condition. Therefore, the theories and experimental justifications have not been accurate. 

B RELATION PATTERN CONVERGENCE ANALYSIS:



C PROOF OF THE THEOREM:

The proof of theorems are provided as follows:

Proof of the Theorem T1 1) Let r be a reflexive relation and condition a) holds. For TransE, we have

EQUATION

Therefore, the relation vector collapses to a null vector (r = 0). As a consequence of r = 0, embedding vectors of head and tail entities will be same which is undesired. Therefore, TransE cannot infer reflexive relation with r = 0.

For TransComplEx, we have h + rh = 0.

We have

EQUATION

Therefore, all entities will have same embedding vectors which is undesired.

2) Using condition (b), we have h + rt = γ 1 .

It gives r = γ 1 . Therefore, in order to infer reflexive relation, the length of the relation vector should be γ 1 . Consequently, TransE and TransComplEx can infer reflexive relation. The same procedure can be used for the conditions (c) and (d).

Proof of the theorem T2: i) Let the relation r be neither reflexive nor irreflexive and two triples (e 1 , r, e 1 ), (e 2 , r, e 2 ) be positive and negative respectively. Therefore the following inequalities hold:

EQUATION

)

Equation 12 is rewritten as follows:

EQUATION

For TransE in real space, Re(r) ≤ γ 1 and Re(r) ≥ γ 2 cannot be held simultaneously when γ 2 > γ 1 . Therefore, TransE in real space cannot encode a relation which is neither reflexive nor irreflexive. In contrast, TransE in complex space can encode the relation by proper assignment of imaginary parts of entities. Therefore, theoretically TransComplEx can infer a relation which is neither reflexive nor irreflexive.

Proof of the theorem T3: i), ii) Let r be a symmetric relation and a) holds. We have

EQUATION

Trivially, we have

EQUATION

For TransE in real space, there is 3) For TransE with condition b), there is

EQUATION

The necessity condition for encoding symmetric relation is h + rt = t + rh . This implies h cos(θ h,r ) = t cos(θ t,r ). Let h -t = u, by 17 we have u + r = γ 1 , ur = γ 1 .

Let γ 1 = α r . We have

EQUATION

Regarding 18, we have

u 2 + (1 -α 2 ) r 2 = -(|u 2 + (1 -α 2 ) r 2 ).

→ u 2 = (α 2 -1) r 2 .

To avoid contradiction, α ≥ 1. If α ≥ 1 we have cos(θ u,r ) = π/2. Therefore, TransE can encode symmetric pattern with condition b), if γ 1 = α r and α ≥ 1. From the proof of condition b), we conclude that TransE can encode symmetric patterns under conditions c) and d).

Proof of the theorem T4: i) The proof of the lemma with condition a) for TransE is mentioned in the paper Kazemi & Poole (2018) . For TransComplEx, the proof is trivial. ii) Now, we prove that the limitation L4 is not valid when b) holds.

Let condition b) holds and relation r be reflexive, we have e 1 + re 1 = γ 1 , e 2 + re 2 = γ 1 .

Let e 1 + re 2 = γ 1 . To violate the limitation L4, the triple (e 2 , r, e 1 ) should be negative i.e., e 2 + re 1 > γ 1 , → e 2 + re 1 2 > γ 2 1 , → e 2 2 + e 1 2 + r 2 + 2 < e 2 , r > -2 < e 2 , e 1 > -2 < e 1 , r > > γ 2 1 . Considering e 1 + re 2 = γ 1 , we have < e 2 , r > -< e 1 , r > > 0, → < e 2 -e 1 , r > > 0, → cos(θ (e2-e1),r ) > 0, Therefore, the limitation L4 is not valid i.e., if a relation r is reflexive, it may not be symmetric. TransE is special case of TransComplEx and also condition b) is special case of condition c). Therefore using conditions b), c) and d), the limitation L4 is not valid for TransE and TransComplEx. likelihood loss as standard optimization problem, showing the the loss put a boundary for the score functions.

We will consider the mentioned points as future work.

D OPTIMAL HYPER-PARAMETERS:

The following tables show the optimal configurations for our models included in the Table -2 and 3 

:

Accordingly, we ran the RotatE code in our setting (embedding dimension

and 10 negative samples). The original paper used very big numbers of 1000 and 1000 respectively.

Figures and Tables:

Figure 1: Top: Visualization of truth region (positive) of a triple according to Table1. The residual vector , (a) becomes 0, (b) lies on the border of a sphere with radius γ 1 , (c) lies inside of a sphere with radius γ 1 , and (d) (h1,r1,t1) lies inside of a sphere with radius γ (h1,r1,t1) . Bottom: The histogram of the scores of triples when TransE is trained on WordNet (WN18RR) using the losses of Equation2(γ 1 = 0), 2 (γ 1 = 4), 4 (γ 1 = 4) and 5 (γ = 6) respectively. Each of the bottom figures is the approximation of the corresponding conditions (a) to (d).

Figure 2: Necessity condition for encoding symmetric relation: (a) when α < 1, the model cannot encode it. (b) when α = 1, the intersection of two hyperspheres is a point. u = 0 means embedding vectors of all entities should be same. Therefore, symmetric cannot be encoded. (c) when α > 1, symmetric can be encoded as there are more than one point in the intersection of two hyperspheres.

Theorem T1. (Addressing L1): TransE and TransComplEx cannot infer a reflexive relation pattern with a non-zero relation vector under (a). However, under (b-d), TransE and TransComplEx can infer reflexive pattern. Theorem T2. (Addressing L2): (i) TransComplEx can infer a relation which is neither reflexive nor irreflexive under (b-d). (ii) TransE cannot infer a relation which is neither reflexive nor irreflexive under (a-d). Theorem T3. (Addressing L3): (i) TransComplEx can infer symmetric relations under (a-d). (ii) TransE cannot infer symmetric relations under (a) with non-zero vector for relation. (iii) TransE can infer a symmetric relation under (b-d).

shows different conditions for encoding symmetric relation. Conditions (c-d) are directly resulted from (b), as it is subsumed by (c) and (d). That completes the proof. Theorem T4. (Addressing L4): For both TransE and TransComplEx, (i) Limitation L4 holds under (a). (ii) Limitation L4 is not valid under (b-d). Theorem T5. (Addressing L5): For both TransE and TransComplEx, (i) Limitation L5 holds under (a). (ii) Limitation L5 holds is not valid under (b-d). Theorem T6. (Addressing L6): For both TransE and TransComplEx, (i) Limitation L6 holds under (a). (ii) Limitation L6 is not valid under (b-d).4.3 ENCODING RELATION PATTERNS IN TRANSCOMPLEXMost of KGE models learn from triples. Recent work incorporates relation patterns such as transitive, symmetric on the top of triples to further improve performance of models. For instance, ComplEx-NNE+AER(Ding et al., 2018) encodes implication relation in the ComplEx model. RUGE(Guo et al., 2018) injects First Order Horn Clause rules in an embedding model. SimplE (Kazemi & Poole, 2018) captures symmetric, antisymmetric and inverse relations by weight tying in the model. Inspired by(Minervini et al., 2017) and considering the score function of TransCom-plEx, in this part, we derive formulae for equivalence, symmetric, inverse and implication to be used as regularization terms in the optimization problem. Therefore, TransComplEx incorporates different relation patterns to optimize the embeddings.

and Ding et al. (2018) extracted different relation patterns from FB15K and WN18 respectively. The relation patterns are provided by their confidence level, e.g. (a, BornIn, b) 0.9

Figure 3: The convergence of symmetric relation loss on FB15K

(a), (b), (c)(with the loss 4) and (c)(with the RotatE loss) with no injected relation patterns. Rows 5-8 TransComplEx trained using condition (a), (b), (c)(with the loss 4) and (c)(with the RotatE loss) with no injected relation patterns. Rows 9-10: RotatE trained using condition (c)(with the loss 4) and (c)(with the RotatE loss) with no injected relation patterns. Analysis of the results: Table 5 presents a comparison of TransE, TransComplEx and RotatE trained by different losses. TransE2(γ 1 = 0) is trained by using the loss 2 when γ 1 = 0. TransE2(γ 1 > 0) refers to the TransE model which is trained by using the loss 2 when γ 1 is a non-zero positive value. The TransE model which is trained by the losses 4 and the RotatE loss (i.e., L RotatE c ) are denoted by TransE4 and TransEL RotatE c

Figure4visualizes the convergence curve of the inverse loss with (RPTransComplEx4) and without (TransComplEx4) injection when the models are trained on WN18. Figure5shows the convergence of the TransComplEx4 model trained on FB15K with and without the relation pattern injection. The

Figure 4: The convergence of inverse relation loss on WN18

Re(h) + Re(r) = Re(t), Re(t) + Re(r) = Re(h), Therefore, Re(r) = 0. It means that TransE cannot infer symmetric relations with condition a). For TransComplEx, additionally we haveIm(h) + Im(r) = -Im(t), Im(t) + Im(r) = -Im(h),It concludes Im(h) + Im(r) + Im(t) = 0. Therefore, TransE in complex space with condition a) can infer symmetric relation. Because a) is an special case of b) and c), TransComplEx can infer symmetric relations in all conditions.

negative samples per each positive. The embedding dimension and learning rate are tuned from the sets {100, 200}, {0.0001, 0.0005, 0.001, 0.005, 0.01} respectively. All hyperparameters are adjusted by early stopping on validation set according to MRR. RPTransComplEx# denotes the TransComplEx model which is trained by the loss function # (2, 4, 5). RP indicates that relation patterns are injected during learning by regularizing the derived formulae (see 7). TransComplEx# refers to our model trained with the loss # without regularizing relation patterns formulae. The same notation is used for TransE#. The optimal configurations for RPTransComplEx are provided in the appendix.

Table content: <table><tr><td/><td/><td colspan="2">FB15k</td><td/><td>WN18</td><td/></tr><tr><td/><td colspan="6">MR MRR Hits@10 MR MRR Hits@10</td></tr><tr><td>TransE (Bordes et al., 2013)</td><td colspan="2">125 -</td><td>47.1</td><td colspan="2">251 -</td><td>89.2</td></tr><tr><td>TransH (bern) (Wang et al., 2014)*</td><td>87</td><td>-</td><td>64.4</td><td colspan="2">388 -</td><td>82.3</td></tr><tr><td>TransR (bern) (Lin et al., 2015b)*</td><td>77</td><td>-</td><td>68.7</td><td colspan="2">225 -</td><td>92.0</td></tr><tr><td>TransD (bern) (Ji et al., 2015)*</td><td>91</td><td>-</td><td>77.3</td><td colspan="2">212 -</td><td>92.2</td></tr><tr><td>TransE-RS (bern) (Zhou et al., 2017a)*</td><td>63</td><td>-</td><td>72.1</td><td colspan="2">371 -</td><td>93.7</td></tr><tr><td>TransH-RS (bern) (Zhou et al., 2017a)*</td><td>77</td><td>-</td><td>75.0</td><td colspan="2">357 -</td><td>94.5</td></tr><tr><td>TorusE (Ebisu &amp; Ichise, 2019)</td><td colspan="2">-73.3</td><td>83.2</td><td colspan="2">-94.7</td><td>95.4</td></tr><tr><td colspan="3">TorusE(with WNP) (Ebisu &amp; Ichise, 2019) -75.1</td><td>83.5</td><td colspan="2">-94.7</td><td>95.4</td></tr><tr><td>R-GCN (Schlichtkrull et al., 2018)+</td><td colspan="2">-65.1</td><td>82.5</td><td colspan="2">-81.4</td><td>95.5</td></tr><tr><td>ConvE (Dettmers et al., 2018)++</td><td colspan="2">51 68.9</td><td>85.1</td><td colspan="2">504 94.2</td><td>95.5</td></tr><tr><td>ComplEx (Trouillon et al., 2016)++</td><td colspan="2">106 67.5</td><td>82.6</td><td colspan="2">543 94.1</td><td>94.7</td></tr><tr><td>ANALOGY (Liu et al., 2017)++</td><td colspan="2">121 72.2</td><td>84.3</td><td colspan="2">-94.2</td><td>94.7</td></tr><tr><td>RotatE (Sun et al., 2019)</td><td colspan="2">48 69.0</td><td>86.1</td><td colspan="2">433 94.8</td><td>95.5</td></tr><tr><td>SimplE (Kazemi &amp; Poole, 2018)</td><td colspan="2">-72.7</td><td>83.8</td><td colspan="2">-94.2</td><td>94.7</td></tr><tr><td>SimplE+ (Fatemi et al., 2018)</td><td colspan="2">-72.5</td><td>84.1</td><td colspan="2">-93.7</td><td>93.9</td></tr><tr><td>PTransE (Lin et al., 2015a)</td><td>58</td><td>-</td><td>84.6</td><td>-</td><td>-</td><td>-</td></tr><tr><td>KALE (Guo et al., 2016)</td><td colspan="2">73 52.3</td><td>76.2</td><td colspan="2">241 53.2</td><td>94.4</td></tr><tr><td>RUGE (Guo et al., 2018)</td><td colspan="2">97 76.8</td><td>86.5</td><td>-</td><td>-</td><td>-</td></tr><tr><td colspan="3">ComplEx-NNE+AER (Ding et al., 2018) 116 80.3</td><td>87.4</td><td colspan="2">450 94.3</td><td>94.8</td></tr><tr><td>RPTransComplEx2</td><td colspan="2">38 70.5</td><td>88.3</td><td colspan="2">451 92.7</td><td>94.8</td></tr><tr><td>RPTransComplEx4</td><td colspan="2">38 72.4</td><td>88.8</td><td colspan="2">275 92.4</td><td>95.4</td></tr><tr><td>RPTransComplEx5</td><td colspan="2">59 61.7</td><td>82.2</td><td colspan="2">547 94.0</td><td>94.7</td></tr><tr><td>TransComplEx4</td><td colspan="2">38 68.2</td><td>87.5</td><td colspan="2">284 92.2</td><td>95.5</td></tr><tr><td>TransE4</td><td colspan="2">46 64.8</td><td>87.2</td><td colspan="2">703 68.7</td><td>94.5</td></tr></table>

Table 2 presents comparison of TransComplEx and its relation pattern encoded variants (RPTransComplEx) with three classes of embedding models on the most famous two datasets of FB15K and WN18. The first category (CAT1) of models consist of translation-based model (e.g.

Table content: <table><tr><td>TransX, TorusE). The second category (CAT2) are embedding models which are not translation-</td></tr><tr><td>based (e.g. ConvE, ComplEx, ANALOGY). The previous two categories (CAT1/2) can learn relation</td></tr><tr><td>patterns that are reflected in triples. In other words they learn patterns implicitly from existing</td></tr><tr><td>triples. The last category (CAT3) consist of models which can explicitly encode (inject) rules in</td></tr><tr><td>their training process (e.g. RUGE, ComplEx-NNE+AER, SimplE, SimplE+). The models in CAT3</td></tr><tr><td>are actually trained on relation patterns as well as triples.</td></tr></table>



Table content: <table><tr><td/><td colspan="2">FB15k-237</td><td/><td colspan="2">WN18RR</td></tr><tr><td/><td colspan="5">MR MRR Hits@10 MR MRR Hits@10</td></tr><tr><td colspan="2">TransE (Bordes et al., 2013) (our implementation) 205 27.1</td><td>45.2</td><td colspan="2">3806 19.5</td><td>45.1</td></tr><tr><td>DistMult (Bordes et al., 2013)+</td><td>-24.1</td><td>41.9</td><td>-</td><td>43.0</td><td>49.0</td></tr><tr><td>ComplEx (Trouillon et al., 2016)+</td><td>-24.0</td><td>41.9</td><td>-</td><td>44.0</td><td>51.0</td></tr><tr><td>R-GCN (Schlichtkrull et al., 2018)+</td><td>-24.8</td><td>41.7</td><td>-</td><td>-</td><td>-</td></tr><tr><td>ConvE (Dettmers et al., 2018)+</td><td>-31.6</td><td>49.1</td><td>-</td><td>46.0</td><td>48.0</td></tr><tr><td>TorusE (Ebisu &amp; Ichise, 2019)</td><td>-30.5</td><td>48.4</td><td>-</td><td>45.2</td><td>51.2</td></tr><tr><td>TorusE (with WNP) (Ebisu &amp; Ichise, 2019)</td><td>-30.7</td><td>48.5</td><td>-</td><td>46.0</td><td>53.4</td></tr><tr><td>RotatE (Sun et al., 2019)</td><td>211 31.1</td><td>49.4</td><td colspan="2">4789 47.3</td><td>54.9</td></tr><tr><td>RPTransComplEx2</td><td>210 27.7</td><td>46.4</td><td>-</td><td>-</td><td>-</td></tr><tr><td>RPTransComplEx4</td><td>226 31.9</td><td>49.5</td><td>-</td><td>-</td><td>-</td></tr><tr><td>RPTransComplEx5</td><td>216 25.3</td><td>43.8</td><td>-</td><td>-</td><td>-</td></tr><tr><td>TransComplEx4</td><td>223 31.7</td><td>49.3</td><td colspan="2">4081 38.9</td><td>49.8</td></tr><tr><td>TransE4</td><td>205 27.2</td><td>45.3</td><td colspan="2">3850 20.0</td><td>47.5</td></tr></table>

Link prediction results. Rows 1-8: basic models with no injected relation patterns. Results labeled with + are taken from(Ebisu & Ichise, 2019) while the rest are taken from original papers/code. Dashes: results could not be obtained.

Table content: <table/>

Link prediction results. RPTransComplEx4(w grounding) is trained on triples and the relation patterns which are grounded. RPTransComplEx4(w/o grounding) is trained on triples and the relation patterns which are not grounded.

Table content: <table><tr><td/><td colspan="2">FB15K</td></tr><tr><td/><td colspan="2">MR MRR Hits@10</td></tr><tr><td>RPTransComplEx4(w grounding)</td><td>38 72.4</td><td>88.8</td></tr><tr><td colspan="2">RPTransComplEx4(w/o grounding) 40 72.8</td><td>88.7</td></tr></table>

Link prediction results. Rows 1-4: TransE trained using condition

Table content: <table><tr><td>TransE2 (γ 1 = 0)</td><td>222 27.4</td><td>45.7</td><td>3014 19.3</td><td>47.4</td></tr><tr><td>TransE2 (γ 1 &gt; 0)</td><td>198 31.3</td><td>50.5</td><td>3942 21.4</td><td>50.3</td></tr><tr><td>TransE4</td><td>181 32.3</td><td>52.1</td><td>3451 23.5</td><td>53.9</td></tr><tr><td>TransEL RotatE c</td><td>179 32.5</td><td>51.9</td><td>3594 23.3</td><td>53.6</td></tr><tr><td colspan="2">TransComplEx2 (γ 1 = 0) 213 28.5</td><td>47.3</td><td>3014 31.2</td><td>49.5</td></tr><tr><td colspan="2">TransComplEx2 (γ 1 &gt; 0) 194 31.9</td><td>50.8</td><td>3942 41.3</td><td>50.8</td></tr><tr><td>TransComplEx4</td><td>177 32.8</td><td>52.1</td><td>3435 44.3</td><td>55.0</td></tr><tr><td>TransComplExL RotatE c</td><td>176 32.7</td><td>51.9</td><td>3537 44.2</td><td>54.7</td></tr><tr><td>RotatE4</td><td>194 33.0</td><td>52.0</td><td>3806 47.8</td><td>56.9</td></tr><tr><td>RotatEL RotatE c</td><td>196 33.0</td><td>51.8</td><td>3943 47.3</td><td>56.5</td></tr></table>

. Optimal Setting. The best hyper-parameter setting for RPTransComplEx2 on several dataset.

Table content: <table><tr><td/><td/><td colspan="2">RPTransComplEx2</td></tr><tr><td/><td colspan="2">FB15k FB15k-237</td><td>WN18</td></tr><tr><td>Embedding Dimension (d)</td><td>200</td><td>200</td><td>200</td></tr><tr><td>λ 0</td><td>100</td><td>100</td><td>100</td></tr><tr><td>γ 1</td><td>0.4</td><td>1.5</td><td>1.0</td></tr><tr><td>γ 2</td><td>0.5</td><td>2.0</td><td>2.0</td></tr><tr><td>Negative Sample (α)</td><td>10</td><td>10</td><td>10</td></tr><tr><td/><td/><td colspan="2">RPTransComplEx4</td></tr><tr><td/><td colspan="2">FB15k FB15k-237</td><td>WN18</td></tr><tr><td>Embedding Dimension (d)</td><td>200</td><td>200</td><td>200</td></tr><tr><td>λ 0</td><td>10</td><td>100</td><td>100</td></tr><tr><td>γ 1</td><td>0.4</td><td>1.5</td><td>0.6</td></tr><tr><td>γ 2</td><td>0.5</td><td>2.0</td><td>1.7</td></tr><tr><td>Negative Sample (α)</td><td>10</td><td>10</td><td>2.0</td></tr></table>

Optimal Setting. The best hyper-parameter setting for RPTransComplEx4 on several dataset.

Table content: <table><tr><td/><td/><td colspan="2">RPTransComplEx5</td></tr><tr><td/><td colspan="2">FB15k FB15k-237</td><td>WN18</td></tr><tr><td>Embedding Dimension (d)</td><td>200</td><td>200</td><td>200</td></tr><tr><td>γ</td><td>5.0</td><td>10</td><td>10</td></tr><tr><td>Negative Sample (α)</td><td>10</td><td>10</td><td>10</td></tr></table>

Optimal Setting. The best hyper-parameter setting for RPTransComplEx5 on several dataset.

Table content: <table><tr><td/><td/><td colspan="2">TransComplEx4</td><td/></tr><tr><td/><td colspan="2">FB15k FB15k-237</td><td colspan="2">WN18 WN18rr</td></tr><tr><td>Embedding Dimension (d)</td><td>200</td><td>200</td><td>200</td><td>200</td></tr><tr><td>λ 0</td><td>10</td><td>100</td><td>100</td><td>1</td></tr><tr><td>γ 1</td><td>0.4</td><td>1.5</td><td>0.6</td><td>1.6</td></tr><tr><td>γ 2</td><td>0.5</td><td>2.0</td><td>1.7</td><td>2.7</td></tr><tr><td>Negative Sample (α)</td><td>10</td><td>10</td><td>2.0</td><td>2.0</td></tr></table>

Optimal Setting. The best hyper-parameter setting for TransComplEx4 on several dataset.

Table content: <table><tr><td/><td/><td>TransE 4</td><td/><td/></tr><tr><td/><td colspan="2">FB15k FB15k-237</td><td colspan="2">WN18 WN18rr</td></tr><tr><td>Embedding Dimension (d)</td><td>200</td><td>200</td><td>200</td><td>200</td></tr><tr><td>λ 0</td><td>10</td><td>100</td><td>1</td><td>1</td></tr><tr><td>γ 1</td><td>0.4</td><td>0.4</td><td>1.0</td><td>0.6</td></tr><tr><td>γ 2</td><td>0.5</td><td>0.5</td><td>2.0</td><td>1.7</td></tr><tr><td>Negative Sample (α)</td><td>10</td><td>10</td><td>10</td><td>2</td></tr></table>

Optimal Setting. The best hyper-parameter setting for TransE4 on several dataset.

Table content: <table/>

Peer review (iclr2020):
Summary: This paper list several limitations of translational-based Knowledge Graph embedding methods, TransE which have been identified by prior works and have theoretically/empirically shown that all limitations can be addressed by altering the loss function and shifting to Complex domain. The authors propose four variants of loss function which address the limitations and propose a method, RPTransComplEx which utilizes their observations for outperforming several existing Knowledge Graph embedding methods. Overall, the proposed method is well motivated and experimental results have been found to be consistent with the theoretical analysis. Suggestions/Questions: 1. It would be great if hyperparameters listed in the “Experimental Setup” section could be presented in a table for better readability. 2. In Section 2, the authors have mentioned that RotatE obtains SOTA results using a very large embedding dimension (1000). However, it gives very similar performance even with smaller dimensional embedding (such as 200) with 1000 negative samples. In Section 5, RotatE results with 200 dimension and 10 negative samples are reported for a fair comparison. Wouldn’t it be better to instead increase the number of negative samples in RPTransComplEx instead of decreasing negative samples in RotatE? 3. In Table 3, it is not clear why authors have not reported their performance on the WN18RR dataset for their methods. Also, the reported performance of TransE in [1] is much better than what is reported in the paper. [1] Sun, Zhiqing, Zhi-Hong Deng, Jian-Yun Nie and Jian Tang. “RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Space.” ArXiv abs/1902.10197 (2019): n. pag.