Research map / Graph learning / Graph self-supervised learning
DHAKR: Learning Deep Hierarchical Attention-Based Kernelized Representations for Graph Classification
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes DHAKR, an end-to-end differentiable hierarchical attention-based kernel model that learns soft assignments of WL-subtree and shortest-path substructure invariants into composite invariants for graph classification.
Problem
R-convolution graph kernels count isomorphic substructures while ignoring global structure, substructure importance, and the classifier is separate from kernel construction; GNNs pool node features and lose topology.
Approach
Learn a soft assignment matrix to hierarchically merge substructure invariants into composite invariants, apply feature-channel and kernel-channel attention to weight substructures and kernel layers, and feed the multi-level kernel embeddings to a classifier trained end-to-end.
Key results
- DHAKR(WL) reaches 89.87% on MUTAG and 68.76% on PTC(MR), outperforming all listed graph kernels on those datasets.
- DHAKR(SP) reaches 55.41% on Shock versus 37.88% for SPGK, and 76.11% on IMDB-B, the best among compared kernels and deep models.
- Against graph deep learning baselines, DHAKR variants obtain the best average ranks (DHAKR(WL) 1.4, DHAKR(SP) 2.4) in the reported table.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Remove kernel attention (ablation) | DHAKR(WL) · MUTAG | accuracy: 89.87±1.03 -> 89.62±0.79 | Hurt |
| Remove kernel attention (ablation) | DHAKR(WL) · PROTEINS | accuracy: 77.47±0.53 -> 77.05±0.33 | Hurt |
| Remove kernel attention (ablation) | DHAKR(WL) · IMDB-M | accuracy: 52.07±0.39 -> 51.87±0.60 | Hurt |
| Remove feature attention (ablation) | DHAKR(WL) · MUTAG | accuracy: 89.87±1.03 -> 86.81±1.29 | Hurt |
| Remove feature attention (ablation) | DHAKR(WL) · PROTEINS | accuracy: 77.47±0.53 -> 72.74±0.96 | Hurt |
| Remove feature attention (ablation) | DHAKR(WL) · IMDB-M | accuracy: 52.07±0.39 -> 49.39±0.79 | Hurt |
| Remove kernel attention (ablation) | DHAKR(SP) · MUTAG | accuracy: 88.44±0.64 -> 88.33±0.59 | Hurt |
| Remove kernel attention (ablation) | DHAKR(SP) · PROTEINS | accuracy: 76.09±0.97 -> 75.79±1.22 | Hurt |
| Remove kernel attention (ablation) | DHAKR(SP) · IMDB-M | accuracy: 53.21±0.54 -> 52.50±0.43 | Hurt |
| Remove feature attention (ablation) | DHAKR(SP) · MUTAG | accuracy: 88.44±0.64 -> 80.05±2.40 | Hurt |
| Remove feature attention (ablation) | DHAKR(SP) · PROTEINS | accuracy: 76.09±0.97 -> 74.55±0.51 | Hurt |
| Remove feature attention (ablation) | DHAKR(SP) · IMDB-M | accuracy: 53.21±0.54 -> 44.36±0.51 | Hurt |
Surprising findings
- Removing feature attention causes large drops (e.g., DHAKR(SP) on MUTAG falls from 88.44% to 80.05%), while removing kernel attention yields only small changes.
Limitations
- Results on PROTEINS are not best overall (DHAKR(SP) is beaten by DHAKR(WL)/others), and the model is limited to WL subtree and shortest-path substructures in the main text.
- Hyperparameter sensitivity: the regularization coefficient gamma has a notable effect, especially for DHAKR(WL).
- Complexity grows with N^2 due to kernel matrices over all sample graphs.
Experimental setup
Datasets: MUTAG PTC(MR) PROTEINS IMDB-BINARY IMDB-MULTI Shock
Benchmarks: MUTAG PTC(MR) PROTEINS IMDB-B IMDB-M Shock
Metrics: classification accuracy average rank
Baselines: GCGK RWGK SPGK WLSK CORE SP CORE WL DGCNN DCNN PATCHY-SAN DGK q-RWNN KerGNN
Builds on: Weisfeiler-Lehman subtree kernel shortest path graph kernel R-convolution kernels channel attention (SENet)
Closest papers by meaning
- Rethinking Kernel Methods for Node Representation Learning on Graphs (NeurIPS 2019 · similarity 0.91)
- Cluster-wise Graph Transformer with Dual-granularity Kernelized Attention (NeurIPS 2024 · similarity 0.87)
- Subgraph Invariant Learning Towards Large-Scale Graph Node Classification (AAAI 2025 · similarity 0.87)
- Exploring Consistency in Graph Representations: from Graph Kernels to Graph Neural Networks (NeurIPS 2024 · similarity 0.86)
- Graph Classification via Reference Distribution Learning: Theory and Practice (NeurIPS 2024 · similarity 0.86)
- Unsupervised Neighborhood Propagation Kernel Layers for Semi-supervised Node Classification (AAAI 2024 · similarity 0.86)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
