One Node One Model: Featuring the Missing-Half for Graph Clustering
Most existing graph clustering methods primarily focus on exploiting topological structure, often neglecting the "missing-half" node feature information, especially how these features can enhance clustering performance. This issue is further compounded by the challenges associated with high-dimensional features. Feature selection in graph clustering is particularly difficult because it requires simultaneously discovering clusters and identifying the relevant features for these clusters. To address this gap, we introduce a novel paradigm called "one node one model", which builds an exclusive model for each node and defines the node label as a combination of predictions for node groups. Specifically, the proposed "Feature Personalized Graph Clustering (FPGC)" method identifies cluster-relevant features for each node using a squeeze-and-excitation block, integrating these features into each model to form the final representations. Additionally, the concept of feature cross is developed as a data augmentation technique to learn low-order feature interactions. Extensive experimental results demonstrate that FPGC outperforms state-of-the-art clustering methods. Moreover, the plug-and-play nature of our method provides a versatile solution to enhance GNN-based models from the feature perspective.
Authors
- Xuanting Xie (ORCID: https://orcid.org/0000-0003-3801-5876)
- Erlin Pan (ORCID: https://orcid.org/0000-0001-8343-8822)
- Zhaochen Guo (ORCID: https://orcid.org/0000-0003-3135-6213)
- Zhao Kang (ORCID: https://orcid.org/0000-0003-4103-0954)
- Bingheng Li (ORCID: https://orcid.org/0009-0000-0950-9012)
- Wenyu Chen (ORCID: https://orcid.org/0000-0003-2610-3342)
Institutions
- University of Electronic Science and Technology of China (CN)
- Alibaba Group (United States) (US)
- Michigan State University (US)
Publication Details
- Journal
- Proceedings of the AAAI Conference on Artificial Intelligence
- Published
- 2025-04-11
- DOI
- https://doi.org/10.1609/aaai.v39i20.35473
- Citations
- 6
- Primary Topic
- Graph Theory and Algorithms
- Type
- article
- Field-Weighted Citation Impact
- 2.59
Funders
- National Natural Science Foundation of China