TypeBandit: Type-Level Context Allocation and Reweighting for Effective Attribute Completion in Heterogeneous Graph Neural Networks

2026-04-29

AI Summary

This work addresses the challenge of missing node attributes in heterogeneous graphs, which is exacerbated by information asymmetry across different node types. The authors propose TypeBandit, the first approach to formally model type-dependent information asymmetry and introduce a type-level bandit sampling mechanism for lightweight, model-agnostic attribute completion under limited annotation budgets. Instead of relying on local neighborhood modeling, TypeBandit leverages a compact global context and integrates topology-aware initialization, joint representation learning, and hybrid pretraining that combines structural degree priors with feature propagation. The method is compatible with mainstream heterogeneous GNNs such as R-GCN, HetGNN, and HGT. Extensive experiments on DBLP, IMDB, ACM, and the large-scale OGBN-MAG dataset demonstrate substantial performance gains, confirming its efficiency, robustness, and superior semantic propagation capability.

Abstract

Heterogeneous graphs are widely used to model multi-relational systems, but missing node attributes remain a major bottleneck for downstream learning. In this paper, we identify and formalize type-dependent information asymmetry: the phenomenon that different node types provide substantially different levels of useful signal for attribute completion. Motivated by this observation, we propose TypeBandit, a lightweight, model-agnostic methodology for heterogeneous attribute completion. TypeBandit combines topology-aware initialization, type-level bandit sampling, and joint representation learning. It allocates a finite global sampling budget across node types, samples representative nodes within each type, and uses the resulting sampled type summaries as shared contextual signals during representation construction. By operating at the type level rather than over each target node's local neighborhood, TypeBandit keeps the adaptive state compact and practical for large heterogeneous graphs. A key advantage of TypeBandit is architectural flexibility. Rather than requiring a new heterogeneous graph neural network architecture, TypeBandit acts as a type-aware front end for representative heterogeneous GNN backbones, including R-GCN, HetGNN, HGT, and SimpleHGN. We further introduce a hybrid pretraining scheme that combines structural degree priors with feature propagation, yielding a more reliable initializer than degree-only pretraining. Under a fixed-split protocol on DBLP, IMDB, and ACM, TypeBandit provides dataset-dependent but practically meaningful gains. Additional ablation, stability, efficiency, semantic-propagation, and sampled OGBN-MAG experiments support TypeBandit as a practical strategy for heterogeneous attribute completion when type-specific information is unevenly distributed and sampling resources are limited.

Problem

• heterogeneous graph
• attribute completion
• information asymmetry
• node types
• missing attributes

Innovation

• type-level context allocation
• heterogeneous graph neural networks
• attribute completion
• bandit sampling
• model-agnostic