Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

On Valid Optimal Assignment Kernels and Applications to Graph Classification

About

The success of kernel methods has initiated the design of novel positive semidefinite functions, in particular for structured data. A leading design paradigm for this is the convolution kernel, which decomposes structured objects into their parts and sums over all pairs of parts. Assignment kernels, in contrast, are obtained from an optimal bijection between parts, which can provide a more valid notion of similarity. In general however, optimal assignments yield indefinite functions, which complicates their use in kernel methods. We characterize a class of base kernels used to compare parts that guarantees positive semidefinite optimal assignment kernels. These base kernels give rise to hierarchies from which the optimal assignment kernels are computed in linear time by histogram intersection. We apply these results by developing the Weisfeiler-Lehman optimal assignment kernel for graphs. It provides high classification accuracy on widely-used benchmark data sets improving over the original Weisfeiler-Lehman kernel.

Nils M. Kriege, Pierre-Louis Giscard, Richard C. Wilson• 2016

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy76.4
742
Graph ClassificationMUTAG
Accuracy84.5
697
Graph ClassificationNCI1
Accuracy86.1
460
Graph ClassificationCOLLAB
Accuracy80.7
329
Graph ClassificationIMDB-B
Accuracy72.7
322
Graph ClassificationENZYMES
Accuracy60.13
305
Graph ClassificationNCI109
Accuracy86.3
223
Graph ClassificationMutag (test)
Accuracy84.5
217
Graph ClassificationMUTAG (10-fold cross-validation)
Accuracy86
206
Graph ClassificationDD
Accuracy79.2
175
Showing 10 of 29 rows

Other info

Code

Follow for update