• Home
  • Search
  • U2B: Scale-unbiased Representation Converter for Graph Classification with Imbalanced and Balanced Scale Distributions
  • https://doi.org/10.1609/aaai.v40i31.39820Copy DOI Icon

U2B: Scale-unbiased Representation Converter for Graph Classification with Imbalanced and Balanced Scale Distributions

  • Mar 14, 2026
  • Guanjun Wang +7 more
Show More
  • Abstract
  • Literature Map
  • Similar Papers
Abstract

Graph classification is a critical task in analyzing graph data, with applications across various domains. While graph neural networks (GNNs) have achieved remarkable results, their ability to generalize across graphs of varying scales remains a challenge. Conventional models often perform well on large-scale graphs but struggle with distributions that are skewed towards small scales. Conversely, models tailored to address scale imbalances frequently prioritize small-scale graphs, leading to diminished performance in more balanced scenarios. To overcome these limitations, we introduce a Unbalanced-Balanced Representation Converter (U2B), which exhibits no explicit bias toward graph scales. U2B employs a two-step workflow: a distillation phase to extract base features from both node-level and graph-level representations, followed by a refinement phase to generate unbiased representations for improved balance. In the distillation phase, a static constraint guides node-level adjustments, improving the representation of nodes in small graphs. Simultaneously, a dynamic constraint in the graph-level process mitigates biases toward features from large graphs. To ensure harmony between the representations, a consistency alignment loss is introduced, aligning node-level and graph-level features to create more cohesive and balanced graph representations. Extensive experiments on multiple datasets show that U2B achieves competitive performance.

Similar Papers
  • Conference Article

SALE-MLP: Structure Aware Latent Embeddings for GNN to Graph-free MLP Distillation

  • Sep 01, 2025
  • Harsh Pal +3
  • Conference Article
  • Citations84

Graph Neural Network (GNN) in Image and Video Understanding Using Deep Learning for Computer Vision Applications

  • Aug 04, 2021
  • P Pradhyumna +2
  • Conference Article
  • Citations39

H2MN

  • Aug 14, 2021
  • Zhen Zhang +6
  • PDF
  • Book Chapter
  • Citations2

A Principled Approach to Analyze Expressiveness and Accuracy of Graph Neural Networks

  • Jan 01, 2020
  • Asma Atamna +2
  • Research Article
  • Citations46

Multistructure Graph Classification Method With Attention-Based Pooling

  • Apr 01, 2023
  • IEEE Transactions on Computational Social Systems
  • Yuhua Xu +4
  • Dissertation

Efficient algorithms for subgraph counting and enumeration on large graphs

  • Jan 01, 2024
  • Kaixin Wang
  • Research Article
  • Citations5

GCARe: Mitigating Subgroup Unfairness in Graph Condensation through Adversarial Regularization

  • Aug 11, 2023
  • Applied Sciences
  • Runze Mao +2
  • Conference Article
  • Citations1

Efficient and Scalable Implicit Graph Neural Networks with Virtual Equilibrium

  • Dec 17, 2022
  • Qi Chen +6
  • Research Article
  • Citations9

Image and video analysis using graph neural network for Internet of Medical Things and computer vision applications

  • Mar 15, 2024
  • CAAI Transactions on Intelligence Technology
  • Amit Sharma +4
  • Conference Article
  • Citations49

Certified Robustness of Graph Neural Networks against Adversarial Structural Perturbation

  • Aug 14, 2021
  • Binghui Wang +3
  • Research Article

Abstract 5353: Graph representation learning of tumor topology from spatial imaging data

  • Apr 04, 2023
  • Cancer Research
  • Nicholas Ceglia +4
  • Research Article
  • Citations11

MolGraph: a Python package for the implementation of molecular graphs and graph neural networks with TensorFlow and Keras.

  • Dec 05, 2024
  • Journal of computer-aided molecular design
  • Alexander Kensert +2
  • Research Article
  • Citations2

Position-Sensing Graph Neural Networks: Proactively Learning Nodes Relative Positions.

  • Mar 01, 2025
  • IEEE transactions on neural networks and learning systems
  • Yiqun Zhang +6
  • Research Article
  • Citations172

EnGN: A High-Throughput and Energy-Efficient Accelerator for Large Graph Neural Networks

  • Aug 13, 2020
  • IEEE Transactions on Computers
  • Shengwen Liang +6
  • Research Article
  • Citations21

DeeperGCN: Training Deeper GCNs With Generalized Aggregation Functions.

  • Jan 01, 2023
  • IEEE Transactions on Pattern Analysis and Machine Intelligence
  • Guohao Li +4
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.