[edit]
Adversarial Sets for Regularising Neural Link Predictors
Proceedings of the 33rd Conference on Uncertainty in Artificial Intelligence, PMLR R15:61-70, 2017.
Abstract
In adversarial training, a set of models learn together by pursuing competing goals, usu- ally defined on single data instances. How- ever, in relational learning and other non-i.i.d domains, goals can also be defined over sets of instances. For example, a link predictor for the IS-A relation needs to be consistent with the transitivity property: if IS-A(x1, x2) and IS-A(x2, x3) hold, IS-A(x1, x3) needs to hold as well. Here we use such assumptions for deriving an inconsistency loss, measuring the degree to which the model violates the assumptions on an adversarially-generated set of examples. The training objective is de- fined as a minimax problem, where an adver- sary finds the most offending adversarial ex- amples by maximising the inconsistency loss, and the model is trained by jointly minimis- ing a supervised loss and the inconsistency loss on the adversarial examples. This yields the first method that can use function-free Horn clauses (as in Datalog) to regularise any neu- ral link predictor, with complexity independent of the domain size. We show that for several link prediction models, the optimisation prob- lem faced by the adversary has efficient closed- form solutions. Experiments on link predic- tion benchmarks indicate that given suitable prior knowledge, our method can significantly improve neural link predictors on all relevant metrics.