COIN: Co-Cluster Infomax for Bipartite Graphs
AI-generated Key Points
- Bipartite graphs are used in recommender systems, information retrieval, and drug discovery.
- Learning informative node embeddings is a fundamental challenge for bipartite graphs.
- Recent self-supervised learning methods discriminate instance-wise positive and negative node pairs that could contain cluster-level errors.
- A novel co-cluster infomax (COIN) framework is introduced to address this issue by maximizing the mutual information of co-clusters.
- COIN can easily calculate mutual information and is an end-to-end coclustering method that can be trained jointly with other objective functions and optimized via backpropagation.
- Theoretical analysis proves that COIN effectively increases the mutual information of node embeddings while being upperbounded by the prior distributions of nodes.
- COIN is extensively evaluated on various benchmark datasets and tasks to demonstrate its effectiveness.
- Sensitivity experiments show that COIN performs best when the number of clusters is around the ground truth number of classes and has small variances for different coefficients λ for mutual information on Wikipedia datasets.
- Overall, COIN provides a powerful solution to learning informative node embeddings in bipartite graphs by capturing cluster level information through maximizing mutual information of co-clusters.
Authors: Baoyu Jing, Yuchen Yan, Yada Zhu, Hanghang Tong
Abstract: Bipartite graphs are powerful data structures to model interactions between two types of nodes, which have been used in a variety of applications, such as recommender systems, information retrieval, and drug discovery. A fundamental challenge for bipartite graphs is how to learn informative node embeddings. Despite the success of recent self-supervised learning methods on bipartite graphs, their objectives are discriminating instance-wise positive and negative node pairs, which could contain cluster-level errors. In this paper, we introduce a novel co-cluster infomax (COIN) framework, which captures the cluster-level information by maximizing the mutual information of co-clusters. Different from previous infomax methods which estimate mutual information by neural networks, COIN could easily calculate mutual information. Besides, COIN is an end-to-end coclustering method which can be trained jointly with other objective functions and optimized via back-propagation. Furthermore, we also provide theoretical analysis for COIN. We theoretically prove that COIN is able to effectively increase the mutual information of node embeddings and COIN is upper-bounded by the prior distributions of nodes. We extensively evaluate the proposed COIN framework on various benchmark datasets and tasks to demonstrate the effectiveness of COIN.
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
Assess the quality of the AI-generated content by voting
Score: 0
Why do we need votes?
Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.