Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Yanqiao Zhu 0001

dblp:67/8383-1 · DBLP profile ↗
← Back
30ranked-venue papers
5as first author
27since 2021 · last 2025
0000-0003-2205-5304ORCID · verified

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 20 · 2 first-author · 18 since 2021Databases, data management, data science and information retrieval · 14 · 3 first-author · 13 since 2021Graphics, computer vision, multimedia, augmented reality and games · 6 · 4 since 2021Applied, interdisciplinary, general and emerging computing · 6 · 1 first-author · 6 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
15 papers
Graph learning · 42% Trustworthy machine learning · 10% Representation and self-supervised learning · 10%
Databases, data mining, and information retrieval
7 papers
Recommender systems · 74% Information retrieval · 26%
Interdisciplinary, comprehensive, and emerging computing
7 papers
Bioinformatics and computational biology · 84% Computational science and engineering · 12% Computing education · 4%

Topics — the 30 heaviest of 52, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Graph learning
graph neural network
2.552024
Learning Over Molecular Conformer Ensembles: Datasets and Benchmarks · ICLR 2024
GSLB: The Graph Structure Learning Benchmark · NeurIPS 2023
TAGNN: Target Attentive Graph Neural Networks for Session-based Recommendation · SIGIR 2020
Machine learning › Graph learning
molecular representation learning
1.522025
Symmetry-Preserving Conformer Ensemble Networks for Molecular Representation Learning · NeurIPS 2025
Uncovering Neural Scaling Laws in Molecular Representation Learning · NeurIPS 2023
Machine learning › Graph learning
graph structure learning
1.422024
Bi-Level Graph Structure Learning for Next POI Recommendation · IEEE Trans. Knowl. Data Eng. 2024
GSLB: The Graph Structure Learning Benchmark · NeurIPS 2023
Machine learning › Representation and self-supervised learning
pre-training
1.322023
Uncovering Neural Scaling Laws in Molecular Representation Learning · NeurIPS 2023
A Systematic Survey of Chemical Pre-trained Models · IJCAI 2023
Recommender systems
multimodal recommendation
1.222023
Latent Structure Mining With Contrastive Modality Fusion for Multimedia Recommendation · IEEE Trans. Knowl. Data Eng. 2023
Mining Latent Structures for Multimedia Recommendation · ACM Multimedia 2021
Bioinformatics and computational biology › molecular informatics
molecular representation learning
1.022024
Learning Over Molecular Conformer Ensembles: Datasets and Benchmarks · ICLR 2024
A Systematic Survey of Chemical Pre-trained Models · IJCAI 2023
Machine learning › Optimization for machine learning
black-box optimization
0.912025
Efficient Evolutionary Search Over Chemical Space with Large Language Models · ICLR 2025
Machine learning › Deep learning architectures and training
equivariant neural network
0.912025
Symmetry-Preserving Conformer Ensemble Networks for Molecular Representation Learning · NeurIPS 2025
Machine learning › Optimization for machine learning
evolutionary computation
0.912025
Efficient Evolutionary Search Over Chemical Space with Large Language Models · ICLR 2025
Natural language and speech › Language models and text generation
large language model
0.912025
Efficient Evolutionary Search Over Chemical Space with Large Language Models · ICLR 2025
Bioinformatics and computational biology
molecule discovery
0.912025
Efficient Evolutionary Search Over Chemical Space with Large Language Models · ICLR 2025
Recommender systems
session-based recommendation
0.822020
TAGNN: Target Attentive Graph Neural Networks for Session-based Recommendation · SIGIR 2020
Session-Based Recommendation with Graph Neural Networks · AAAI 2019
Natural language and speech › Language models and text generation
large language model evaluation
0.812024
SciBench: Evaluating College-Level Scientific Problem-Solving Abilities of Large Language Models · ICML 2024
Machine learning › Graph learning › graph representation learning
molecular graph representation
0.812024
Learning Over Molecular Conformer Ensembles: Datasets and Benchmarks · ICLR 2024
Information retrieval › document retrieval › domain-specific retrieval
biomedical information retrieval
0.812024
BMRetriever: Tuning Large Language Models as Better Biomedical Text Retrievers · EMNLP 2024
Information retrieval › retrieval models › neural retrieval
dense retrieval
0.812024
BMRetriever: Tuning Large Language Models as Better Biomedical Text Retrievers · EMNLP 2024
Recommender systems › point-of-interest recommendation
next POI recommendation
0.812024
Bi-Level Graph Structure Learning for Next POI Recommendation · IEEE Trans. Knowl. Data Eng. 2024
Recommender systems
point-of-interest recommendation
0.812024
Bi-Level Graph Structure Learning for Next POI Recommendation · IEEE Trans. Knowl. Data Eng. 2024
Information retrieval
retrieval models
0.812024
BMRetriever: Tuning Large Language Models as Better Biomedical Text Retrievers · EMNLP 2024
Machine learning › Graph learning › graph neural network
graph convolution
0.712023
Latent Structure Mining With Contrastive Modality Fusion for Multimedia Recommendation · IEEE Trans. Knowl. Data Eng. 2023
Machine learning › Trustworthy machine learning › robustness
learning with noisy labels
0.712023
Neighborhood-Regularized Self-Training for Learning with Few Labels · AAAI 2023
Machine learning › Trustworthy machine learning › robustness › learning with noisy labels
pseudo-label denoising
0.712023
Neighborhood-Regularized Self-Training for Learning with Few Labels · AAAI 2023
Machine learning › Learning paradigms › semi-supervised learning › pseudo-labeling
pseudo-label selection
0.712023
Neighborhood-Regularized Self-Training for Learning with Few Labels · AAAI 2023
Machine learning › Trustworthy machine learning
robustness evaluation
0.712023
GSLB: The Graph Structure Learning Benchmark · NeurIPS 2023
Machine learning › Deep learning architectures and training
scaling laws
0.712023
Uncovering Neural Scaling Laws in Molecular Representation Learning · NeurIPS 2023
Machine learning › Transfer learning and domain adaptation › domain adaptation › unsupervised domain adaptation
self-training
0.712023
Neighborhood-Regularized Self-Training for Learning with Few Labels · AAAI 2023
Machine learning › Learning paradigms
semi-supervised learning
0.712023
Neighborhood-Regularized Self-Training for Learning with Few Labels · AAAI 2023
Computational science and engineering › materials science
materials discovery
0.712023
M2Hub: Unlocking the Potential of Machine Learning for Materials Discovery · NeurIPS 2023
Bioinformatics and computational biology › molecular informatics › molecular modeling
molecular structure modeling
0.712023
M2Hub: Unlocking the Potential of Machine Learning for Materials Discovery · NeurIPS 2023
Empirical software engineering
open source software
0.712023
Code Recommendation for Open Source Software Developers · WWW 2023

Methods — techniques the papers use, named apart from their topics

graph neural network · 6.7contrastive learning · 1.8large language model · 1.7hierarchical ensemble encoding · 1.7geometric mixture-of-experts · 1.7evolutionary algorithm · 1.7equivariance · 1.7prompting strategies · 1.5ensemble learning · 1.5benchmarking study · 1.5heterogeneous graph · 1.3modality-aware structure learning · 1.2unsupervised pre-training · 0.8synthetic data generation · 0.8instruction fine-tuning · 0.8hierarchical graph learning · 0.8deep neural network · 0.7collaborative filtering · 0.7
YearPublicationVenuePosition
2025 Efficient Evolutionary Search Over Chemical Space with Large Language Models
abstract
Molecular discovery, when formulated as an optimization problem, presents significant computational challenges because optimization objectives can be non-differentiable. Evolutionary Algorithms (EAs), often used to optimize black-box objectives in molecular discovery, traverse chemical space by performing random mutations and crossovers, leading to a large number of expensive objective evaluations. In this work, we ameliorate this shortcoming by incorporating chemistry-aware Large Language Models (LLMs) into EAs. Namely, we redesign crossover and mutation operations in EAs using LLMs trained on large corpora of chemical information. We perform extensive empirical studies on both commercial and open-source models on multiple tasks involving property optimization, molecular rediscovery, and structure-based drug design, demonstrating that the joint usage of LLMs with EAs yields superior performance over all baseline models across single- and multi-objective settings. We demonstrate that our algorithm improves both the quality of the final solution and convergence speed, thereby reducing the number of required objective evaluations.
Haorui Wang, Marta Skreta, Cher Tian Ser, Wenhao Gao 0001, Felix Strieth-Kalthoff, Chenru Duan, Yuchen Zhuang, Yue Yu 0001, Yanqiao Zhu 0001, Yuanqi Du, Alán Aspuru-Guzik, Kirill Neklyudov, Chao Zhang 0014
ICLR10
2025 Symmetry-Preserving Conformer Ensemble Networks for Molecular Representation Learning
abstract
Molecular representation learning has emerged as a promising approach for modeling molecules with deep learning in chemistry and beyond. While 3D geometric models effectively capture molecular structure, they typically process single static conformers, overlooking the inherent flexibility and dynamics of molecules. In reality, many molecular properties depend on distributions of thermodynamically accessible conformations rather than single structures. Recent works show that learning from conformer ensembles can improve molecular representations, but existing approaches either produce unphysical structures through averaging or require restrictive molecular alignment. In this paper, we propose SymmetryPreserving Conformer Ensemble networks (SPiCE), which introduces two key innovations: (1) geometric mixture-of-experts for selective processing of scalar and vector features, and (2) hierarchical ensemble encoding that combines ensemblelevel representation with cross-conformer integration. Crucially, SPiCE ensures physically meaningful representations by maintaining joint equivariance to geometric transformations of individual conformers and conformer permutations. Extensive experiments demonstrate that SPiCE consistently outperforms existing conformer ensemble methods and state-of-the-art structural aggregation models across quantum mechanical and biological property prediction tasks.
Yanqiao Zhu 0001, Yidan Shi, Yuanzhou Chen, Yizhou Sun, Wei Wang 0010
NeurIPS1
2024 BMRetriever: Tuning Large Language Models as Better Biomedical Text Retrievers
abstract
Developing effective biomedical retrieval models is important for excelling at knowledge-intensive biomedical tasks but still challenging due to the lack of sufficient publicly annotated biomedical data and computational resources. We present BMRetriever, a series of dense retrievers for enhancing biomedical retrieval via unsupervised pre-training on large biomedical corpora, followed by instruction fine-tuning on a combination of labeled datasets and synthetic pairs. Experiments on 5 biomedical tasks across 11 datasets verify BMRetriever's efficacy on various biomedical applications. BMRetriever also exhibits strong parameter efficiency, with the 410M variant outperforming baselines up to 11.7 times larger, and the 2B variant matching the performance of models with over 5B parameters. The training data and model checkpoints are released at https://huggingface.co/BMRetriever to ensure transparency, reproducibility, and application to new domains.
Ran Xu 0002, Wenqi Shi 0002, Yue Yu 0001, Yuchen Zhuang, Yanqiao Zhu 0001, May D. Wang, Joyce C. Ho, Chao Zhang 0014, Carl Yang 0001
EMNLP5
2024 Learning Over Molecular Conformer Ensembles: Datasets and Benchmarks
abstract
Molecular Representation Learning (MRL) has proven impactful in numerous biochemical applications such as drug discovery and enzyme design. While Graph Neural Networks (GNNs) are effective at learning molecular representations from a 2D molecular graph or a single 3D structure, existing works often overlook the flexible nature of molecules, which continuously interconvert across conformations via chemical bond rotations and minor vibrational perturbations. To better account for molecular flexibility, some recent works formulate MRL as an ensemble learning problem, focusing on explicitly learning from a set of conformer structures. However, most of these studies have limited datasets, tasks, and models. In this work, we introduce the first MoleculAR Conformer Ensemble Learning (MARCEL) benchmark to thoroughly evaluate the potential of learning on con- former ensembles and suggest promising research directions. MARCEL includes four datasets covering diverse molecule- and reaction-level properties of chemically diverse molecules including organocatalysts and transition-metal catalysts, extending beyond the scope of common GNN benchmarks that are confined to drug-like molecules. In addition, we conduct a comprehensive empirical study, which benchmarks representative 1D, 2D, and 3D MRL models, along with two strategies that explicitly incorporate conformer ensembles into 3D models. Our findings reveal that direct learning from an accessible conformer space can improve performance on a variety of tasks and models.
Yanqiao Zhu 0001, Jeehyun Hwang, Keir Adams, Zhen Liu 0069, Bozhao Nan, Brock Stenfors, Yuanqi Du, Jatin Chauhan, Olaf Wiest, Olexandr Isayev, Connor W. Coley, Yizhou Sun, Wei Wang 0010
ICLR1
2024 SciBench: Evaluating College-Level Scientific Problem-Solving Abilities of Large Language Models
abstract
Most existing Large Language Model (LLM) benchmarks on scientific problem reasoning focus on problems grounded in high-school subjects and are confined to elementary algebraic operations. To systematically examine the reasoning capabilities required for solving complex scientific problems, we introduce an expansive benchmark suite SciBench for LLMs. SciBench contains a carefully curated dataset featuring a range of collegiate-level scientific problems from mathematics, chemistry, and physics domains. Based on the dataset, we conduct an in-depth benchmarking study of representative open-source and proprietary LLMs with various prompting strategies. The results reveal that current LLMs fall short of delivering satisfactory performance, with the best overall score of merely 43.22%. Furthermore, through a detailed user study, we categorize the errors made by LLMs into ten problem-solving abilities. Our analysis indicates that no single prompting strategy significantly outperforms the others and some strategies that demonstrate improvements in certain problem-solving skills could result in declines in other skills. We envision that SciBench will catalyze further developments in the reasoning abilities of LLMs, thereby ultimately contributing to scientific research and discovery.
Ziniu Hu, Pan Lu, Yanqiao Zhu 0001, Jieyu Zhang 0001, Satyen Subramaniam, Arjun R. Loomba, Shichang Zhang, Yizhou Sun, Wei Wang 0010
ICML4
2024 Bi-Level Graph Structure Learning for Next POI Recommendation
abstract
Next point-of-interest (POI) recommendation aims to predict a user's next destination based on sequential check-in history and a set of POI candidates. Graph neural networks (GNNs) have demonstrated a remarkable capability in this endeavor by exploiting the extensive global collaborative signals present among POIs. However, most of the existing graph-based approaches construct graph structures based on pre-defined heuristics, failing to consider inherent hierarchical structures of POI features such as geographical locations and visiting peaks, or suffering from noisy and incomplete structures in graphs. To address the aforementioned issues, this paper presents a novelBi-levelGraphStructureLearning (${\sf BiGSL}$) for next POI recommendation.${\sf BiGSL}$first learns a hierarchical graph structure to capture the fine-to-coarse connectivity between POIs and prototypes, and then uses a pairwise learning module to dynamically infer relationships between POI pairs and prototype pairs. Based on the learned bi-level graphs, our model then employs a multi-relational graph network that considers both POI- and prototype-level neighbors, resulting in improved POI representations. Our bi-level structure learning scheme is more robust to data noise and incompleteness, and improves the exploration ability for recommendation by alleviating sparsity issues. Experimental results on three real-world datasets demonstrate the superiority of our model over existing state-of-the-art methods, with a significant improvement in recommendation accuracy and exploration performance.
Liang Wang 0056, Qiang Liu 0006, Yanqiao Zhu 0001, Xiang Tao, Mengdi Zhang 0002, Liang Wang 0001
IEEE Trans. Knowl. Data Eng.4
2023 Neighborhood-Regularized Self-Training for Learning with Few Labels
abstract
Training deep neural networks (DNNs) with limited supervision has been a popular research topic as it can significantly alleviate the annotation burden. Self-training has been successfully applied in semi-supervised learning tasks, but one drawback of self-training is that it is vulnerable to the label noise from incorrect pseudo labels. Inspired by the fact that samples with similar labels tend to share similar representations, we develop a neighborhood-based sample selection approach to tackle the issue of noisy pseudo labels. We further stabilize self-training via aggregating the predictions from different rounds during sample selection. Experiments on eight tasks show that our proposed method outperforms the strongest self-training baseline with 1.83% and 2.51% performance gain for text and graph datasets on average. Our further analysis demonstrates that our proposed data selection strategy reduces the noise of pseudo labels by 36.8% and saves 57.3% of the time when compared with the best baseline. Our code and appendices will be uploaded to https://github.com/ritaranx/NeST.
Ran Xu 0002, Yue Yu 0001, Hejie Cui, Xuan Kan, Yanqiao Zhu 0001, Joyce C. Ho, Chao Zhang 0014, Carl Yang 0001
AAAI5
2023 AKE-GNN: Effective Graph Learning with Adaptive Knowledge Exchange
abstract
Graph Neural Networks (GNNs) have already been widely used in various graph mining tasks. However, recent works reveal that the learned weights (channels) in well-trained GNNs are highly redundant, which inevitably limits the performance of GNNs. Instead of removing these redundant channels for efficiency consideration, we aim to reactivate them to enlarge the representation capacity of GNNs for effective graph learning. In this paper, we propose to substitute these redundant channels with other informative channels to achieve this goal. We introduce a novel GNN learning framework named AKE-GNN, which performs the Adaptive Knowledge Exchange strategy among multiple graph views generated by graph augmentations. AKE-GNN first trains multiple GNNs each corresponding to one graph view to obtain informative channels. Then, AKE-GNN iteratively exchanges redundant channels in the weight parameter matrix of one GNN with informative channels of another GNN in a layer-wise manner. Additionally, existing GNNs can be seamlessly incorporated into our framework. AKE-GNN achieves superior performance compared with various baselines across a suite of experiments on node classification, link prediction, and graph classification. In particular, we conduct a series of experiments on 15 public benchmark datasets, 8 popular GNN models, and 3 graph tasks and show that AKE-GNN consistently outperforms existing popular GNN models and even their ensembles. Extensive ablation studies and analyses on knowledge exchange methods validate the effectiveness of AKE-GNN.
Liang Zeng 0002, Jin Xu 0010, Zijun Yao 0002, Yanqiao Zhu 0001, Jian Li 0015
CIKM4
2023 A Systematic Survey of Chemical Pre-trained Models
abstract
Deep learning has achieved remarkable success in learning representations for molecules, which is crucial for various biochemical applications, ranging from property prediction to drug design. However, training Deep Neural Networks (DNNs) from scratch often requires abundant labeled molecules, which are expensive to acquire in the real world. To alleviate this issue, tremendous efforts have been devoted to Chemical Pre-trained Models (CPMs), where DNNs are pre-trained using large-scale unlabeled molecular databases and then fine-tuned over specific downstream tasks. Despite the prosperity, there lacks a systematic review of this fast-growing field. In this paper, we present the first survey that summarizes the current progress of CPMs. We first highlight the limitations of training molecular representation models from scratch to motivate CPM studies. Next, we systematically review recent advances on this topic from several key perspectives, including molecular descriptors, encoder architectures, pre-training strategies, and applications. We also highlight the challenges and promising avenues for future research, providing a useful resource for both machine learning and scientific communities.
Jun Xia 0001, Yanqiao Zhu 0001, Yuanqi Du, Stan Z. Li
IJCAI2
2023 Uncovering Neural Scaling Laws in Molecular Representation Learning
abstract
Molecular Representation Learning (MRL) has emerged as a powerful tool for drug and materials discovery in a variety of tasks such as virtual screening and inverse design. While there has been a surge of interest in advancing model-centric techniques, the influence of both data quantity and quality on molecular representations is not yet clearly understood within this field. In this paper, we delve into the neural scaling behaviors of MRL from a data-centric viewpoint, examining four key dimensions: (1) data modalities, (2) dataset splitting, (3) the role of pre-training, and (4) model capacity.Our empirical studies confirm a consistent power-law relationship between data volume and MRL performance across these dimensions. Additionally, through detailed analysis, we identify potential avenues for improving learning efficiency.To challenge these scaling laws, we adapt seven popular data pruning strategies to molecular data and benchmark their performance. Our findings underline the importance of data-centric MRL and highlight possible directions for future research.
Dingshuo Chen, Yanqiao Zhu 0001, Jieyu Zhang 0001, Yuanqi Du, Zhixun Li, Qiang Liu 0006, Liang Wang 0056
NeurIPS2
2023 M2Hub: Unlocking the Potential of Machine Learning for Materials Discovery
abstract
We introduce M$^2$Hub, a toolkit for advancing machine learning in materials discovery. Machine learning has achieved remarkable progress in modeling molecular structures, especially biomolecules for drug discovery. However, the development of machine learning approaches for modeling materials structures lag behind, which is partly due to the lack of an integrated platform that enables access to diverse tasks for materials discovery. To bridge this gap, M$^2$Hub will enable easy access to materials discovery tasks, datasets, machine learning methods, evaluations, and benchmark results that cover the entire workflow. Specifically, the first release of M$^2$Hub focuses on three key stages in materials discovery: virtual screening, inverse design, and molecular simulation, including 9 datasets that covers 6 types of materials with 56 tasks across 8 types of material properties. We further provide 2 synthetic datasets for the purpose of generative tasks on materials. In addition to random data splits, we also provide 3 additional data partitions to reflect the real-world materials discovery scenarios. State-of-the-art machine learning methods (including those are suitable for materials structures but never compared in the literature) are benchmarked on representative tasks. Our codes and library are publicly available at \url{https://github.com/yuanqidu/M2Hub}.
Yuanqi Du, Yingheng Wang, Yining Huang, Jianan Canal Li, Yanqiao Zhu 0001, Chenru Duan, John M. Gregoire, Carla P. Gomes
NeurIPS5
2023 GSLB: The Graph Structure Learning Benchmark
abstract
Graph Structure Learning (GSL) has recently garnered considerable attention due to its ability to optimize both the parameters of Graph Neural Networks (GNNs) and the computation graph structure simultaneously. Despite the proliferation of GSL methods developed in recent years, there is no standard experimental setting or fair comparison for performance evaluation, which creates a great obstacle to understanding the progress in this field. To fill this gap, we systematically analyze the performance of GSL in different scenarios and develop a comprehensive Graph Structure Learning Benchmark (GSLB) curated from 20 diverse graph datasets and 16 distinct GSL algorithms. Specifically, GSLB systematically investigates the characteristics of GSL in terms of three dimensions: effectiveness, robustness, and complexity. We comprehensively evaluate state-of-the-art GSL algorithms in node- and graph-level tasks, and analyze their performance in robust learning and model complexity. Further, to facilitate reproducible research, we have developed an easy-to-use library for training, evaluating, and visualizing different GSL methods. Empirical results of our extensive experiments demonstrate the ability of GSL and reveal its potential benefits on various downstream tasks, offering insights and opportunities for future research. The code of GSLB is available at: https://github.com/GSL-Benchmark/GSLB.
Zhixun Li, Yanqiao Zhu 0001, Dingshuo Chen, Yingtao Luo, Xiangxin Zhou, Qiang Liu 0006, Liang Wang 0001, Jeffrey Xu Yu
NeurIPS4
2023 Code Recommendation for Open Source Software Developers
abstract
Open Source Software (OSS) is forming the spines of technology infrastructures, attracting millions of talents to contribute. Notably, it is challenging and critical to consider both the developers’ interests and the semantic features of the project code to recommend appropriate development tasks to OSS developers. In this paper, we formulate the novel problem of code recommendation, whose purpose is to predict the future contribution behaviors of developers given their interaction history, the semantic features of source code, and the hierarchical file structures of projects. We introduce CODER, a novel graph-based CODE Recommendation framework for open source software developers, which accounts for the complex interactions among multiple parties within the system. CODER jointly models microscopic user-code interactions and macroscopic user-project interactions via a heterogeneous graph and further bridges the two levels of information through aggregation on file-structure graphs that reflect the project hierarchy. Moreover, to overcome the lack of reliable benchmarks, we construct three large-scale datasets to facilitate future research in this direction. Extensive experiments show that our CODER framework achieves superior performance under various experimental settings, including intra-project, cross-project, and cold-start recommendation.
Yiqiao Jin, Yunsheng Bai, Yanqiao Zhu 0001, Yizhou Sun, Wei Wang 0010
WWW3
2023 Unsupervised Graph Representation Learning with Cluster-aware Self-training and Refining
abstract
Unsupervised graph representation learning aims to learn low-dimensional node embeddings without supervision while preserving graph topological structures and node attributive features. Previous Graph Neural Networks (GNN) require a large number of labeled nodes, which may not be accessible in real-world applications. To this end, we present a novel unsupervised graph neural network model with Cluster-aware Self-training and Refining ( CLEAR ). Specifically, in the proposed CLEAR model, we perform clustering on the node embeddings and update the model parameters by predicting the cluster assignments. To avoid degenerate solutions of clustering, we formulate the graph clustering problem as an optimal transport problem and leverage a balanced clustering strategy. Moreover, we observe that graphs often contain inter-class edges, which mislead the GNN model to aggregate noisy information from neighborhood nodes. Therefore, we propose to refine the graph topology by strengthening intra-class edges and reducing node connections between different classes based on cluster labels, which better preserves cluster structures in the embedding space. We conduct comprehensive experiments on two benchmark tasks using real-world datasets. The results demonstrate the superior performance of the proposed model over baseline methods. Notably, our model gains over 7% improvements in terms of accuracy on node clustering over state-of-the-arts.
Yanqiao Zhu 0001, Yichen Xu 0008, Feng Yu 0001, Qiang Liu 0006
ACM Trans. Intell. Syst. Technol.1
2023 Latent Structure Mining With Contrastive Modality Fusion for Multimedia Recommendation
abstract
Multimedia contents are of predominance in the modern Web era. Recent years have witnessed growing research interests in multimedia recommendation, which aims to predict whether a user will interact with an item with multimodal contents. Most previous studies focus on modeling user-item interactions with multimodal features included as side information. However, this scheme is not well-designed for multimedia recommendation. First, onlycollaborativeitem-item relationships are implicitly modeled through high-order item-user-item co-occurrences. Considering that items are associated with rich contents in multiple modalities, we argue that the latentsemanticitem-item structures underlying these multimodal contents could be beneficial for learning better item representations and assist the recommender models to comprehensively discover candidate items. Second, although previous studies consider multiple modalities, their ways of fusing multiple modalities by linear combination or concatenation is insufficient to fully capture content information of items and item relationships. To address these deficiencies, we propose a latent structureMIning withContRastive mOdality fusion model, which we term MICRO for brevity. To be specific, we devise a novel modality-aware structure learning module, which learns item-item relationships for each modality. Based on the learned modality-aware latent item relationships, we perform graph convolutions to explicitly inject item affinities into modality-aware item representations. Additionally, we design a novel multimodal contrastive framework to facilitate item-level multimodal fusion by mining both modality-shared and modality-specific information. Finally, the item representations are plugged into existing collaborative filtering methods to make accurate recommendation. Extensive experiments on three real-world datasets demonstrate the superiority of our method over state-of-arts and rationalize the design choice of our work.
Yanqiao Zhu 0001, Qiang Liu 0006, Mengqi Zhang 0002, Liang Wang 0001
IEEE Trans. Knowl. Data Eng.2
2023 BrainGB: A Benchmark for Brain Network Analysis With Graph Neural Networks
abstract
Mapping the connectome of the human brain using structural or functional connectivity has become one of the most pervasive paradigms for neuroimaging analysis. Recently, Graph Neural Networks (GNNs) motivated from geometric deep learning have attracted broad interest due to their established power for modeling complex networked data. Despite their superior performance in many fields, there has not yet been a systematic study of how to design effective GNNs for brain network analysis. To bridge this gap, we present BrainGB, a benchmark for brain network analysis with GNNs. BrainGB standardizes the process by (1) summarizing brain network construction pipelines for both functional and structural neuroimaging modalities and (2) modularizing the implementation of GNN designs. We conduct extensive experiments on datasets across cohorts and modalities and recommend a set of general recipes for effective GNN designs on brain networks. To support open and reproducible research on GNN-based brain network analysis, we host the BrainGB website at https://braingb.us with models, tutorials, examples, as well as an out-of-box Python package. We hope that this work will provide useful empirical evidence and offer insights for future research in this novel and promising direction.
Hejie Cui, Wei Dai 0013, Yanqiao Zhu 0001, Xuan Kan, Antonio Aodong Chen Gu, Joshua Lukemire, Liang Zhan, Lifang He 0001, Ying Guo 0003, Carl Yang 0001
IEEE Trans. Medical Imaging3
2022 BrainGB: A Benchmark for Brain Network Analysis with Graph Neural Networks (Extended Abstract)
abstract
Mapping the connectome of the human brain using structural or functional connectivity has become one of the most pervasive paradigms for neuroimaging analysis. Recently, Graph Neural Networks (GNNs) motivated from geometric deep learning have attracted broad interest due to their established power for modeling complex networked data. Despite their superior performance in many fields, there has not yet been a systematic study of how to design effective GNNs for brain network analysis. To bridge this gap, we present BrainGB, a benchmark for brain network analysis with GNNs. BrainGB standardizes the process by (1) summarizing brain network construction pipelines for both functional and structural neuroimaging modalities and (2) modularizing the implementation of GNN designs. We conduct extensive experiments on datasets across cohorts and modalities and recommend a set of general recipes for effective GNN designs on brain networks. To support open and reproducible research on GNN-based brain network analysis, we host the BrainGBwebsite at https://braingb.us with models, tutorials, examples, as well as an out-of-box Python package. We hope that this work will provide useful empirical evidence and offer insights for future research in this novel and promising direction.
Hejie Cui, Wei Dai 0013, Yanqiao Zhu 0001, Xuan Kan, Antonio Aodong Chen Gu, Joshua Lukemire, Liang Zhan, Lifang He 0001, Ying Guo 0003, Carl Yang 0001
IEEE Big Data3
2022 Learning Task-Aware Effective Brain Connectivity for fMRI Analysis with Graph Neural Networks (Extended Abstract)
abstract
Functional magnetic resonance imaging (fMRI) has become one of the most common imaging modalities for brain function analysis. Recently, graph neural networks (GNN) have been adopted for fMRI analysis with superior performance. Unfortunately, traditional functional brain networks are mainly constructed based on similarities among region of interests (ROI), which are noisy and agnostic to the downstream prediction tasks and can lead to inferior results for GNN-based models. To better adapt GNNs for fMRI analysis, we propose TBDS, an end-to-end framework based on Task-aware Brain connectivity DAG (short for Directed Acyclic Graph) Structure generation for fMRI analysis. The key component of TBDS is the brain network generator which adopts a DAG learning approach to transform the raw time-series into task-aware brain connectivities. Besides, we design an additional contrastive regularization to inject task-specific knowledge during the brain network generation process. Comprehensive experiments on two fMRI datasets, namely Adolescent Brain Cognitive Development (ABCD) and Philadelphia Neuroimaging Cohort (PNC) datasets demonstrate the efficacy of TBDS. In addition, the generated brain networks also highlight the prediction-related brain regions and thus provide unique interpretations of the prediction results. Our implementation will be published upon acceptance.
Yue Yu 0001, Xuan Kan, Hejie Cui, Ran Xu 0002, Yujia Zheng 0001, Xiangchen Song, Yanqiao Zhu 0001, Kun Zhang 0001, Razieh Nabi, Ying Guo 0003, Chao Zhang 0014, Carl Yang 0001
IEEE Big Data7
2022 Deep Contrastive Multiview Network Embedding
abstract
Multiview network embedding aims at projecting nodes in the network to low-dimensional vectors, while preserving their multiple relations and attribute information. Contrastive learning approaches have shown promising performance in this task. However, they neglect the semantic consistency between fused and view representations and have difficulty in modeling complementary information between different views. To deal with these deficiencies, this work presents a novel Contrastive leaRning framEwork for Multiview network Embedding (CREME). In our work, different views can be obtained based on the various relations among nodes. Then, we generate view embeddings via proper view encoders and utilize an attentive multiview aggregator to fuse these representations. Particularly, we design two collaborative contrastive objectives, view fusion InfoMax and inter-view InfoMin, to train the model in a self-supervised manner. The former objective distills information from embeddings generated from different views, while the latter captures complementary information among views to promote distinctive view embeddings. We also show that the two objectives can be unified into one objective for model training. Extensive experiments on three real-world datasets demonstrate that our proposed CREME is able to consistently outperform state-of-the-art methods.
Mengqi Zhang 0002, Yanqiao Zhu 0001, Qiang Liu 0006, Liang Wang 0001
CIKM2
2022 Data-Efficient Brain Connectome Analysis via Multi-Task Meta-Learning
abstract
Brain networks characterize complex connectivities among brain regions as graph structures, which provide a powerful means to study brain connectomes. In recent years, graph neural networks have emerged as a prevalent paradigm of learning with structured data. However, most brain network datasets are limited in sample sizes due to the relatively high cost of data collection, which hinders the deep learning models from sufficient training. Inspired by meta-learning that learns new concepts fast with limited training examples, this paper studies data-efficient training strategies for analyzing brain connectomes in a cross-dataset setting. Specifically, we propose to meta-train the model on datasets of large sample sizes and transfer the knowledge to small datasets. In addition, we also explore brain-network-oriented designs, including atlas transformation and adaptive task reweighing. Compared to other pre-training strategies, our meta-learning-based approach achieves higher and stabler performance, which demonstrates the effectiveness of our proposed solutions. The framework is also able to derive new insights regarding the similarities among datasets and diseases in a data-driven fashion.
Yi Yang 0094, Yanqiao Zhu 0001, Hejie Cui, Xuan Kan, Lifang He 0001, Ying Guo 0003, Carl Yang 0001
KDD2
2022 Interpretable Graph Neural Networks for Connectome-Based Brain Disorder Analysis
Hejie Cui, Wei Dai 0013, Yanqiao Zhu 0001, Lifang He 0001, Carl Yang 0001
MICCAI (8)3
2022 Structure-Enhanced Heterogeneous Graph Contrastive Learning
abstract
Recent years have seen a growing interest in Graph Contrastive Learning (GCL), which trains Graph Neural Network (GNN) model to discriminate similar and dissimilar pairs of nodes without human annotations. Most prior GCL work focuses on homogeneous graphs and little attention has been paid to Heterogeneous Graphs (HGs) that involve different types of nodes and edges. Moreover, earlier studies reveal that the explicit use of structure information of underlying graphs is useful for learning representations. Conventional GCL methods merely measure the likelihood of contrastive pairs according to node representations, which may not align with the true semantic similarities. How to leverage such structure information for GCL is not yet well-understood. To address the aforementioned challenges, this paper presents a novel method dubbed STructure-EnhaNced heterogeneous graph ContrastIve Learning, STENCIL for brevity. At first, we generate multiple semantic views for HGs based on metapaths. Unlike most methods that maximize the consistency among these views, we propose a novel multiview contrastive aggregation objective to adaptively distill information from each view. In addition, we advocate the explicit use of structure embedding, which enriches the model with local structural patterns of the underlying HGs, so as to better mine true and hard negatives for GCL. Empirical studies on three real-world datasets show that our proposed method consistently outperforms existing state-of-the-art methods and even surpasses several supervised counterparts.
Yanqiao Zhu 0001, Yichen Xu 0008, Hejie Cui, Carl Yang 0001, Qiang Liu 0006
SDM1
2021 Deep Active Learning for Text Classification with Diverse Interpretations
abstract
Recently, Deep Neural Networks (DNNs) have made remarkable progress for text classification, which, however, still require a large number of labeled data. To train high-performing models with the minimal annotation cost, active learning is proposed to select and label the most informative samples, yet it is still challenging to measure informativeness of samples used in DNNs. In this paper, inspired by piece-wise linear interpretability of DNNs, we propose a novel Active Learning with DivErse iNterpretations (ALDEN) approach. With local interpretations in DNNs, ALDEN identifies linearly separable regions of samples. Then, it selects samples according to their diversity of local interpretations and queries their labels. To tackle the text classification problem, we choose the word with the most diverse interpretations to represent the whole sentence. Extensive experiments demonstrate that ALDEN consistently outperforms several state-of-the-art deep active learning methods.
Qiang Liu 0006, Yanqiao Zhu 0001, Zhaocheng Liu
CIKM2
2021 Disentangled Self-Attentive Neural Networks for Click-Through Rate Prediction
abstract
Click-Through Rate (CTR) prediction, whose aim is to predict the probability of whether a user will click on an item, is an essential task for many online applications. Due to the nature of data sparsity and high dimensionality of CTR prediction, a key to making effective prediction is to model high-order feature interaction. An efficient way to do this is to perform inner product of feature embeddings with self-attentive neural networks. To better model complex feature interaction, in this paper we propose a novel DisentanglEd Self-atTentIve NEtwork (DESTINE) framework for CTR prediction that explicitly decouples the computation of unary feature importance from pairwise interaction. Specifically, the unary term models the general importance of one feature on all other features, whereas the pairwise interaction term contributes to learning the pure impact for each feature pair. We conduct extensive experiments using two real-world benchmark datasets. The results show that DESTINE not only maintains computational efficiency but achieves consistent improvements over state-of-the-art baselines.
Yichen Xu 0008, Yanqiao Zhu 0001, Feng Yu 0001, Qiang Liu 0006
CIKM2
2021 Mining Latent Structures for Multimedia Recommendation
abstract
Multimedia content is of predominance in the modern Web era. Investigating how users interact with multimodal items is a continuing concern within the rapid development of recommender systems. The majority of previous work focuses on modeling user-item interactions with multimodal features included as side information. However, this scheme is not well-designed for multimedia recommendation. Specifically, only collaborative item-item relationships are implicitly modeled through high-order item-user-item relations. Considering that items are associated with rich contents in multiple modalities, we argue that the latent semantic item-item structures underlying these multimodal contents could be beneficial for learning better item representations and further boosting recommendation. To this end, we propose a LATent sTructure mining method for multImodal reCommEndation, which we term LATTICE for brevity. To be specific, in the proposed LATTICE model, we devise a novel modality-aware structure learning layer, which learns item-item structures for each modality and aggregates multiple modalities to obtain latent item graphs. Based on the learned latent graphs, we perform graph convolutions to explicitly inject high-order item affinities into item representations. These enriched item representations can then be plugged into existing collaborative filtering methods to make more accurate recommendations. Extensive experiments on three real-world datasets demonstrate the superiority of our method over state-of-the-art multimedia recommendation methods and validate the efficacy of mining latent item-item relationships from multimodal features.
Yanqiao Zhu 0001, Qiang Liu 0006, Shuhui Wang, Liang Wang 0001
ACM Multimedia2
2021 Graph Contrastive Learning with Adaptive Augmentation
abstract
Recently, contrastive learning (CL) has emerged as a successful method for unsupervised graph representation learning. Most graph CL methods first perform stochastic augmentation on the input graph to obtain two graph views and maximize the agreement of representations in the two views. Despite the prosperous development of graph CL methods, the design of graph augmentation schemes—a crucial component in CL—remains rarely explored. We argue that the data augmentation schemes should preserve intrinsic structures and attributes of graphs, which will force the model to learn representations that are insensitive to perturbation on unimportant nodes and edges. However, most existing methods adopt uniform data augmentation schemes, like uniformly dropping edges and uniformly shuffling features, leading to suboptimal performance. In this paper, we propose a novel graph contrastive representation learning method with adaptive augmentation that incorporates various priors for topological and semantic aspects of the graph. Specifically, on the topology level, we design augmentation schemes based on node centrality measures to highlight important connective structures. On the node attribute level, we corrupt node features by adding more noise to unimportant node features, to enforce the model to recognize underlying semantic information. We perform extensive experiments of node classification on a variety of real-world datasets. Experimental results demonstrate that our proposed method consistently outperforms existing state-of-the-art baselines and even surpasses some supervised counterparts, which validates the effectiveness of the proposed contrastive framework with adaptive augmentation.
Yanqiao Zhu 0001, Yichen Xu 0008, Feng Yu 0001, Qiang Liu 0006, Liang Wang 0001
WWW1
2021 GraphAIR: Graph representation learning with neighborhood aggregation and interaction
Fenyu Hu, Yanqiao Zhu 0001, Weiran Huang 0002, Liang Wang 0001, Tieniu Tan
Pattern Recognit.2
2020 TAGNN: Target Attentive Graph Neural Networks for Session-based Recommendation
abstract
Session-based recommendation nowadays plays a vital role in many websites, which aims to predict users' actions based on anonymous sessions. There have emerged many studies that model a session as a sequence or a graph via investigating temporal transitions of items in a session. However, these methods compress a session into one fixed representation vector without considering the target items to be predicted. The fixed vector will restrict the representation ability of the recommender model, considering the diversity of target items and users' interests. In this paper, we propose a novel target attentive graph neural network (TAGNN) model for session-based recommendation. In TAGNN, target-aware attention adaptively activates different user interests with respect to varied target items. The learned interest representation vector varies with different target items, greatly improving the expressiveness of the model. Moreover, TAGNN harnesses the power of graph neural networks to capture rich item transitions in sessions. Comprehensive experiments conducted on real-world datasets demonstrate its superiority over state-of-the-art methods.
Feng Yu 0001, Yanqiao Zhu 0001, Qiang Liu 0006, Liang Wang 0001, Tieniu Tan
SIGIR2
2019 Session-Based Recommendation with Graph Neural Networks
abstract
The problem of session-based recommendation aims to predict user actions based on anonymous sessions. Previous methods model a session as a sequence and estimate user representations besides item representations to make recommendations. Though achieved promising results, they are insufficient to obtain accurate user vectors in sessions and neglect complex transitions of items. To obtain accurate item embedding and take complex transitions of items into account, we propose a novel method, i.e. Session-based Recommendation with Graph Neural Networks, SR-GNN for brevity. In the proposed method, session sequences are modeled as graphstructured data. Based on the session graph, GNN can capture complex transitions of items, which are difficult to be revealed by previous conventional sequential methods. Each session is then represented as the composition of the global preference and the current interest of that session using an attention network. Extensive experiments conducted on two real datasets show that SR-GNN evidently outperforms the state-of-the-art session-based recommendation methods consistently.
Yuyuan Tang, Yanqiao Zhu 0001, Liang Wang 0001, Xing Xie 0001, Tieniu Tan
AAAI3
2019 Hierarchical Graph Convolutional Networks for Semi-supervised Node Classification
abstract
Graph convolutional networks (GCNs) have been successfully applied in node classification tasks of network mining. However, most of these models based on neighborhood aggregation are usually shallow and lack the “graph pooling” mechanism, which prevents the model from obtaining adequate global information. In order to increase the receptive field, we propose a novel deep Hierarchical Graph Convolutional Network (H-GCN) for semi-supervised node classification. H-GCN first repeatedly aggregates structurally similar nodes to hyper-nodes and then refines the coarsened graph to the original to restore the representation for each node. Instead of merely aggregating one- or two-hop neighborhood information, the proposed coarsening procedure enlarges the receptive field for each node, hence more global information can be captured. The proposed H-GCN model shows strong empirical performance on various public benchmark graph datasets, outperforming state-of-the-art methods and acquiring up to 5.9% performance improvement in terms of accuracy. In addition, when only a few labeled samples are provided, our model gains substantial improvements.
Fenyu Hu, Yanqiao Zhu 0001, Liang Wang 0001, Tieniu Tan
IJCAI2