Share your thoughts, 1 month free Claude Pro on usSee more
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
1252
Graph ClassificationMUTAG
Accuracy84.5
1103
Graph ClassificationNCI1
Accuracy86.1
658
Graph ClassificationCOLLAB
Accuracy80.7
469
Graph ClassificationIMDB-B
Accuracy72.7
425
Graph ClassificationENZYMES
Accuracy60.13
328
Graph ClassificationDD
Accuracy79.2
300
Graph ClassificationNCI109
Accuracy86.3
267
Graph ClassificationPTC-MR
Accuracy63.6
244
Graph ClassificationMUTAG (10-fold cross-validation)
Accuracy86
227
Showing 10 of 29 rows

Other info

Code

Follow for update