IEEE Access (Jan 2024)

Geometric Pooling: Maintaining More Representative Information

  • Hao Xu,
  • Jia Liu,
  • Yang Shen,
  • Kenan Lou,
  • Yanxia Bao,
  • Ruihua Zhang,
  • Shuyue Zhou,
  • Hongsen Zhao,
  • Xinmiao Zhu,
  • Shuai Wang

DOI
https://doi.org/10.1109/ACCESS.2024.3387703
Journal volume & issue
Vol. 12
pp. 54066 – 54072

Abstract

Read online

Graph Pooling technology plays an important role in graph node classification tasks. Sorting pooling technologies maintain large-value units for pooling graphs of varying sizes. However, by analyzing the statistical characteristic of activated units after pooling, we found that a large number of units dropped by sorting pooling are negative-value units that contain representative information and can contribute considerably to the final decision. To maintain more representative information, we proposed a novel pooling technology, called Geometric Pooling (GP), containing the unique node features with negative values by measuring the similarity of all node features. We reveal the effectiveness of GP from the entropy reduction view. The experiments were conducted on TUdatasets to show the effectiveness of GP. The results showed that the proposed GP outperforms the SOTA graph pooling technologies by $1\%~\sim ~5\%$ with fewer parameters.

Keywords