Residential College | false |
Status | 已發表Published |
Informative Nodes Mining for Class-Imbalanced Representation Learning | |
Zhou, Mengting; Gong, Zhiguo | |
2024-03 | |
Source Publication | IEEE Transactions on Computational Social Systems |
ISSN | 2329-924X |
Abstract | Online social networks often enforce a restricted web interface to allow third part users to access their data objects in one-by-one manner along links. The heavy cost of collecting data from them makes it very difficult to train a graph neural networks (GNNs) algorithm by using valuable data resources from those social network services. In this article, we endeavor to harness the potential of collected data in order to cultivate robust node representations, all while taking into account the prevalent class-imbalanced scenario. To this end, we propose a framework based on the reinforcement learning technique that can implement training while sampling. We employee deep Q-learning network (DQN) as the basic reinforcement learning framework which maintains an experience replay buffer to store experiences in order to improve efficiency and generalization ability of the algorithm. The proposed framework can guide the training process according to the label distribution of the currently collected training data. To maximize the utility of the collected data, we introduce two ways to guide the agent to learn more robust node representations under the natural class-imbalanced scenario. Specifically, effective-number GNNs (ENG) try to employ a reward function with the technique of effective number and resampling GNNs (RSG) attempts to modify the state transition function motivated by resampling technique. Extensive experiments are conducted on several real-world imbalanced datasets and the proposed methods significantly outperform the state-of-the-art methods in node classification task, accomplishing higher performance while utilizing a reduced volume of data. |
Keyword | Class Imbalanced Learning Costs Graph Neural Network (Gnn) Graph Neural Networks Node Classification Representation Learning Social Networking (Online) Task Analysis Training Training Data |
DOI | 10.1109/TCSS.2024.3350740 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Computer Science |
WOS Subject | Computer Science, Cybernetics ; Computer Science, Information Systems |
WOS ID | WOS:001166970700001 |
Publisher | IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC445 HOES LANE, PISCATAWAY, NJ 08855-4141 |
Scopus ID | 2-s2.0-85184326940 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | Faculty of Science and Technology THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Gong, Zhiguo |
Affiliation | State Key Laboratory of Internet of Things for Smart City and the Department of Computer and Information Science, University of Macau, Macao, China |
First Author Affilication | University of Macau |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Zhou, Mengting,Gong, Zhiguo. Informative Nodes Mining for Class-Imbalanced Representation Learning[J]. IEEE Transactions on Computational Social Systems, 2024. |
APA | Zhou, Mengting., & Gong, Zhiguo (2024). Informative Nodes Mining for Class-Imbalanced Representation Learning. IEEE Transactions on Computational Social Systems. |
MLA | Zhou, Mengting,et al."Informative Nodes Mining for Class-Imbalanced Representation Learning".IEEE Transactions on Computational Social Systems (2024). |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment