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

Yijun Tian 0001

dblp:234/9123-1 · DBLP profile ↗
← Back
32ranked-venue papers
9as first author
31since 2021 · last 2026
0000-0003-2795-6080ORCID · conflict

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

Artificial intelligence and machine learning · 28 · 8 first-author · 28 since 2021Databases, data management, data science and information retrieval · 11 · 3 first-author · 10 since 2021Graphics, computer vision, multimedia, augmented reality and games · 9 · 5 first-author · 9 since 2021Applied, interdisciplinary, general and emerging computing · 3 · 1 first-author · 2 since 2021Human-computer interaction and ubiquitous computing · 1 · 1 first-author

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
24 papers
Graph learning · 33% Trustworthy machine learning · 13% Language models and text generation · 12%
Interdisciplinary, comprehensive, and emerging computing
4 papers
Bioinformatics and computational biology · 100%
Databases, data mining, and information retrieval
3 papers
Information retrieval · 78% Recommender systems · 22%

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

TopicWeightPapersLastEvidence papers
Machine learning › Graph learning
graph neural network
5.182024
Graph Cross Supervised Learning via Generalized Knowledge · KDD 2024
S3GCL: Spectral, Swift, Spatial Graph Contrastive Learning · ICML 2024
Mitigating Emergent Robustness Degradation while Scaling Graph Learning · ICLR 2024
Machine learning › Representation and self-supervised learning › contrastive learning
graph contrastive learning
2.432026
Adaptive and Context-rich Generative Self-supervised Learning on Graphs · AAAI 2026
S3GCL: Spectral, Swift, Spatial Graph Contrastive Learning · ICML 2024
When Sparsity Meets Contrastive Models: Less Graph Data Can Bring Better Class-Balanced Representations · ICML 2023
Machine learning › Graph learning
graph representation learning
2.332026
Adaptive and Context-rich Generative Self-supervised Learning on Graphs · AAAI 2026
When Sparsity Meets Contrastive Models: Less Graph Data Can Bring Better Class-Balanced Representations · ICML 2023
Chasing All-Round Graph Representation Robustness: Model, Training, and Optimization · ICLR 2023
Machine learning › Efficient and distributed learning › model compression
knowledge distillation
1.932026
Pareto-Based Heterogeneous Knowledge Distillation for MLPs on Graphs · AAAI 2026
Boosting Graph Neural Networks via Adaptive Knowledge Distillation · AAAI 2023
Beyond Answers: Transferring Reasoning Capabilities to Smaller LLMs Using Multi-Teacher Knowledge Distillation · WSDM 2025
Machine learning › Efficient and distributed learning
model compression
1.932026
Pareto-Based Heterogeneous Knowledge Distillation for MLPs on Graphs · AAAI 2026
Boosting Graph Neural Networks via Adaptive Knowledge Distillation · AAAI 2023
Beyond Answers: Transferring Reasoning Capabilities to Smaller LLMs Using Multi-Teacher Knowledge Distillation · WSDM 2025
Machine learning › Trustworthy machine learning
robustness
1.632024
Mitigating Emergent Robustness Degradation while Scaling Graph Learning · ICLR 2024
Chasing All-Round Graph Representation Robustness: Model, Training, and Optimization · ICLR 2023
Graph Cross Supervised Learning via Generalized Knowledge · KDD 2024
Machine learning › Graph learning › graph neural network
heterogeneous graph neural network
1.622026
Pareto-Based Heterogeneous Knowledge Distillation for MLPs on Graphs · AAAI 2026
RecipeRec: A Heterogeneous Graph Learning Model for Recipe Recommendation · IJCAI 2022
Natural language and speech › Language models and text generation
retrieval-augmented generation
1.522024
G-Retriever: Retrieval-Augmented Generation for Textual Graph Understanding and Question Answering · NeurIPS 2024
Graph Neural Prompting with Large Language Models · AAAI 2024
Computer vision › Vision and language › visual question answering
document visual question answering
1.012026
ALDEN: Reinforcement Learning for Active Navigation and Evidence Gathering in Long Documents · ACL (1) 2026
Machine learning › Representation and self-supervised learning › representation learning › unsupervised representation learning › self-supervised representation learning
generative self-supervised learning
1.012026
Adaptive and Context-rich Generative Self-supervised Learning on Graphs · AAAI 2026
Machine learning › Graph learning
graph self-supervised learning
1.012026
Adaptive and Context-rich Generative Self-supervised Learning on Graphs · AAAI 2026
Natural language and speech › Language models and text generation › natural language understanding
long document understanding
1.012026
ALDEN: Reinforcement Learning for Active Navigation and Evidence Gathering in Long Documents · ACL (1) 2026
Machine learning › Graph learning › graph self-supervised learning
masked graph modeling
1.012026
Adaptive and Context-rich Generative Self-supervised Learning on Graphs · AAAI 2026
Machine learning › Reinforcement learning
self-improving agent
1.012026
Reinforcement Learning for Self-Improving Agent with Skill Library · ACL (1) 2026
Computer vision › Vision and language
vision-language model
1.012026
ALDEN: Reinforcement Learning for Active Navigation and Evidence Gathering in Long Documents · ACL (1) 2026
Information retrieval › retrieval models › neural retrieval
dense retrieval
1.012026
LACONIC: Dense-Level Effectiveness for Scalable Sparse Retrieval via a Two-Phase Training Curriculum · SIGIR 2026
Information retrieval › retrieval models › sparse retrieval
learned sparse retrieval
1.012026
LACONIC: Dense-Level Effectiveness for Scalable Sparse Retrieval via a Two-Phase Training Curriculum · SIGIR 2026
Natural language and speech › Language models and text generation
chain-of-thought reasoning
0.912025
Beyond Answers: Transferring Reasoning Capabilities to Smaller LLMs Using Multi-Teacher Knowledge Distillation · WSDM 2025
Machine learning › Efficient and distributed learning › distillation
knowledge distillation for language models
0.912025
Beyond Answers: Transferring Reasoning Capabilities to Smaller LLMs Using Multi-Teacher Knowledge Distillation · WSDM 2025
Natural language and speech › Language models and text generation
large language model
0.912025
Beyond Answers: Transferring Reasoning Capabilities to Smaller LLMs Using Multi-Teacher Knowledge Distillation · WSDM 2025
Machine learning › Trustworthy machine learning › adversarial machine learning
adversarial defense
0.812024
Mitigating Emergent Robustness Degradation while Scaling Graph Learning · ICLR 2024
Natural language and speech › Question answering and dialogue systems
knowledge base question answering
0.812024
G-Retriever: Retrieval-Augmented Generation for Textual Graph Understanding and Question Answering · NeurIPS 2024
Knowledge, reasoning and agents › Knowledge representation and reasoning
knowledge graph
0.812024
Graph Neural Prompting with Large Language Models · AAAI 2024
Natural language and speech › Language models and text generation › pre-trained language model › knowledge-enhanced pre-trained language model
knowledge graph-enhanced language model
0.812024
Graph Neural Prompting with Large Language Models · AAAI 2024
Knowledge, reasoning and agents › Knowledge representation and reasoning
knowledge graph reasoning
0.812024
G-Retriever: Retrieval-Augmented Generation for Textual Graph Understanding and Question Answering · NeurIPS 2024
Natural language and speech › Language models and text generation › large language model › large language model adaptation › personalization
LLM personalization
0.812024
Democratizing Large Language Models via Personalized Parameter-Efficient Fine-tuning · EMNLP 2024
Machine learning › Trustworthy machine learning
machine unlearning
0.812024
Breaking the Trilemma of Privacy, Utility, and Efficiency via Controllable Machine Unlearning · WWW 2024
Machine learning › Representation and self-supervised learning › representation learning › unsupervised representation learning › self-supervised representation learning
masked modeling
0.812024
MAPE-PPI: Towards Effective and Efficient Protein-Protein Interaction Prediction via Microenvironment-Aware Protein Embedding · ICLR 2024
Machine learning › Graph learning › graph neural network
node classification
0.812024
Graph Cross Supervised Learning via Generalized Knowledge · KDD 2024
Machine learning › Trustworthy machine learning
out-of-distribution generalization
0.812024
Graph Cross Supervised Learning via Generalized Knowledge · KDD 2024

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

graph neural network · 5.4mixture of experts · 2.2reinforcement learning · 2.0knowledge distillation · 1.3node classification · 1.1adversarial attack strategy · 1.1weakly supervised pre-finetuning · 1.0ranking-based reconstruction · 1.0pareto learning · 1.0multi-layer perceptron · 1.0hard negative mining · 1.0bootstrapping · 1.0bidirectional contextualization · 1.0adaptive feature masking · 1.0KL divergence constraint · 1.0unlearning proxy · 0.8self-supervised link prediction · 0.8pre-training · 0.8
YearPublicationVenuePosition
2026 Adaptive and Context-rich Generative Self-supervised Learning on Graphs
abstract
Generative self-supervised learning on graphs has emerged as a popular learning paradigm and demonstrated its efficacy in handling non-Euclidean data. However, several remaining issues limit the capability of existing methods: 1) the disregard of uneven node significance in masking, 2) the underutilization of holistic graph information, 3) the ignorance of semantic knowledge in the representation space due to the exclusive use of reconstruction loss in the output space, and 4) the unstable reconstructions caused by the large volume of masked contents. In light of this, we propose ACE-GSL, an adaptive and context-rich graph self-supervised learning framework to address these issues from the perspectives of adaptivity, integrity, complementarity, and consistency. Specifically, we first develop an adaptive feature mask generator to account for the unique significance of nodes and sample informative masks (adaptivity). We then design a ranking-based structure reconstruction objective joint with feature reconstruction to capture holistic graph information and emphasize the topological proximity between neighbors (integrity). After that, we present a bootstrapping-based similarity module to encode the high-level semantic knowledge in the representation space, complementary to the low-level reconstruction in the output space (complementarity). Finally, we build a consistency assurance module to provide reconstruction objectives with extra stabilized consistency targets (consistency). Extensive experiments demonstrate that ACE-GSL achieves state-of-the-art performance over 28 methods on 20 datasets across 3 tasks.
Yijun Tian 0001, Chuxu Zhang, Ziyi Kou, Zheyuan Liu 0010, Xiangliang Zhang 0001, Nitesh V. Chawla
AAAI1
2026 Pareto-Based Heterogeneous Knowledge Distillation for MLPs on Graphs
abstract
Heterogeneous Graph Neural Networks (HGNNs) have demonstrated remarkable capabilities in capturing effective information in heterogeneous graphs, achieving outstanding performance in various learning tasks. However, the heavy dependency of HGNNs on neighbors information may result in high latency, which restricts their practicality in real-world applications. Recent studies have attempted to overcome such latency in Graph Neural Networks (GNNs) by distilling knowledge into student models that do not rely on graph structure. But these approaches primarily focus on replicating teachers' predictive outcomes while neglecting the structural knowledge they encoded. This limitation makes such approach less effective when graphs become complex, particularly on heterogeneous graphs. Motivated by this challenge, we propose HGKD, a novel hierarchical knowledge distillation framework that transfers both structural knowledge and predictive outcomes from HGNN teachers to a multi-layer perceptron student. Additionally, we provide two variants of HGKD that help the student learn from multiple teacher models through Pareto learning and incorporate low-cost neighbor information. We evaluate HGKD and its variants on a range of heterogeneous graph datasets. The results demonstrate that our student model achieves performance comparable to or exceeding that of HGNN teachers, despite not relying on graph structures during inference.
Yijun Tian 0001, Zhichao Xu 0001, Chuxu Zhang
AAAI2
2026 Reinforcement Learning for Self-Improving Agent with Skill Library
abstract
Jiongxiao Wang, Qiaojing Yan, Yawei Wang, Yijun Tian, Soumya Smruti Mishra, Zhichao Xu, Megha Gandhi, Panpan Xu, Lin Lee Cheong. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026.
Jiongxiao Wang, Qiaojing Yan, Yijun Tian 0001, Soumya Smruti Mishra, Zhichao Xu 0001, Megha Gandhi, Lin Lee Cheong
ACL (1)4
2026 ALDEN: Reinforcement Learning for Active Navigation and Evidence Gathering in Long Documents
abstract
While Vision-language models (VLMs) interpret text-rich images effectively, they struggle with reasoning across long, multi-page documents.We present Active Long-DocumEnt Navigation (ALDEN), a multi-turn reinforcement learning framework that fine-tunes VLMs as interactive agents capable of actively navigating long, visually rich documents rather than passive readers.ALDEN features a novel fetch action that allows direct page indexing, complementing the classic search action and better exploiting document structure.To ensure training efficiency and stability, we introduce a rule-based cross-level reward for dense supervision and a visual-semantic anchoring mechanism using dual-path KL-divergence constraints.We train ALDEN on a curated corpus built from open-source datasets, filtering out trivial samples and rewriting queries to incentivize multi-turn navigation and fetch usage.Empirically, ALDEN achieves state-of-the-art results on five long-document benchmarks, offering a more accurate and efficient path for long-document understanding.All of our code and datasets will be made publicly available at Github 1 .
Terry Ruas, Yijun Tian 0001, Jan Philip Wahle, Daniel Kurzawe, Bela Gipp
ACL (1)3
2026 LACONIC: Dense-Level Effectiveness for Scalable Sparse Retrieval via a Two-Phase Training Curriculum
abstract
While dense retrieval models have been the standard for state-of-the-art information retrieval, their deployment is often constrained by high memory requirements and reliance on GPU accelerators for vector similarity search at scale. Learned sparse retrieval offers a compelling alternative by enabling efficient search via inverted indices, yet it has historically received less attention than dense approaches. In this paper, we introduce LACONIC, a family of learned sparse retrievers based on the Llama3 architecture (1B, 3B, and 8B). We propose a streamlined two-phase training curriculum consisting of (1) weakly supervised pre-finetuning to adapt causal LLMs for bidirectional contextualization and (2) high-signal finetuning using curated hard negatives. Our results demonstrate that LACONIC effectively bridges the performance gap with dense models: the 8B variant achieves a state-of-the-art 60.2 nDCG@10 on the MTEB Retrieval benchmark, ranking 15th on the leaderboard as of February 5th, 2026, while utilizing 74% less index memory than an equivalent dense model. By delivering high retrieval effectiveness on commodity CPU hardware with a fraction of the compute budget required by competing models, LACONIC provides a scalable and efficient solution for real-world search applications. We fully open source our code implementation and trained checkpoints to facilitate reproducibility.
Zhichao Xu 0001, Shengyao Zhuang, Xinyu Zhang 0018, Xueguang Ma, Yijun Tian 0001, Maitrey Mehta, Jimmy Lin, Vivek Srikumar
SIGIR5
2025 Towards Few-shot Chemical Reaction Outcome Prediction
abstract
Accurate chemical reaction prediction is essential for drug discovery and synthetic planning. However, this task becomes particularly challenging in low-data scenarios, where novel reaction types lack sufficient training examples. To address this challenge, we propose FewRxn, a novel model-agnostic few-shot reaction prediction framework that enables rapid adaptation to unseen reaction types using only a few training samples. FewRxn integrates several key innovations, including segmentation masks for enhanced reactant representation, fingerprint embeddings for richer molecular context, and task-aware meta-learning for effective knowledge transfer. Through extensive evaluations, FewRxn achieves state-of-the-art accuracy in few-shot settings, significantly outperforming traditional fine-tuning methods. Additionally, our work provides insights into the impact of molecular representations on reaction knowledge transfer, demonstrating that knowledge captured under molecular graph-based formulation consistently outperforms those learned in forms of SMILES generation in few-shot learning.
Yili Shen, Yijun Tian 0001, Cheng-Wei Ju, Olaf Wiest, Xiangliang Zhang 0001
CIKM2
2025 Beyond Answers: Transferring Reasoning Capabilities to Smaller LLMs Using Multi-Teacher Knowledge Distillation
abstract
Transferring the reasoning capability from stronger large language models (LLMs) to smaller ones has been quite appealing, as smaller LLMs are more flexible to deploy with less expense. Among the existing solutions, knowledge distillation stands out due to its outstanding efficiency and generalization. However, existing methods suffer from several drawbacks, including limited knowledge diversity and the lack of rich contextual information. To solve the problems and facilitate the learning of compact language models, we propose TinyLLM, a new knowledge distillation paradigm to learn a small student LLM from multiple large teacher LLMs. In particular, we encourage the student LLM to not only generate the correct answers but also understand the rationales behind these answers. Given that different LLMs possess diverse reasoning skills, we guide the student model to assimilate knowledge from various teacher LLMs. We further introduce an in-context example generator and a teacher-forcing Chain-of-Thought strategy to ensure that the rationales are accurate and grounded in contextually appropriate scenarios. Extensive experiments on six datasets across two reasoning tasks demonstrate the superiority of our method. Results show that TinyLLM can outperform large teacher LLMs significantly, despite a considerably smaller model size. The source code is available at: https://github.com/YikunHan42/TinyLLM.
Yijun Tian 0001, Yikun Han, Xiusi Chen, Wei Wang 0010, Nitesh V. Chawla
WSDM1
2025 Mixture of Spatial-Temporal Graph Transformer Networks for urban congestion prediction using multimodal transportation data
Jian Zhang 0088, Chuan-Zhi Xie, Yijun Tian 0001
Expert Syst. Appl.5
2024 Graph Neural Prompting with Large Language Models
abstract
Large language models (LLMs) have shown remarkable generalization capability with exceptional performance in various language modeling tasks. However, they still exhibit inherent limitations in precisely capturing and returning grounded knowledge. While existing work has explored utilizing knowledge graphs (KGs) to enhance language modeling via joint training and customized model architectures, applying this to LLMs is problematic owing to their large number of parameters and high computational cost. Therefore, how to enhance pre-trained LLMs using grounded knowledge, e.g., retrieval-augmented generation, remains an open question. In this work, we propose Graph Neural Prompting (GNP), a novel plug-and-play method to assist pre-trained LLMs in learning beneficial knowledge from KGs. GNP encompasses various designs, including a standard graph neural network encoder, a cross-modality pooling module, a domain projector, and a self-supervised link prediction objective. Extensive experiments on multiple datasets demonstrate the superiority of GNP on both commonsense and biomedical reasoning tasks across different LLM sizes and settings. Code is available at https://github.com/meettyj/GNP.
Yijun Tian 0001, Huan Song, Haozhu Wang, Ziqing Hu, Nitesh V. Chawla
AAAI1
2024 FaDE: A Face Segment Driven Identity Anonymization Framework For Fair Face Recognition
abstract
Current face recognition (FR) algorithms frequently encounter discrimination issues in terms of various attributes (e.g., gender, age) due to the biased demographic distribution of the training datasets towards specific groups. In this paper, we study an identity protected fair FR problem where the goal is to augment the datasets with external face images while ensuring the anonymity of the corresponding face identities. Our problem is motivated by the limitation of current fairness driven data augmentation approaches that directly utilize the external face images accessed by FR algorithm developers while ignoring the protection on the face identities of the image owners. To address the problem, we develop FaDE, a face segment driven identity anonymization framework that augments biased face image datasets by identifying specific face segments with diversified demographic characteristics from external face images but with least identity disclosure, and then reconstructing the segments to full face images with new identities. As a result, the augmented dataset is under a more balanced demographic distribution and improves the fairness performance of the optimized FR algorithms. We evaluate FaDE on two public face datasets, CelebA and LFW that suffer from various demographic imbalance. The results show that FaDE significantly enhances both fairness and accuracy performance of the optimized FR algorithms, while keeping effective anonymity for the identities of external face images.
Ziyi Kou, Yijun Tian 0001, Meng Jiang 0001, Xiangliang Zhang 0001
CIKM2
2024 ChefFusion: Multimodal Foundation Model Integrating Recipe and Food Image Generation
abstract
Significant work has been conducted in the domain of food computing, yet these studies typically focus on single tasks such as t2t (instruction generation from food titles and ingredients), i2t (recipe generation from food images), or t2i (food image generation from recipes). None of these approaches integrate all modalities simultaneously. To address this gap, we introduce a novel food computing foundation model that achieves true multimodality, encompassing tasks such as t2t, t2i, i2t, it2t, and t2ti. By leveraging large language models (LLMs) and pre-trained image encoder and decoder models, our model can perform a diverse array of food computing-related tasks, including food understanding, food recognition, recipe generation, and food image generation. Compared to previous models, our foundation model demonstrates a significantly broader range of capabilities and exhibits superior performance, particularly in food image generation and recipe generation tasks. We open-sourced ChefFusion at https://github.com/Peiyu-Georgia-Li/ChefFusion-Multimodal-Foundation-Model-Integrating-Recipe-and-Food-Image-Generation.git.
Xiaobao Huang, Yijun Tian 0001, Nitesh V. Chawla
CIKM3
2024 Democratizing Large Language Models via Personalized Parameter-Efficient Fine-tuning
abstract
Personalization in large language models (LLMs) is increasingly important, aiming to align the LLMs' interactions, content, and recommendations with individual user preferences.Recent advances have highlighted effective prompt design by enriching user queries with non-parametric knowledge through behavior history retrieval and textual profiles.However, these methods faced limitations due to a lack of model ownership, resulting in constrained customization and privacy issues, and often failed to capture complex, dynamic user behavior patterns.To address these shortcomings, we introduce One PEFT Per User (OPPU) 1 , employing personalized parameter-efficient finetuning (PEFT) modules to store user-specific behavior patterns and preferences.By plugging in personal PEFT parameters, users can own and use their LLMs individually.OPPU integrates parametric user knowledge in the personal PEFT parameters with non-parametric knowledge from retrieval and profiles, adapting LLMs to user behavior shifts.Experimental results demonstrate that OPPU significantly outperforms existing prompt-based methods across seven diverse tasks in the LaMP benchmark.Further studies reveal OPPU's enhanced capabilities in handling user behavior shifts, modeling users at different activity levels, maintaining robustness across various user history formats, and displaying versatility with different PEFT methods.
Zhaoxuan Tan, Qingkai Zeng 0001, Yijun Tian 0001, Zheyuan Liu 0010, Meng Jiang 0001
EMNLP3
2024 MAPE-PPI: Towards Effective and Efficient Protein-Protein Interaction Prediction via Microenvironment-Aware Protein Embedding
abstract
Protein-Protein Interactions (PPIs) are fundamental in various biological processes and play a key role in life activities. The growing demand and cost of experimental PPI assays require computational methods for efficient PPI prediction. While existing methods rely heavily on protein sequence for PPI prediction, it is the protein structure that is the key to determine the interactions. To take both protein modalities into account, we define the microenvironment of an amino acid residue by its sequence and structural contexts, which describe the surrounding chemical properties and geometric features. In addition, microenvironments defined in previous work are largely based on experimentally assayed physicochemical properties, for which the "vocabulary" is usually extremely small. This makes it difficult to cover the diversity and complexity of microenvironments. In this paper, we propose Microenvironment-Aware Protein Embedding for PPI prediction (MPAE-PPI), which encodes microenvironments into chemically meaningful discrete codes via a sufficiently large microenvironment "vocabulary" (i.e., codebook). Moreover, we propose a novel pre-training strategy, namely Masked Codebook Modeling (MCM), to capture the dependencies between different microenvironments by randomly masking the codebook and reconstructing the input. With the learned microenvironment codebook, we can reuse it as an off-the-shelf tool to efficiently and effectively encode proteins of different sizes and functions for large-scale PPI prediction. Extensive experiments show that MAPE-PPI can scale to PPI prediction with millions of PPIs with superior trade-offs between effectiveness and computational efficiency than the state-of-the-art competitors.
Lirong Wu, Yijun Tian 0001, Yufei Huang 0002, Siyuan Li 0002, Nitesh V. Chawla, Stan Z. Li
ICLR2
2024 Mitigating Emergent Robustness Degradation while Scaling Graph Learning
abstract
Although graph neural networks have exhibited remarkable performance in various graph tasks, a significant concern is their vulnerability to adversarial attacks. Consequently, many defense methods have been proposed to alleviate the deleterious effects of adversarial attacks and learn robust graph representations. However, most of them are difficult to *simultaneously* avoid two major limitations: (i) an emergent and severe degradation in robustness when exposed to very intense attacks, and (ii) heavy computation complexity hinders them from scaling to large graphs. In response to these challenges, we introduce an innovative graph defense method for unpredictable real-world scenarios by *designing a graph robust learning framework that is resistant to robustness degradation* and *refraining from unscalable designs with heavy computation*: specifically, our method employs a denoising module, which eliminates edges that are associated with attacked nodes to reconstruct a cleaner graph; Then, it applies Mixture-of-Experts to select differentially private noises with varying magnitudes to counteract the hidden features attacked at different intensities toward robust predictions; Moreover, our overall design avoids the reliance on heavy adjacency matrix computations, such as SVD, thus facilitating its applicability even on large graphs. Comprehensive experiments have been conducted to demonstrate the anti-degraded robustness and scalability of our method, as compared to popular graph adversarial learning methods, under diverse attack intensities and various datasets of different sizes.
Xiangchi Yuan, Yijun Tian 0001, Yanfang Ye 0001, Chuxu Zhang
ICLR3
2024 S3GCL: Spectral, Swift, Spatial Graph Contrastive Learning
abstract
Graph Contrastive Learning (GCL) has emerged as a highly effective self-supervised approach in graph representation learning. However, prevailing GCL methods confront two primary challenges: 1) They predominantly operate under homophily assumptions, focusing on low-frequency signals in node features while neglecting heterophilic edges that connect nodes with dissimilar features. 2) Their reliance on neighborhood aggregation for inference leads to scalability challenges and hinders deployment in real-time applications. In this paper, we introduce S3GCL, an innovative framework designed to tackle these challenges. Inspired by spectral GNNs, we initially demonstrate the correlation between frequency and homophily levels. Then, we propose a novel cosine-parameterized Chebyshev polynomial as low/high-pass filters to generate biased graph views. To resolve the inference dilemma, we incorporate an MLP encoder and enhance its awareness of graph context by introducing structurally and semantically neighboring nodes as positive pairs in the spatial domain. Finally, we formulate a cross-pass GCL objective between full-pass MLP and biased-pass GNN filtered features, eliminating the need for augmentation. Extensive experiments on real-world tasks validate S3GCL proficiency in generalization to diverse homophily levels and its superior inference efficiency.
Guancheng Wan, Yijun Tian 0001, Wenke Huang 0003, Nitesh V. Chawla, Mang Ye
ICML2
2024 Learning to Predict Mutational Effects of Protein-Protein Interactions by Microenvironment-aware Hierarchical Prompt Learning
abstract
Protein-protein bindings play a key role in a variety of fundamental biological processes, and thus predicting the effects of amino acid mutations on protein-protein binding is crucial. To tackle the scarcity of annotated mutation data, pre-training with massive unlabeled data has emerged as a promising solution. However, this process faces a series of challenges: (1) complex higher-order dependencies among multiple (more than paired) structural scales have not yet been fully captured; (2) it is rarely explored how mutations alter the local conformation of the surrounding microenvironment; (3) pre-training is costly, both in data size and computational burden. In this paper, we first construct a hierarchical prompt codebook to record common microenvironmental patterns at different structural scales independently. Then, we develop a novel codebook pre-training task, namely masked microenvironment modeling, to model the joint distribution of each mutation with their residue types, angular statistics, and local conformational changes in the microenvironment. With the constructed prompt codebook, we encode the microenvironment around each mutation into multiple hierarchical prompts and combine them to flexibly provide information to wild-type and mutated protein complexes about their microenvironmental differences. Such a hierarchical prompt learning framework has demonstrated superior performance and training efficiency over state-of-the-art pre-training-based methods in mutation effect prediction and a case study of optimizing human antibodies against SARS-CoV-2.
Lirong Wu, Yijun Tian 0001, Yufei Huang 0002, Siyuan Li 0002, Nitesh V. Chawla, Stan Z. Li
ICML2
2024 Graph Cross Supervised Learning via Generalized Knowledge
abstract
The success of GNNs highly relies on the accurate labeling of data. Existing methods of ensuring accurate labels, such as weakly-supervised learning, mainly focus on the existing nodes in the graphs. However, in reality, new nodes always continuously emerge on dynamic graphs, with different categories and even label noises. To this end, we formulate a new problem, Graph Cross-Supervised Learning, or Graph Weak-Shot Learning, that describes the challenges of modeling new nodes with novel classes and potential label noises. To solve this problem, we propose Lipshitz-regularized Mixture-of-Experts similarity network (LIME), a novel framework to encode new nodes and handle label noises. Specifically, we first design a node similarity network to capture the knowledge from the original classes, aiming to obtain insights for the emerging novel classes. Then, to enhance the similarity network's generalization to new nodes that could have a distribution shift, we employ the Mixture-of-Experts technique to increase the generalization of knowledge learned by the similarity network. To further avoid losing generalization ability during training, we introduce the Lipschitz bound to stabilize model output and alleviate the distribution shift issue. Empirical experiments validate LIME's effectiveness: we observe a substantial enhancement of up to 11.34% in node classification accuracy compared to the backbone model when subjected to the challenges of label noise on novel classes across five benchmark datasets. The code can be accessed through https://github.com/xiangchi-yuan/Graph-Cross-Supervised-Learning.
Xiangchi Yuan, Yijun Tian 0001, Yanfang Ye 0001, Nitesh V. Chawla, Chuxu Zhang
KDD2
2024 G-Retriever: Retrieval-Augmented Generation for Textual Graph Understanding and Question Answering
abstract
Given a graph with textual attributes, we enable users to `chat with their graph': that is, to ask questions about the graph using a conversational interface. In response to a user's questions, our method provides textual replies and highlights the relevant parts of the graph. While existing works integrate large language models (LLMs) and graph neural networks (GNNs) in various ways, they mostly focus on either conventional graph tasks (such as node, edge, and graph classification), or on answering simple graph queries on small or synthetic graphs. In contrast, we develop a flexible question-answering framework targeting real-world textual graphs, applicable to multiple applications including scene graph understanding, common sense reasoning, and knowledge graph reasoning. Toward this goal, we first develop a Graph Question Answering (GraphQA) benchmark with data collected from different tasks. Then, we propose our \textit{G-Retriever} method, introducing the first retrieval-augmented generation (RAG) approach for general textual graphs, which can be fine-tuned to enhance graph understanding via soft prompting. To resist hallucination and to allow for textual graphs that greatly exceed the LLM's context window size, \textit{G-Retriever} performs RAG over a graph by formulating this task as a Prize-Collecting Steiner Tree optimization problem. Empirical evaluations show that our method outperforms baselines on textual graph tasks from multiple domains, scales well with larger graph sizes, and mitigates hallucination.~\footnote{Our codes and datasets are available at: \url{https://github.com/XiaoxinHe/G-Retriever}}
Xiao-Xin He, Yijun Tian 0001, Yifei Sun 0002, Nitesh V. Chawla, Thomas Laurent 0001, Yann LeCun, Xavier Bresson, Bryan Hooi
NeurIPS2
2024 Breaking the Trilemma of Privacy, Utility, and Efficiency via Controllable Machine Unlearning
abstract
Machine Unlearning (MU) algorithms have become increasingly critical due to the imperative adherence to data privacy regulations.The primary objective of MU is to erase the influence of specific data samples on a given model without the need to retrain it from scratch.Accordingly, existing methods focus on maximizing user privacy protection.However, there are different degrees of privacy regulations for each real-world web-based application.Exploring the full spectrum of trade-offs between privacy, model utility, and runtime efficiency is critical for practical unlearning scenarios.Furthermore, designing the MU algorithm with simple control of the aforementioned trade-off is desirable but challenging due to the inherent complex interaction.To address the challenges, we present Controllable Machine Unlearning (ConMU), a novel framework designed to facilitate the calibration of MU.The ConMU framework contains three integral modules: an important data selection module that reconciles the runtime efficiency and model generalization, a progressive Gaussian mechanism module that balances privacy and model generalization, and an unlearning proxy that controls the trade-offs between privacy and runtime efficiency.Comprehensive experiments on various benchmark datasets have demonstrated the robust adaptability of our control mechanism and its superiority over established unlearning methods.ConMU explores the full spectrum of the Privacy-Utility-Efficiency trade-off and allows practitioners to account for different real-world regulations.
Zheyuan Liu 0010, Guangyao Dou, Eli Chien, Yijun Tian 0001, Ziwei Zhu 0001
WWW5
2023 Heterogeneous Graph Masked Autoencoders
abstract
Generative self-supervised learning (SSL), especially masked autoencoders, has become one of the most exciting learning paradigms and has shown great potential in handling graph data. However, real-world graphs are always heterogeneous, which poses three critical challenges that existing methods ignore: 1) how to capture complex graph structure? 2) how to incorporate various node attributes? and 3) how to encode different node positions? In light of this, we study the problem of generative SSL on heterogeneous graphs and propose HGMAE, a novel heterogeneous graph masked autoencoder model to address these challenges. HGMAE captures comprehensive graph information via two innovative masking techniques and three unique training strategies. In particular, we first develop metapath masking and adaptive attribute masking with dynamic mask rate to enable effective and stable learning on heterogeneous graphs. We then design several training strategies including metapath-based edge reconstruction to adopt complex structural information, target attribute restoration to incorporate various node attributes, and positional feature prediction to encode node positional information. Extensive experiments demonstrate that HGMAE outperforms both contrastive and generative state-of-the-art baselines on several tasks across multiple datasets. Codes are available at https://github.com/meettyj/HGMAE.
Yijun Tian 0001, Kaiwen Dong, Chuxu Zhang, Nitesh V. Chawla
AAAI1
2023 Boosting Graph Neural Networks via Adaptive Knowledge Distillation
abstract
Graph neural networks (GNNs) have shown remarkable performance on diverse graph mining tasks. While sharing the same message passing framework, our study shows that different GNNs learn distinct knowledge from the same graph. This implies potential performance improvement by distilling the complementary knowledge from multiple models. However, knowledge distillation (KD) transfers knowledge from high-capacity teachers to a lightweight student, which deviates from our scenario: GNNs are often shallow. To transfer knowledge effectively, we need to tackle two challenges: how to transfer knowledge from compact teachers to a student with the same capacity; and, how to exploit student GNN's own learning ability. In this paper, we propose a novel adaptive KD framework, called BGNN, which sequentially transfers knowledge from multiple GNNs into a student GNN. We also introduce an adaptive temperature module and a weight boosting module. These modules guide the student to the appropriate knowledge for effective learning. Extensive experiments have demonstrated the effectiveness of BGNN. In particular, we achieve up to 3.05% improvement for node classification and 6.35% improvement for graph classification over vanilla GNNs.
Zhichun Guo, Yujie Fan, Yijun Tian 0001, Chuxu Zhang, Nitesh V. Chawla
AAAI4
2023 Learning MLPs on Graphs: A Unified View of Effectiveness, Robustness, and Efficiency
Yijun Tian 0001, Chuxu Zhang, Zhichun Guo, Xiangliang Zhang 0001, Nitesh V. Chawla
ICLR1
2023 Chasing All-Round Graph Representation Robustness: Model, Training, and Optimization
Yijun Tian 0001, Mingxuan Ju, Zheyuan Liu 0010, Yanfang Ye 0001, Nitesh V. Chawla, Chuxu Zhang
ICLR2
2023 When Sparsity Meets Contrastive Models: Less Graph Data Can Bring Better Class-Balanced Representations
abstract
Graph Neural Networks (GNNs) are powerful models for non-Euclidean data, but their training is often accentuated by massive unnecessary computation: on the one hand, training on non-Euclidean data has relatively high computational cost due to its irregular density properties; on the other hand, the class imbalance property often associated with non-Euclidean data cannot be alleviated by the massiveness of the data, thus hindering the generalisation of the models. To address the above issues, theoretically, we start with a hypothesis about the effectiveness of using a subset of training data for GNNs, which is guaranteed by the gradient distance between the subset and the full set. Empirically, we also observe that a subset of the data can provide informative gradients for model optimization and which changes over time dynamically. We name this phenomenon dynamic data sparsity. Additionally, we find that pruned sparse contrastive models may miss valuable information, leading to a large loss value on the informative subset. Motivated by the above findings, we develop a unified data model dynamic sparsity framework called Data Decantation (DataDec) to address the above challenges. The key idea of DataDec is to identify the informative subset dynamically during the training process by applying sparse graph contrastive learning. The effectiveness of DataDec is comprehensively evaluated on graph benchmark datasets and we also verify its generalizability on image data.
Chao Huang 0001, Yijun Tian 0001, Qianlong Wen, Zhongyu Ouyang, Youhuan Li, Yanfang Ye 0001, Chuxu Zhang
ICML3
2023 Graph-based Molecular Representation Learning
abstract
Molecular representation learning (MRL) is a key step to build the connection between machine learning and chemical science. In particular, it encodes molecules as numerical vectors preserving the molecular structures and features, on top of which the downstream tasks (e.g., property prediction) can be performed. Recently, MRL has achieved considerable progress, especially in methods based on deep molecular graph learning. In this survey, we systematically review these graph-based molecular representation techniques, especially the methods incorporating chemical domain knowledge. Specifically, we first introduce the features of 2D and 3D molecular graphs. Then we summarize and categorize MRL methods into three groups based on their input. Furthermore, we discuss some typical chemical applications supported by MRL. To facilitate studies in this fast-developing area, we also list the benchmarks and commonly used datasets in the paper. Finally, we share our thoughts on future research directions.
Zhichun Guo, Kehan Guo, Bozhao Nan, Yijun Tian 0001, Roshni G. Iyer, Yihong Ma, Olaf Wiest, Xiangliang Zhang 0001, Wei Wang 0010, Chuxu Zhang, Nitesh V. Chawla
IJCAI4
2023 Character As Pixels: A Controllable Prompt Adversarial Attacking Framework for Black-Box Text Guided Image Generation Models
abstract
In this paper, we study a controllable prompt adversarial attacking problem for text guided image generation (Text2Image) models in the black-box scenario, where the goal is to attack specific visual subjects (e.g., changing a brown dog to white) in a generated image by slightly, if not imperceptibly, perturbing the characters of the driven prompt (e.g., ``brown'' to ``br0wn''). Our study is motivated by the limitations of current Text2Image attacking approaches that still rely on manual trials to create adversarial prompts. To address such limitations, we develop CharGrad, a character-level gradient based attacking framework that replaces specific characters of a prompt with pixel-level similar ones by interactively learning the perturbation direction for the prompt and updating the attacking examiner for the generated image based on a novel proxy perturbation representation for characters. We evaluate CharGrad using the texts from two public image captioning datasets. Results demonstrate that CharGrad outperforms existing text adversarial attacking approaches on attacking various subjects of generated images by black-box Text2Image models in a more effective and efficient way with less perturbation on the characters of the prompts.
Ziyi Kou, Shichao Pei, Yijun Tian 0001, Xiangliang Zhang 0001
IJCAI3
2023 Fair Graph Representation Learning via Diverse Mixture-of-Experts
abstract
Graph Neural Networks (GNNs) have demonstrated a great representation learning capability on graph data and have been utilized in various downstream applications. However, real-world data in web-based applications (e.g., recommendation and advertising) always contains bias, preventing GNNs from learning fair representations. Although many works were proposed to address the fairness issue, they suffer from the significant problem of insufficient learnable knowledge with limited attributes after debiasing. To address this problem, we develop Graph-Fairness Mixture of Experts (G-Fame), a novel plug-and-play method to assist any GNNs to learn distinguishable representations with unbiased attributes. Furthermore, based on G-Fame, we propose G-Fame++, which introduces three novel strategies to improve the representation fairness from node representations, model layer, and parameter redundancy perspectives. In particular, we first present the embedding diversified method to learn distinguishable node representations. Second, we design the layer diversified strategy to maximize the output difference of distinct model layers. Third, we introduce the expert diversified method to minimize expert parameter similarities to learn diverse and complementary representations. Extensive experiments demonstrate the superiority of G-Fame and G-Fame++ in both accuracy and fairness, compared to state-of-the-art methods across multiple graph datasets.
Zheyuan Liu 0010, Yijun Tian 0001, Erchi Zhang, Chao Huang 0001, Yanfang Ye 0001, Chuxu Zhang
WWW3
2022 Hierarchical Spatio-Temporal Graph Neural Networks for Pandemic Forecasting
abstract
The spread of COVID-19 throughout the world has led to cataclysmic consequences on the global community, which poses an urgent need to accurately understand and predict the trajectories of the pandemic. Existing research has relied on graph-structured human mobility data for the task of pandemic forecasting. To perform pandemic forecasting of COVID-19 in the United States, we curate Large-MG, a large-scale mobility dataset that contains 66 dynamic mobility graphs, with each graph having over 3k nodes and an average of 540k edges. One drawback with existing Graph Neural Networks (GNNs) for pandemic forecasting is that they generally perform information propagation in a flat way and thus ignore the inherent community structure in a mobility graph. To bridge this gap, we propose a Hierarchical Spatio-Temporal Graph Neural Network (HiSTGNN) to perform pandemic forecasting, which learns both spatial and temporal information from a sequence of dynamic mobility graphs. HiSTGNN consists of two network architectures. One is a hierarchical graph neural network (HiGNN) that constructs a two-level neural architecture: county-level and region-level, and performs information propagation in a hierarchical way. The other network architecture is a Transformer-based model that captures the temporal dynamics among the sequence of learned node representations from HiGNN. Additionally, we introduce a joint learning objective to further optimize HiSTGNN. Extensive experiments have demonstrated HiSTGNN's superior predictive power of COVID-19 new case/death counts compared with state-of-the-art baselines.
Yihong Ma, Patrick Gérard, Yijun Tian 0001, Zhichun Guo, Nitesh V. Chawla
CIKM3
2022 RecipeRec: A Heterogeneous Graph Learning Model for Recipe Recommendation
abstract
Recipe recommendation systems play an essential role in helping people decide what to eat. Existing recipe recommendation systems typically focused on content-based or collaborative filtering approaches, ignoring the higher-order collaborative signal such as relational structure information among users, recipes and food items. In this paper, we formalize the problem of recipe recommendation with graphs to incorporate the collaborative signal into recipe recommendation through graph modeling. In particular, we first present URI-Graph, a new and large-scale user-recipe-ingredient graph. We then propose RecipeRec, a novel heterogeneous graph learning model for recipe recommendation. The proposed model can capture recipe content and collaborative signal through a heterogeneous graph neural network with hierarchical attention and an ingredient set transformer. We also introduce a graph contrastive augmentation strategy to extract informative graph knowledge in a self-supervised manner. Finally, we design a joint objective function of recommendation and contrastive learning to optimize the model. Extensive experiments demonstrate that RecipeRec outperforms state-of-the-art methods for recipe recommendation. Dataset and codes are available at https://github.com/meettyj/RecipeRec.
Yijun Tian 0001, Chuxu Zhang, Zhichun Guo, Chao Huang 0001, Ronald A. Metoyer, Nitesh V. Chawla
IJCAI1
2022 Recipe2Vec: Multi-modal Recipe Representation Learning with Graph Neural Networks
abstract
Learning effective recipe representations is essential in food studies. Unlike what has been developed for image-based recipe retrieval or learning structural text embeddings, the combined effect of multi-modal information (i.e., recipe images, text, and relation data) receives less attention. In this paper, we formalize the problem of multi-modal recipe representation learning to integrate the visual, textual, and relational information into recipe embeddings. In particular, we first present Large-RG, a new recipe graph data with over half a million nodes, making it the largest recipe graph to date. We then propose Recipe2Vec, a novel graph neural network based recipe embedding model to capture multi-modal information. Additionally, we introduce an adversarial attack strategy to ensure stable learning and improve performance. Finally, we design a joint objective function of node classification and adversarial learning to optimize the model. Extensive experiments demonstrate that Recipe2Vec outperforms state-of-the-art baselines on two classic food study tasks, i.e., cuisine category classification and region prediction. Dataset and codes are available at https://github.com/meettyj/Recipe2Vec.
Yijun Tian 0001, Chuxu Zhang, Zhichun Guo, Yihong Ma, Ronald A. Metoyer, Nitesh V. Chawla
IJCAI1
2021 Recipe Representation Learning with Networks
abstract
Learning effective representations for recipes is essential in food studies for recommendation, classification, and other applications. Unlike what has been developed for learning textual or cross-modal embeddings for recipes, the structural relationship among recipes and food items are less explored. In this paper, we formalize the problem recipe representation learning with networks to involve both the textual feature and the structural relational feature into recipe representations. Specifically, we first present RecipeNet, a new and large-scale corpus of recipe data to facilitate network based food studies and recipe representation learning research. We then propose a novel heterogeneous recipe network embedding model, rn2vec, to learn recipe representations. The proposed model is able to capture textual, structural, and nutritional information through several neural network modules, including textual CNN, inner-ingredients transformer, and a graph neural network with hierarchical attention. We further design a combined objective function of node classification and link prediction to jointly optimize the model. The extensive experiments show that our model outperforms state-of-the-art baselines on two classic food study tasks. Dataset and codes are available at https://github.com/meettyj/rn2vec.
Yijun Tian 0001, Chuxu Zhang, Ronald A. Metoyer, Nitesh V. Chawla
CIKM1
2020 Quasi-Experimental Designs for Assessing Response on Social Media to Policy Changes
Yijun Tian 0001, Rumi Chunara
ICWSM1