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

Xiang Li 0067

dblp:40/1491-67 · DBLP profile ↗
← Back
109ranked-venue papers
16as first author
93since 2021 · last 2026
—ORCID · conflict

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

Artificial intelligence and machine learning · 58 · 8 first-author · 52 since 2021Databases, data management, data science and information retrieval · 58 · 10 first-author · 45 since 2021Applied, interdisciplinary, general and emerging computing · 12 · 4 first-author · 8 since 2021Graphics, computer vision, multimedia, augmented reality and games · 10 · 1 first-author · 9 since 2021Systems, architecture and hardware · 4 · 4 since 2021Computer networks · 4 · 4 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
46 papers
Graph learning · 39% Efficient and distributed learning · 25% Language models and text generation · 8%
Databases, data mining, and information retrieval
21 papers
Recommender systems · 41% Data mining · 25% Spatial and temporal data management · 14%
Computer architecture, parallel and distributed computing, and storage systems
8 papers
Cloud and datacenter computing · 25% High-performance computing · 25% Performance modeling and evaluation · 18%

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

TopicWeightPapersLastEvidence papers
Machine learning › Graph learning
graph neural network
10.2152025
RELIEF: Reinforcement Learning Empowered Graph Feature Prompt Tuning · KDD (1) 2025
Learning Prioritized Node-Wise Message Propagation in Graph Neural Networks (Extended Abstract) · ICDE 2025
Can Large Language Models Act as Ensembler for Multi-GNNs? · EMNLP 2025
Machine learning › Efficient and distributed learning
distributed training
3.842026
MTGenRec: An Efficient Distributed Training System for Generative Recommendation Models in Meituan · KDD (1) 2026
Laminar: A Scalable Asynchronous RL Post-Training Framework · EuroSys 2026
MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production · EuroSys 2026
Machine learning › Efficient and distributed learning
federated learning
2.842024
FwdLLM: Efficient Federated Finetuning of Large Language Models with Perturbed Inferences · USENIX ATC 2024
DFDG: Data-Free Dual-Generator Adversarial Distillation for One-Shot Federated Learning · ICDM 2024
DFRD: Data-Free Robustness Distillation for Heterogeneous Federated Learning · NeurIPS 2023
Machine learning › Graph learning › graph neural network
node classification
2.232025
RELIEF: Reinforcement Learning Empowered Graph Feature Prompt Tuning · KDD (1) 2025
Leveraging Large Language Models for Node Generation in Few-Shot Learning on Text-Attributed Graphs · AAAI 2025
Leveraging Meta-path Contexts for Classification in Heterogeneous Information Networks · ICDE 2021
Data mining
clustering
2.052022
SCHAIN-IRAM: An Efficient and Effective Semi-Supervised Clustering Algorithm for Attributed Heterogeneous Information Networks · IEEE Trans. Knowl. Data Eng. 2022
CAST: A Correlation-based Adaptive Spectral Clustering Algorithm on Multi-scale Data · KDD 2020
Spectral Clustering in Heterogeneous Information Networks · AAAI 2019
Machine learning › Representation and self-supervised learning
contrastive learning
2.032024
Heterogeneous Graph Contrastive Learning With Meta-Path Contexts and Adaptively Weighted Negative Samples · IEEE Trans. Knowl. Data Eng. 2024
Graph Self-Contrast Representation Learning · ICDM 2023
CGMN: A Contrastive Graph Matching Network for Self-Supervised Graph Similarity Learning · IJCAI 2022
Machine learning › Graph learning › graph neural network
scalable graph neural network
2.032024
Scalable decoupling graph neural network with feature-oriented optimization · VLDB J. 2024
LD2: Scalable Heterophilous Graph Neural Network with Decoupled Embeddings · NeurIPS 2023
SCARA: Scalable Graph Neural Networks with Feature-Oriented Optimization · Proc. VLDB Endow. 2022
Machine learning › Graph learning › graph neural network
heterophily
1.832023
LD2: Scalable Heterophilous Graph Neural Network with Decoupled Embeddings · NeurIPS 2023
RAW-GNN: RAndom Walk Aggregation based Graph Neural Network · IJCAI 2022
Finding Global Homophily in Graph Neural Networks When Meeting Heterophily · ICML 2022
Edge and fog computing › edge server
edge server architecture
1.822026
Prototyping and Analyzing Mobile SoC Clusters as Modern Edge Servers · IEEE Trans. Mob. Comput. 2026
More is Different: Prototyping and Analyzing a New Form of Edge Server with Massive Mobile SoCs · USENIX ATC 2024
Machine learning › Graph learning
graph representation learning
1.732025
Hierarchical Vector Quantized Graph Autoencoder with Annealing-Based Code Selection · WWW 2025
Graph Self-Contrast Representation Learning · ICDM 2023
LD2: Scalable Heterophilous Graph Neural Network with Decoupled Embeddings · NeurIPS 2023
Machine learning › Trustworthy machine learning
robustness
1.732025
Resurrecting Label Propagation for Graphs with Heterophily and Label Noise · KDD 2024
DFRD: Data-Free Robustness Distillation for Heterogeneous Federated Learning · NeurIPS 2023
Text Detoxification: Data Efficiency, Semantic Preservation and Model Generalization · EMNLP 2025
Recommender systems
conversion rate prediction
1.722026
Hierarchical Denoising Entire Space Multi-Task Model for Post-Click Conversion Rate Prediction with Noisy Labels · SIGIR 2026
CEC: Towards Learning Global Optimized Recommendation through Causality Enhanced Conversion Model · SIGIR 2023
Machine learning › Graph learning
network embedding
1.632025
Learning Prioritized Node-Wise Message Propagation in Graph Neural Networks (Extended Abstract) · ICDE 2025
Finding Global Homophily in Graph Neural Networks When Meeting Heterophily · ICML 2022
LD2: Scalable Heterophilous Graph Neural Network with Decoupled Embeddings · NeurIPS 2023
Machine learning › Graph learning
graph autoencoder
1.522025
Hierarchical Vector Quantized Graph Autoencoder with Annealing-Based Code Selection · WWW 2025
SeeGera: Self-supervised Semi-implicit Graph Variational Auto-encoders with Masking · WWW 2023
Machine learning › Graph learning
graph self-supervised learning
1.522025
Hierarchical Vector Quantized Graph Autoencoder with Annealing-Based Code Selection · WWW 2025
SeeGera: Self-supervised Semi-implicit Graph Variational Auto-encoders with Masking · WWW 2023
Machine learning › Graph learning › graph autoencoder
variational graph autoencoder
1.522025
Variational Graph Autoencoder for Heterogeneous Information Networks with Missing and Inaccurate Attributes · KDD (1) 2025
SeeGera: Self-supervised Semi-implicit Graph Variational Auto-encoders with Masking · WWW 2023
Machine learning › Graph learning › graph neural network
message passing
1.422025
Learning Prioritized Node-Wise Message Propagation in Graph Neural Networks (Extended Abstract) · ICDE 2025
RAW-GNN: RAndom Walk Aggregation based Graph Neural Network · IJCAI 2022
Machine learning › Efficient and distributed learning › model compression › knowledge distillation
data-free knowledge distillation
1.422024
DFDG: Data-Free Dual-Generator Adversarial Distillation for One-Shot Federated Learning · ICDM 2024
DFRD: Data-Free Robustness Distillation for Heterogeneous Federated Learning · NeurIPS 2023
Machine learning › Representation and self-supervised learning › contrastive learning
graph contrastive learning
1.422024
Heterogeneous Graph Contrastive Learning With Meta-Path Contexts and Adaptively Weighted Negative Samples · IEEE Trans. Knowl. Data Eng. 2024
Graph Self-Contrast Representation Learning · ICDM 2023
Machine learning › Graph learning
heterogeneous graph
1.422025
Variational Graph Autoencoder for Heterogeneous Information Networks with Missing and Inaccurate Attributes · KDD (1) 2025
Leveraging Meta-path Contexts for Classification in Heterogeneous Information Networks · ICDE 2021
Machine learning › Efficient and distributed learning
on-device inference
1.322024
Mobile Foundation Model as Firmware · MobiCom 2024
A Comprehensive Benchmark of Deep Learning Libraries on Mobile Devices · WWW 2022
Machine learning › Trustworthy machine learning
interpretability
1.222023
Explaining Temporal Graph Models through an Explorer-Navigator Framework · ICLR 2023
Reinforcement Learning Enhanced Explainer for Graph Neural Networks · NeurIPS 2021
Recommender systems
causal recommendation
1.222023
CEC: Towards Learning Global Optimized Recommendation through Causality Enhanced Conversion Model · SIGIR 2023
Disentangling User Interest and Conformity for Recommendation with Causal Embedding · WWW 2021
Data mining › clustering
spectral clustering
1.132020
CAST: A Correlation-based Adaptive Spectral Clustering Algorithm on Multi-scale Data · KDD 2020
Spectral Clustering in Heterogeneous Information Networks · AAAI 2019
ROSC: Robust Spectral Clustering on Multi-scale Data · WWW 2018
Machine learning › Efficient and distributed learning › model compression › lightweight neural network
small language models
1.122025
Demystifying Small Language Models for Edge Deployment · ACL (1) 2025
DialCoT Meets PPO: Decomposing and Exploring Reasoning Paths in Smaller Language Models · EMNLP 2023
Cloud and datacenter computing › cluster resource management and scheduling
cluster resource management
1.122026
MuxFlow: efficient GPU sharing in production-level clusters with more than 10000 GPUs · Sci. China Inf. Sci. 2024
MTGenRec: An Efficient Distributed Training System for Generative Recommendation Models in Meituan · KDD (1) 2026
Machine learning › Efficient and distributed learning › distributed training
communication-efficient training
1.012026
MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production · EuroSys 2026
Machine learning › Efficient and distributed learning › inference efficiency
inference optimization
1.012026
SwiftEP: Accelerating MoE Inference with Buffer Fusion and TMA Offloading · NSDI 2026
Natural language and speech › Question answering and dialogue systems
knowledge base question answering
1.012026
Human Cognition Inspired RAG with Knowledge Graph for Complex Problem Solving · AAAI 2026
Machine learning › Efficient and distributed learning › distributed training › large model training
mixture-of-experts training
1.012026
MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production · EuroSys 2026

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

graph neural network · 2.3chain-of-thought prompting · 2.2reinforcement learning · 2.0trace analysis · 2.0feature ID deduplication · 2.0dynamic sequence balancing · 2.0dynamic hash table · 2.0dual-process verification · 2.0cross-platform benchmarking · 2.0large language model · 1.5knowledge distillation · 1.5combinatorial optimization · 1.4meta-path · 1.3causal inference · 1.2attention mechanism · 1.2contrastive learning · 1.1sharpe ratio · 1.0multi-task learning · 1.0
YearPublicationVenuePosition
2026 Human Cognition Inspired RAG with Knowledge Graph for Complex Problem Solving
abstract
Large Language Models (LLMs) have demonstrated significant potential across various domains. However, they often struggle with integrating external knowledge and performing complex reasoning, leading to hallucinations and unreliable outputs. Retrieval Augmented Generation (RAG) has emerged as a promising paradigm to mitigate these issues by incorporating external knowledge. Yet, conventional RAG approaches, especially those based on vector similarity, fail to effectively capture relational dependencies and support multi-step reasoning. In this work, we propose CogGRAG, a human cognition-inspired, graph-based RAG framework designed for Knowledge Graph Question Answering (KGQA). CogGRAG models the reasoning process as a tree-structured mind map that decomposes the original problem into interrelated subproblems and explicitly encodes their semantic relationships. This structure not only provides a global view to guide subsequent retrieval and reasoning but also enables self-consistent verification across reasoning paths. The framework operates in three stages: (1) top-down problem decomposition via mind map construction, (2) structured retrieval of both local and global knowledge from external Knowledge Graphs (KGs), and (3) bottom-up reasoning with dual-process self-verification. Unlike previous tree-based decomposition methods such as MindMap or Graph-CoT, CogGRAG unifies problem decomposition, knowledge retrieval, and reasoning under a single graph-structured cognitive framework, allowing early integration of relational knowledge and adaptive verification. Extensive experiments demonstrate that CogGRAG achieves superior accuracy and reliability compared to existing methods.
Yao Cheng 0009, Yibo Zhao 0005, Jiapeng Zhu 0002, Xiang Li 0067
AAAI6
2026 MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production
abstract
We present MegaScale-MoE, a production system tailored for the efficient training of large-scale mixture-of-experts (MoE) models. MoE emerges as a promising architecture to scale large language models (LLMs) to unprecedented sizes, thereby enhancing model performance. However, existing MoE training systems experience a degradation in training efficiency, exacerbated by the escalating scale of MoE models and the continuous evolution of hardware.
Chao Jin 0007, Ziheng Jiang, Zhihao Bai, Juncai Liu, Xiang Li 0067, Ningxin Zheng, Qi Huang 0001, Wen Heng, Yiyuan Ma, Wenlei Bao, Size Zheng 0001, Xuegui Zheng, Yanghua Peng, Haibin Lin, Xuanzhe Liu, Xin Jin 0008, Xin Liu 0086
EuroSys6
2026 Laminar: A Scalable Asynchronous RL Post-Training Framework
abstract
Reinforcement learning (RL) post-training for Large Language Models (LLMs) is now scaling to large clusters and running for extended durations to enhance model reasoning performance. However, the scalability of existing RL frameworks is limited, as extreme long-tail skewness in RL trajectory generation causes severe GPU underutilization. Current asynchronous RL systems attempt to mitigate this, but they rely on global weight synchronization between the actor and all rollouts, which creates a rigid model update schedule. This global synchronization is ill-suited for the highly skewed and evolving distribution of trajectory generation latency in RL training, crippling training efficiency. Our key insight is that efficient scaling requires breaking this lockstep through trajectory-level asynchrony, which generates and consumes each trajectory independently. We propose Laminar, a scalable and robust RL post-training system built on a fully decoupled architecture. First, we replace global updates with a tier of relay workers acting as a distributed parameter service. This enables asynchronous and fine-grained weight synchronization, allowing rollouts to pull the latest weight anytime without stalling the actor's training loop. Second, a dynamic repack mechanism consolidates long-tail trajectories onto a few dedicated rollouts, maximizing generation throughput. The fully decoupled design also isolates failures, ensuring robustness for long-running jobs. Our evaluation on a 1024-GPU cluster shows that Laminar achieves up to 5.48$\times$ training throughput speedup over state-of-the-art systems, while reducing model convergence time.
Guangming Sheng, Yuxuan Tong, Borui Wan, Wang Zhang 0017, Chaobo Jia, Xibin Wu, Xiang Li 0067, Chi Zhang 0022, Yanghua Peng, Haibin Lin, Xin Liu 0086, Chuan Wu 0001
EuroSys8
2026 MTGenRec: An Efficient Distributed Training System for Generative Recommendation Models in Meituan
abstract
Recommendation is crucial for both user experience and company revenue in Meituan as a leading lifestyle company, and generative recommendation models (GRMs) are shown to produce quality recommendations recently. However, existing systems are limited by insufficient functionality support and inefficient implementations for training GRMs in industrial scenarios. As such, we introduce MTGenRec as an efficient and scalable system for GRM training. Specifically, to handle real-time insertions/deletions of sparse embeddings, MTGenRec employs dynamic hash tables to replace static ones. To improve training efficiency, MTGenRec conducts dynamic sequence balancing to address the computation load imbalances among GPUs and adopts feature ID deduplication alongside automatic table merging to accelerate embedding lookup. Extensive experiments show that MTGenRec improves training throughput by 1.6x - 2.4x while achieving good scalability when running over 100 GPUs. MTGenRec has been deployed for many applications in Meituan and is now handling hundreds of millions of requests on a daily basis. On the delivery platform, we observe a 1.22% growth in user order volume and a 1.31% enhancement in online PV_CTR.
Yuxiang Wang 0013, Xiao Yan 0002, Mincong Huang, Ruidong Han, Bin Yin 0004, Shangyu Chen, Xiang Li 0067, Fei Jiang 0009, Wei Lin 0022, Haowei Han, Xiaokai Zhou, Bo Du 0001, Jiawei Jiang 0001
KDD (1)9
2026 SwiftEP: Accelerating MoE Inference with Buffer Fusion and TMA Offloading
Xingyi Li 0004, Shangguang Wang, Zhehao Lin, Yinben Xia, Qihang Liu, Xiang Li 0067, Zekun He, Yachen Wang, Xianneng Zou
NSDI13
2026 Optimizing Marketing Subsidies via Counterfactual Learning with Asymmetric Reward Function
abstract
In marketing, optimizing subsidy allocation to maximize overall profits is of substantial economic importance. Prior research has employed treatment effect estimation techniques to identify subsidy-sensitive items and design corresponding allocation strategies. However, more accurate treatment effect estimations do not necessarily lead to better allocations, underscoring the critical influence of decision boundaries in decision-making. This paper argues that optimal allocation fundamentally depends on predicting the expected optimal subsidy, a challenge distinct from conventional treatment effect estimation or causal decision-making, which existing approaches fail to address. To fill this gap, we introduce a two-stage Counterfactual optimal subsidy Learning method with an Asymmetric reward (CoLA). In the first stage, we derive a coarse estimate of the expected subsidy threshold by exploiting order information and the conditional independence between expected and observed subsidies. In the second stage, we refine these estimates using an asymmetric loss function, leading to more robust predictions. Under practical budget constraints, we prioritize candidates based on their Sharpe ratios to determine the final subsidy allocation strategy. Experiments on three public datasets and an online A/B test show that our method achieves significant performance improvements, yielding the highest total profit and incremental leverage ratios.
Xiang Li 0067, Yanghao Xiao, Chunyuan Zheng 0001, Qian Zou, Cheng Bing, Wei Lin 0022, Haoxuan Li 0001, Zhouchen Lin
SIGIR1
2026 User Activity Modeling under Inflated Distribution
Xiang Li 0067, Chunyuan Zheng 0001, Qingying Chen, Huiyou Jiang, Haoxuan Li 0001, Zhouchen Lin
SIGIR1
2026 Hierarchical Denoising Entire Space Multi-Task Model for Post-Click Conversion Rate Prediction with Noisy Labels
Haoxuan Li 0001, Xiang Li 0067, Chunyuan Zheng 0001, Xiao-Hua Zhou
SIGIR4
2026 A survey on learning from graphs with heterophily: recent advances and future directions
abstract
Abstract Graphs are structured data that models complex relations between real-world entities. Heterophilic graphs, where linked nodes trend to have different labels or dissimilar features, have recently attracted significant attention and found many real-world applications. Meanwhile, increasing efforts have been made to advance learning from graphs with heterophily. Various graph heterophily measures, benchmark datasets, and learning paradigms are emerging rapidly. In this survey, we comprehensively review existing works on learning from graphs with heterophily. First, we overview over 500 publications, of which more than 300 are directly related to heterophilic graphs. After that, we survey existing metrics of graph heterophily and list recent benchmark datasets. Further, we systematically categorize existing methods based on a hierarchical taxonomy including GNN models, learning paradigms and practical applications. In addition, broader topics related to graph heterophily are also included. Finally, we discuss the primary challenges of existing studies and highlight promising avenues for future research.
Cheng-Hua Gong, Yao Cheng 0009, Jian-Xiang Yu, Can Xu 0005, Siqiang Luo, Xiang Li 0067
Frontiers Comput. Sci.7
2026 Class-Balanced and fast active learning for graph neural networks via reinforcement learning
Chengcheng Yu, Jiapeng Zhu 0002, Xiang Li 0067
Knowl. Based Syst.3
2026 Geometric Contrastive Ensemble Distillation with calibration margin induction
Yang Yang 0080, Junyao Hou, Xiang Li 0067, Zhenghua Chen, Yingxue Gao, Chao Wang 0003, Min Wu 0008, Quanjun Yin
Pattern Recognit.5
2026 Prototyping and Analyzing Mobile SoC Clusters as Modern Edge Servers
abstract
The rapidly growing edge computing platforms, coupled with the still imperfect edge infrastructure, present an excellent opportunity for the emergence of new edge hardware. However, it remains unclear whether alternative architectures built from energy-efficient mobile System-on-Chips (SoCs) can meet the stringent performance, cost, and energy demands of modern edge workloads. In this paper, we propose a new type of edge server composed of 60 Qualcomm Snapdragon 865 mobile SoCs in a 2U rack, referred to as SoC Cluster. We demonstrate its successful deployment on existing edge cloud platforms and its ability to natively serve mobile cloud gaming services. Despite the emergence of new hardware on edge platforms and its successful operation in serving mobile cloud gaming, our trace analysis revealed low hardware utilization and significant dynamic fluctuations in usage. To assess its broader applicability, we conducted the first measurement study of SoC Cluster to reveal its ability to run two popular and modern edge applications: deep learning inference and video transcoding. We developed a cross-platform benchmark suite to evaluate throughput, latency, power consumption, and application-specific metrics like video quality. We then directly compare SoC Cluster with a traditional edge server equipped with Intel CPUs and NVIDIA GPUs in terms of energy efficiency, space efficiency, and monetary cost. Results show that SoC Cluster exhibits up to 6.5? higher energy efficiency and 7.7? higher space efficiency. We also disclose its limitations in serving computation-intensive workloads such as large deep learning models. The outcomes provide insightful implications and offer practical direction for refining SoC Cluster toward broader deployment in edge scenarios.
Li Zhang 0133, Boqing Shi, Xiang Li 0067, Ao Zhou 0001, Xiao Ma 0009, Shangguang Wang, Mengwei Xu 0001
IEEE Trans. Mob. Comput.3
2025 Leveraging Large Language Models for Node Generation in Few-Shot Learning on Text-Attributed Graphs
abstract
Text-attributed graphs have recently garnered significant attention due to their wide range of applications in web domains. Existing methodologies employ word embedding models for acquiring text representations as node features, which are subsequently fed into Graph Neural Networks (GNNs) for training. Recently, the advent of Large Language Models (LLMs) has introduced their powerful capabilities in information retrieval and text generation, which can greatly enhance the text attributes of graph data. Furthermore, the acquisition and labeling of extensive datasets are both costly and time-consuming endeavors. Consequently, few-shot learning has emerged as a crucial problem in the context of graph learning tasks. In order to tackle this challenge, we propose a lightweight paradigm called LLM4NG, which adopts a plug-and-play approach to establish supervision signals by leveraging LLMs for node generation. Specifically, we utilize LLMs to extract semantic information from the labels and generate samples that belong to these categories as exemplars. Subsequently, we employ an edge predictor to capture the structural information inherent in the raw dataset and integrate the newly generated samples into the original graph. This approach harnesses LLMs for enhancing class-level information and seamlessly introduces labeled nodes and edges without modifying the raw dataset, thereby facilitating the node classification task in few-shot scenarios. Extensive experiments demonstrate the outstanding performance of our proposed paradigm, particularly in low-shot scenarios. For instance, in the 1-shot setting of the ogbn-arxiv dataset, LLM4NG achieves a 76% improvement over the baseline model.
Jianxiang Yu 0001, Yuxiang Ren, Chenghua Gong, Jiaqi Tan 0006, Xiang Li 0067, Xuecang Zhang
AAAI5
2025 Demystifying Small Language Models for Edge Deployment
abstract
Small language models (SLMs) have emerged as a promising solution for deploying resource-constrained devices, such as smartphones and Web of Things. This work presents the first comprehensive study of over 60 SLMs such as Microsoft Phi and Google Gemma that are publicly accessible. Our findings show that state-of-the-art SLMs outperform 7B models in general tasks, proving their practical viability. However, SLMs’ in-context learning capabilities remain limited, and their efficiency has significant optimization potential. We identify key SLM optimization opportunities, including dynamic task-specific routing, model-hardware co-design, and vocabulary/KV cache compression. Overall, we expect the work to reveal an all-sided landscape of SLMs, benefiting the research community across algorithm, model, system, and hardware levels.
Zhenyan Lu, Xiang Li 0067, Dongqi Cai 0001, Rongjie Yi, Fangming Liu, Wei Liu 0302, Jian Luan 0001, Nicholas D. Lane, Mengwei Xu 0001
ACL (1)2
2025 Multi-level Relevance Document Identifier Learning for Generative Retrieval
abstract
Fuwei Zhang, Xiaoyu Liu, Xinyu Jia, Yingfei Zhang, Shuai Zhang, Xiang Li, Fuzhen Zhuang, Wei Lin, Zhao Zhang. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025.
Yingfei Zhang, Xiang Li 0067, Fuzhen Zhuang, Wei Lin 0022, Zhao Zhang 0011
ACL (1)6
2025 MTGR: Industrial-Scale Generative Recommendation Framework in Meituan
abstract
Scaling law has recently been validated in the recommendation system, adopting generative recommendation strategies to achieve scalability. However, these generative approaches require abandoning the meticulously constructed cross features of traditional recommendation models,leading to a significant decline in model performance. To address this challenge, we propose Meituan Generative Recommendation, which is based on the HSTU architecture and is capable of retaining the original deep learning recommendation model (DLRM) features, including cross features. Additionally, MTGR achieves training and inference acceleration through user-level compression to ensure efficient scaling. We also propose Group-Layer Normalization (GLN) to enhance the performance of encoding within different semantic spaces and the dynamic masking strategy to avoid information leakage. We further optimize the training frameworks, enabling support for our models with 10 to 100 times computational complexity compared to the DLRM, without significant cost increases. MTGR achieved 65x FLOPs for single-sample forward inference compared to the DLRM model, resulting in the largest gain in nearly two years both offline and online. This breakthrough was successfully deployed on Meituan, the world's largest food delivery platform, where it has been handling the main traffic.
Ruidong Han, Bin Yin 0004, Shangyu Chen, Fei Jiang 0009, Xiang Li 0067, Mincong Huang, Chunzhen Jing, Yueming Han, MengLei Zhou, Wei Lin 0022
CIKM6
2025 A Soft-partitioned Semi-supervised Collaborative Transfer Learning Approach for Multi-Domain Recommendation
abstract
n industrial practice, Multi-domain Recommendation (MDR) plays a crucial role. Shared-specific architectures are widely used in industrial solutions to capture shared and unique attributes via shared and specific parameters. However, with imbalanced data across different domains, these models face two key issues: (1) Overwhelming: Dominant domain data skews model performance, neglecting non-dominant domains. (2) Overfitting: Sparse data in non-dominant domains leads to overfitting in specific parameters. To tackle these challenges, we propose Soft-partitioned Semi-supervised Collaborative Transfer Learning (SSCTL) for multi-domain recommendation. SSCTL generates dynamic parameters to address the overwhelming issue, thus shifting focus towards samples from non-dominant domains. To combat overfitting, it leverages pseudo-labels with weights from dominant domain instances to enhance non-dominant domain data. We conduct comprehensive experiments, both online and offline, to validate the efficacy of our proposed method. Online tests yielded significant improvements across various domains, with increases in GMV ranging from 0.54% to 2.90% and enhancements in CTR ranging from 0.22% to 1.69%.
Yiqing Wu, Ruidong Han, Fuzhen Zhuang, Xiang Li 0067, Wei Lin 0022
CIKM5
2025 Towards Instance-wise Personalized Federated Learning via Semi-Implicit Bayesian Prompt Tuning
abstract
Federated learning (FL) is a privacy-preserving machine learning paradigm that enables collaborative model training across multiple distributed clients without disclosing their raw data. Personalized federated learning (pFL) has gained increasing attention for its ability to address data heterogeneity. However, most existing pFL methods assume that each client's data follows a single distribution and learn one client-level personalized model for each client. This assumption often fails in practice, where a single client may possess data from multiple sources or domains, resulting in significant intra-client heterogeneity and suboptimal performance. To tackle this challenge, we propose pFedBayesPT, a fine-grained instance-wise pFL framework based on visual prompt tuning. Specifically, we formulate instance-wise prompt generation from a Bayesian perspective and model the prompt posterior as an implicit distribution to capture diverse visual semantics. We derive a variational training objective under the semi-implicit variational inference framework. Extensive experiments on benchmark datasets demonstrate that pFedBayesPT consistently outperforms existing pFL methods under both feature and label heterogeneity settings.
Tiandi Ye, Wenyan Liu 0001, Lichun Li, Shangchao Su, Cen Chen 0001, Xiang Li 0067, Ming Gao 0001
CIKM7
2025 LogiGraph: Logical Reasoning with Contrastive Learning and Lightweight Graph Networks
abstract
Logical reasoning is a crucial factor in machine reading comprehension tasks (MRC). Existing methods suffer from the balance between semantic and explicit logical relation representations, in which some emphasize contextual semantics, while others pay more attention to explicit logical features. Additionally, previous methods utilize graph convolutional networks (GCN) for node updates, still exhibiting some shortcomings. To address these challenges, in this paper, we propose a logical reasoning method with contrastive learning and lightweight graph networks (LogiGraph). Our method focuses on the lightweight aspect of the GCN, which greatly improves the shortcomings of the GCN, and employs conjunction and punctuation marks as two types of edges to construct a dual graph. Besides, we combine contrastive learning with graph reasoning, which changes the logical expression’s content as the negative sample of the original context, enabling the model to capture negative logical relationships and improving generalization ability. We conduct extensive experiments on two public datasets, ReClor and LogiQA. Experimental results demonstrate that LogiGraph can achieve state-of-the-art performance on both datasets.
Xiang Li 0067, Yong Xu 0007, Jun Huang 0007
COLING1
2025 GraphCBAL-Sys: A Class-Balanced Active Learning System for Graphs
Chengcheng Yu, Wenqian Zhou, Fangshu Chen, Jiapeng Zhu 0002, Xiang Li 0067
DASFAA (6)6
2025 Can Large Language Models Act as Ensembler for Multi-GNNs?
abstract
Graph Neural Networks (GNNs) have emerged as powerful models for learning from graphstructured data.However, GNNs lack the inherent semantic understanding capability of rich textual node attributes, limiting their effectiveness in applications.On the other hand, we empirically observe that for existing GNN models, no one can consistently outperforms others across diverse datasets.In this paper, we study whether LLMs can act as an ensembler for multi-GNNs and propose the LensGNN model.The model first aligns multiple GNNs, mapping the representations of different GNNs into the same space.Then, through LoRA fine-tuning, it aligns the space between the GNN and the LLM, injecting graph tokens and textual information into LLMs.This allows LensGNN to ensemble multiple GNNs and take advantage of the strengths of LLM, leading to a deeper understanding of both textual semantic information and graph structural information.The experimental results show that LensGNN outperforms existing models.This research advances text-attributed graph ensemble learning by providing a robust and superior solution for integrating semantic and structural information.We provide our code and data here: https: //github.com/AquariusAQ/LensGNN.
Hanqi Duan, Yao Cheng 0009, Jianxiang Yu 0001, Xiang Li 0067
EMNLP5
2025 Text Detoxification: Data Efficiency, Semantic Preservation and Model Generalization
abstract
The widespread dissemination of toxic content on social media poses a serious threat to both online environments and public discourse, highlighting the urgent need for detoxification methods that effectively remove toxicity while preserving the original semantics.However, existing approaches often struggle to simultaneously achieve strong detoxification performance, semantic preservation, and robustness to out-of-distribution data.Moreover, they typically rely on costly, manually annotated parallel corpora while showing poor data efficiency.To address these challenges, we propose GEM, a two-stage training framework that jointly optimizes Model Generalization, Data Efficiency, and Semantic Preservation.We first perform supervised fine-tuning on a small set of high-quality, filtered parallel data to establish a strong initialization.Then, we leverage unlabeled toxic inputs and a custom-designed reward model to train the LLM using Group Relative Policy Optimization.Experimental results demonstrate that our method effectively mitigates the trade-offs faced by previous work, achieving state-of-the-art performance with improved generalization and significantly reduced dependence on annotated data.
Yibo Zhao 0005, Jiapeng Zhu 0002, Wenming Shao, Zhao Zhang 0011, Xiang Li 0067
EMNLP7
2025 Learning Prioritized Node-Wise Message Propagation in Graph Neural Networks (Extended Abstract)
abstract
Graphs are ubiquitous in the real world, in graphs, nodes represent entities and edges capture their relationships. Recently, graph neural networks (GNNs) [3]–[6] have been proposed to integrate these two sources of information. In GNNs, a node's embedding is learned by aggregating messages from its neighbors.
Yao Cheng 0009, Xiang Li 0067
ICDE4
2025 Having It Both Ways: Single Trajectory Embedding for Similarity Computation with Pairwise Learning
abstract
Trajectory similarity measure is a fundamental component in trajectory databases, supporting many down-stream trajectory tasks. Existing similarity functions often exhibit unacceptable time complexities, hampering their efficiency for real-world scenarios. To address this limitation, learning-based approximation techniques utilizing trajectory embeddings have been proposed. However, creating a robust embedding model presents challenges, including the lack of direct involvement in the computational similarity process, adherence to non-metric similarity spaces, and the integration of precise similarity computation alignments. To address these challenges, we introduce DTisT, a novel embedding framework that enhances trajectory embeddings by pairwise learning from dual-trajectory input models. DTisT not only captures the dynamics of trajectory similarity computation through a dual-trajectory learning model but also integrates a learnable virtual trajectory to align the embedding space with non-metric similarity spaces effectively. Additionally, we incorporate aligned information from actual similarity computations into our embedding process using an attention mask mechanism. To ensure effective learning, we adopt a pre-train and fine-tune strategy, utilizing contrastive learning during the pre-training stage. Extensive experiments conducted on two real datasets demonstrate that DTisT surpasses state-of-the-art methods, showcasing its effectiveness in trajectory similarity embedding.
Jianing Si, Haitao Yuan 0002, Xiang Li 0067, Xiao Ma 0009, Guoliang Li 0001, Shangguang Wang
ICDE3
2025 RELIEF: Reinforcement Learning Empowered Graph Feature Prompt Tuning
abstract
The advent of the "pre-train, prompt'' paradigm has recently extended its generalization ability and data efficiency to graph representation learning, following its achievements in Natural Language Processing (NLP). Initial graph prompt tuning approaches tailored specialized prompting functions for Graph Neural Network (GNN) models pre-trained with specific strategies, such as edge prediction, thus limiting their applicability. In contrast, another pioneering line of research has explored universal prompting via adding prompts to the input graph's feature space, thereby removing the reliance on specific pre-training strategies. However, the necessity to add feature prompts to all nodes remains an open question. Motivated by findings from prompt tuning research in the NLP domain, which suggest that highly capable pre-trained models need less conditioning signal to achieve desired behaviors, we advocate for strategically incorporating necessary and lightweight feature prompts to certain graph nodes to enhance downstream task performance. This introduces a combinatorial optimization problem, requiring a policy to decide 1) which nodes to prompt and 2) what specific feature prompts to attach. We then address the problem by framing the prompt incorporation process as a sequential decision-making problem and propose our method, RELIEF, which employs Reinforcement Learning (RL) to optimize it. At each step, the RL agent selects a node (discrete action) and determines the prompt content (continuous action), aiming to maximize cumulative performance gain. Extensive experiments on graph and node-level tasks with various pre-training strategies in few-shot scenarios demonstrate that our RELIEF outperforms fine-tuning and other prompt-based approaches in classification performance and data efficiency. The code is available at https://github.com/JasonZhujp/RELIEF.
Jiapeng Zhu 0002, Zichen Ding 0002, Jianxiang Yu 0001, Jiaqi Tan 0006, Xiang Li 0067, Weining Qian
KDD (1)5
2025 Variational Graph Autoencoder for Heterogeneous Information Networks with Missing and Inaccurate Attributes
abstract
Heterogeneous Information Networks (HINs), which consist of various types of nodes and edges, have recently witnessed excellent performance in graph mining. However, most existing heterogeneous graph neural networks (HGNNs) fail to simultaneously handle the problems of missing attributes, inaccurate attributes and scarce node labels, which limits their expressiveness. In this paper, we propose a generative self-supervised model GraMI to address these issues simultaneously. Specifically, GraMI first initializes all the nodes in the graph with a low-dimensional representation matrix. After that, based on the variational graph autoencoder framework, GraMI learns both node-level and attribute-level embeddings in the encoder, which can provide fine-grained semantic information to construct node attributes. In the decoder, GraMI reconstructs both links and attributes. Instead of directly reconstructing raw features for attributed nodes, GraMI generates the initial low-dimensional representation matrix for all the nodes, based on which raw features of attributed nodes are further reconstructed. In this way, GraMI can not only complete informative features for non-attributed nodes, but rectify inaccurate ones for attributed nodes. Finally, we conduct extensive experiments to show the superiority of GraMI in tackling HINs with missing and inaccurate attributes. Our code and data can be found here: https://github.com/See-r/GraMI.
Yige Zhao, Jianxiang Yu 0001, Yao Cheng 0009, Chengcheng Yu, Xiang Li 0067, Shuaiqiang Wang
KDD (1)6
2025 PA-RAG: RAG Alignment via Multi-Perspective Preference Optimization
abstract
Jiayi Wu, Hengyi Cai, Lingyong Yan, Hao Sun, Xiang Li, Shuaiqiang Wang, Dawei Yin, Ming Gao. Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers). 2025.
Jiayi Wu 0001, Hengyi Cai, Lingyong Yan, Hao Sun 0015, Xiang Li 0067, Shuaiqiang Wang, Dawei Yin 0001, Ming Gao 0001
NAACL (Long Papers)5
2025 Hierarchical Vector Quantized Graph Autoencoder with Annealing-Based Code Selection
abstract
Graph self-supervised learning has gained significant attention recently. However, many existing approaches heavily depend on perturbations, and inappropriate perturbations may corrupt the graph's inherent information. The Vector Quantized Variational Autoencoder (VQ-VAE) is a powerful autoencoder extensively used in fields such as computer vision; however, its application to graph data remains underexplored. In this paper, we provide an empirical analysis of vector quantization in the context of graph autoencoders, demonstrating its significant enhancement of the model's capacity to capture graph topology. Furthermore, we identify two key challenges associated with vector quantization when applying in graph data: codebook underutilization and codebook space sparsity. For the first challenge, we propose an annealing-based encoding strategy that promotes broad code utilization in the early stages of training, gradually shifting focus toward the most effective codes as training progresses. For the second challenge, we introduce a hierarchical two-layer codebook that captures relationships between embeddings through clustering. The second layer codebook links similar codes, encouraging the model to learn closer embeddings for nodes with similar features and structural topology in the graph. Our proposed model outperforms 16 representative baseline methods in self-supervised link prediction and node classification tasks across multiple datasets. Our implementation is available at https://github.com/vitaminzl/hqa-gae.
Long Zeng 0004, Jianxiang Yu 0001, Jiapeng Zhu 0002, Qingsong Zhong, Xiang Li 0067
WWW5
2024 Boosting Language Models Reasoning with Chain-of-Knowledge Prompting
abstract
Recently, Chain-of-Thought (CoT) prompting has delivered success on complex reasoning tasks, which aims at designing a simple prompt like "Let's think step by step" or multiple incontext exemplars with well-designed rationales to elicit Large Language Models (LLMs) to generate intermediate reasoning steps.However, the generated rationales often come with hallucinations, making unfactual and unfaithful reasoning chains.To mitigate this brittleness, we propose a novel Chain-of-Knowledge (CoK) prompting, where we aim at eliciting LLMs to generate explicit pieces of knowledge evidence in the form of structure triple.This is inspired by our human behaviors, i.e., we can draw a mind map or knowledge map as the reasoning evidence in the brain before answering a complex question.Benefiting from CoK, we additionally introduce a F 2 -Verification method to estimate the reliability of the reasoning chains in terms of factuality and faithfulness.For the unreliable response, the wrong evidence can be indicated to prompt the LLM to rethink.Extensive experiments demonstrate that our method can further improve the performance of commonsense, factual, symbolic, and arithmetic reasoning tasks 1 .
Jianing Wang 0002, Qiushi Sun, Xiang Li 0067, Ming Gao 0001
ACL (1)3
2024 Enhancing CTR Prediction through Sequential Recommendation Pre-training: Introducing the SRP4CTR framework
abstract
In sequential recommendation, pre-training from user historical behaviors through self-supervised learning can better comprehend user dynamic preferences, presenting the potential for direct integration with Click-Through Rate (CTR) prediction tasks. Previous methods have integrated pre-trained models into downstream tasks with the sole purpose of extracting semantic information or well-represented user features, which are then incorporated as new features. However, these approaches tend to ignore the additional inference costs and do not consider how to transfer the effective information from the pre-trained models for specific estimated items in CTR prediction. In this paper, we propose a Sequential Recommendation Pre-training framework for CTR prediction (SRP4CTR) to tackle the above problems. Initially, we discuss the impact of introducing pre-trained models on inference costs. Subsequently, we introduced a pre-trained method to encode sequence side information concurrently. During the fine-tuning process, we incorporate a cross-attention block to establish a bridge between estimated items and the pre-trained model at a low cost. Moreover, we develop a querying transformer technique to facilitate the knowledge transfer from the pre-trained model. Offline and online experiments show that our method outperforms previous baseline models.
Ruidong Han, Qianzhong Li, Rui Li 0044, Yurou Zhao, Xiang Li 0067, Wei Lin 0022
CIKM6
2024 GraphCBAL: Class-Balanced Active Learning for Graph Neural Networks via Reinforcement Learning
abstract
Graph neural networks (GNNs) have recently demonstrated significant success. Active learning for GNNs aims to query the valuable samples from the unlabeled data for annotation to maximize the GNNs' performance at a low cost. However, most existing methods for reinforced active learning in GNNs may lead to a highly imbalanced class distribution, especially in highly skewed class scenarios. This further adversely affects the classification performance. To tackle this issue, in this paper, we propose a novel reinforced class-balanced active learning framework for GNNs, namely, GraphCBAL. It learns an optimal policy to acquire class-balanced and informative nodes for annotation, maximizing the performance of GNNs trained with selected labeled nodes. GraphCBAL designs class-balance-aware states, as well as a reward function that achieves trade-off between model performance and class balance. We further upgrade GraphCBAL to GraphCBAL++ by introducing a punishment mechanism to obtain a more class-balanced labeled set. Extensive experiments on multiple datasets demonstrate the effectiveness of the proposed approaches, achieving superior performance over state-of-the-art baselines. In particular, our methods can strike the balance between classification results and class balance. We provide our code and data at https://github.com/cici-chengcheng/GraphCBAL.
Chengcheng Yu, Jiapeng Zhu 0002, Xiang Li 0067
CIKM3
2024 Aligning Explanations for Recommendation with Rating and Feature via Maximizing Mutual Information
abstract
Providing natural language-based explanations to justify recommendations helps to improve users' satisfaction and gain users' trust. However, as current explanation generation methods are commonly trained with an objective to mimic existing user reviews, the generated explanations are often not aligned with the predicted ratings or some important features of the recommended items, and thus, are suboptimal in helping users make informed decision on the recommendation platform. To tackle this problem, we propose a flexible model-agnostic method named MMI (Maximizing Mutual Information) framework to enhance the alignment between the generated natural language explanations and the predicted rating/important item features. Specifically, we propose to use mutual information (MI) as a measure for the alignment and train a neural MI estimator. Then, we treat a well-trained explanation generation model as the backbone model and further fine-tune it through reinforcement learning with guidance from the MI estimator, which rewards a generated explanation that is more aligned with the predicted rating or a pre-defined feature of the recommended item. Experiments on three datasets demonstrate that our MMI framework can boost different backbone models, enabling them to outperform existing baselines in terms of alignment with predicted ratings and item features. Additionally, user studies verify that MI-enhanced explanations indeed facilitate users' decisions and are favorable compared with other baselines due to their better alignment properties.
Yurou Zhao, Ruidong Han, Fei Jiang 0009, Lu Guan, Xiang Li 0067, Wei Lin 0022, Weizhi Ma, Jiaxin Mao
CIKM6
2024 Conjoin after Decompose: Improving Few-Shot Performance of Named Entity Recognition
abstract
Prompt-based methods have been widely used in few-shot named entity recognition (NER). In this paper, we first conduct a preliminary experiment and observe that the key to affecting the performance of prompt-based NER models is the capability to detect entity boundaries. However, most existing models fail to boost such capability. To solve the issue, we propose a novel model, ParaBART, which consists of a BART encoder and a specially designed parabiotic decoder. Specifically, the parabiotic decoder includes two BART decoders and a conjoint module. The two decoders are responsible for entity boundary detection and entity type classification, respectively. They are connected by the conjoint module, which is used to replace unimportant tokens’ embeddings in one decoder with the average embedding of all the tokens in the other. We further present a novel boundary expansion strategy to enhance the model’s capability in entity type classification. Experimental results show that ParaBART can achieve significant performance gains over state-of-the-art competitors.
Chengcheng Han 0004, Renyu Zhu, Jun Kuang, Fengjiao Chen, Xiang Li 0067, Ming Gao 0001, Xuezhi Cao, Yunsen Xian
LREC/COLING5
2024 AlphaFin: Benchmarking Financial Analysis with Retrieval-Augmented Stock-Chain Framework
abstract
The task of financial analysis primarily encompasses two key areas: stock trend prediction and the corresponding financial question answering. Currently, machine learning and deep learning algorithms (ML&DL) have been widely applied for stock trend predictions, leading to significant progress. However, these methods fail to provide reasons for predictions, lacking interpretability and reasoning processes. Also, they can not integrate textual information such as financial news or reports. Meanwhile, large language models (LLM) have remarkable textual understanding and generation ability. But due to the scarcity of financial training datasets and limited integration with real-time knowledge, LLM still suffer from hallucinations and unable to keep up with the latest information. To tackle these challenges, we first release AlphaFin datasets, combining traditional research datasets, real-time financial data, and handwritten chain-of-thought (CoT) data. It has positive impact on training LLM for completing financial analysis. We then use AlphaFin datasets to benchmark a state-of-the-art method, called Stock-Chain, for effectively tackling the financial analysis task, which integrates retrieval-augmented generation (RAG) techniques. Extensive experiments are conducted to demonstrate the effectiveness of our framework on financial analysis.
Xiang Li 0067, Yong Xu 0007, Mingkui Tan, Jun Huang 0007
LREC/COLING1
2024 Make Prompt-based Black-Box Tuning Colorful: Boosting Model Generalization from Three Orthogonal Perspectives
abstract
Large language models (LLMs) have shown increasing power on various natural language processing (NLP) tasks. However, tuning these models for downstream tasks usually needs exorbitant costs or is unavailable due to commercial considerations. Recently, black-box tuning has been proposed to address this problem by optimizing task-specific prompts without accessing the gradients and hidden representations. However, most existing works have yet fully exploited the potential of gradient-free optimization under the scenario of few-shot learning. In this paper, we describe BBT-RGB, a suite of straightforward and complementary techniques for enhancing the efficiency and performance of black-box optimization. Specifically, our method includes three plug-and-play components: (1) Two-stage derivative-free optimization strategy that facilitates fast convergence and mitigates overfitting; (2) Automatic verbalizer construction with its novel usage under few-shot settings; (3) Better prompt initialization policy based on instruction search and auto-selected demonstration. Extensive experiments across various tasks on natural language understanding and inference demonstrate the effectiveness of our method. Our codes are available at https://github.com/QiushiSun/BBT-RGB.
Qiushi Sun, Chengcheng Han 0004, Nuo Chen 0002, Renyu Zhu, Jingyang Gong, Xiang Li 0067, Ming Gao 0001
LREC/COLING6
2024 TransCoder: Towards Unified Transferable Code Representation Learning Inspired by Human Skills
abstract
Code pre-trained models (CodePTMs) have recently demonstrated a solid capacity to process various code intelligence tasks, e.g., code clone detection, code translation, and code summarization. The current mainstream method that deploys these models to downstream tasks is to fine-tune them on individual tasks, which is generally costly and needs sufficient data for large models. To tackle the issue, in this paper, we present TransCoder, a unified Transferable fine-tuning strategy for Code representation learning. Inspired by human inherent skills of knowledge generalization, TransCoder drives the model to learn better code-related knowledge like human programmers. Specifically, we employ a tunable prefix encoder to first capture cross-task and cross-language transferable knowledge, subsequently applying the acquired knowledge for optimized downstream adaptation. Besides, our approach confers benefits for tasks with minor training sample sizes and languages with smaller corpora, underscoring versatility and efficacy. Extensive experiments conducted on representative datasets clearly demonstrate that our method can lead to superior performance on various code-related tasks and encourage mutual reinforcement, especially in low-resource scenarios. Our codes are available at https://github.com/QiushiSun/TransCoder.
Qiushi Sun, Nuo Chen 0002, Jianing Wang 0002, Ming Gao 0001, Xiang Li 0067
LREC/COLING5
2024 Structure-aware Fine-tuning for Code Pre-trained Models
abstract
Over the past few years, we have witnessed remarkable advancements in Code Pre-trained Models (CodePTMs). These models achieved excellent representation capabilities by designing structure-based pre-training tasks for code. However, how to enhance the absorption of structural knowledge when fine-tuning CodePTMs still remains a significant challenge. To fill this gap, in this paper, we present SAT, a novel structure-enhanced and plug-and-play fine-tuning method for CodePTMs. We first propose a structure loss to quantify the difference between the information learned by CodePTMs and the knowledge extracted from code structure. Specifically, we use the attention scores from Transformer layer as the learned information, and the shortest path length between leaves in abstract syntax trees as the structural knowledge. Subsequently, multi-task learning is introduced to improve the performance of fine-tuning. Experiments conducted on four pre-trained models and two generation tasks demonstrate the effectiveness of our proposed method as a plug-and-play solution. Furthermore, we observed that SAT can benefit CodePTMs more with limited training data.
Jiayi Wu 0001, Renyu Zhu, Nuo Chen 0002, Qiushi Sun, Xiang Li 0067, Ming Gao 0001
LREC/COLING5
2024 Variational Kernel Density Estimation Recommendation Algorithm for Users with Diverse Activity Levels
Wei Liu 0061, Shangsong Liang, Huaijie Zhu, Leong Hou U, Jianxing Yu, Xiang Li 0067, Jian Yin 0001
DASFAA (2)6
2024 Federated Learning via Consensus Mechanism on Heterogeneous Data: A New Perspective on Convergence
abstract
Federated learning (FL) on heterogeneous data (non-IID data) has recently received great attention. Most existing methods focus on studying the convergence guarantees for the global objective. While these methods can guarantee the decrease of the global objective in each communication round, they fail to ensure risk decrease for each client. In this paper, we propose FedCOME, which introduces a consensus mechanism aiming decreased risk for each client after each training round. In particular, we allow a slight adjustment to a client’s gradient on the server-side, producing an acute angle between the corrected and original gradients of participated clients. To generalize the consensus mechanism to the partial participation FL scenario, we devise a novel client sampling strategy to enhance the representativeness of the selected client subset to more accurately reflect the global population.. Training on these selected clients with the consensus mechanism could empirically lead to risk decrease for clients that are not selected. Finally, we conduct extensive experiments on four benchmark datasets to show the superiority of FedCOME against other state-of-the-art methods in terms of effectiveness, efficiency. For reproducibility, we make our source code publicly available at: https://github.com/fedcome/fedcome.
Shu Zheng, Tiandi Ye, Xiang Li 0067, Ming Gao 0001
ICASSP3
2024 DFDG: Data-Free Dual-Generator Adversarial Distillation for One-Shot Federated Learning
abstract
Federated Learning (FL) is a distributed machine learning scheme in which clients jointly participate in the collaborative training of a global model by sharing model information rather than their private datasets. In light of concerns associated with communication and privacy, one-shot FL with a single communication round has emerged as a de facto promising solution. However, existing one-shot FL methods either require public datasets, focus on model homogeneous settings, or distill limited knowledge from local models, making it difficult or even impractical to train a robust global model. To address these limitations, we propose a new data-free dual-generator adversarial distillation method (namely DFDG) for one-shot FL, which can explore a broader local models' training space via training dual generators. DFDG is executed in an adversarial manner and comprises two parts: dual-generator training and dual-model distillation. In dual-generator training, we delve into each generator concerning fidelity, transferability and diversity to ensure its utility, and additionally tailor the cross-divergence loss to lessen the overlap of dual generators' output spaces. In dual-model distillation, the trained dual generators work together to provide the training data for updates of the global model. At last, our extensive experiments on various image classification tasks show that DFDG achieves significant performance gains in accuracy compared to SOTA baselines. We provide our code here: https://anonymous.4open.science/r/DFDG-7BDB.
Kangyang Luo, Yexuan Fu, Renrong Shao, Xiang Li 0067, Yunshi Lan, Ming Gao 0001, Jinlong Shu
ICDM5
2024 Training-free Multi-objective Diffusion Model for 3D Molecule Generation
abstract
Searching for novel and diverse molecular candidates is a critical undertaking in drug and material discovery. Existing approaches have successfully adapted the diffusion model, the most effective generative model in image generation, to create 1D SMILES strings, 2D chemical graphs, or 3D molecular conformers. However, these methods are not efficient and flexible enough to generate 3D molecules with multiple desired properties, as they require additional training for the models for each new property or even a new combination of existing properties. Moreover, some properties may potentially conflict, making it impossible to find a molecule that satisfies all of them simultaneously. To address these challenges, we present a training-free conditional 3D molecular generation algorithm based on off-the-shelf unconditional diffusion models and property prediction models. The key techniques include modeling the loss of property prediction models as energy functions, considering the property relation between multiple conditions as a probabilistic graph, and developing a stable posterior estimation for computing the conditional score function. We conducted experiments on both single-objective and multi-objective 3D molecule generation, focusing on quantum properties, and compared our approach with the trained or fine-tuned diffusion models. Our proposed model achieves superior performance in generating molecules that meet the conditions, without any additional training cost.
Xu Han 0012, Yifei Shen 0004, Can Xu 0005, Xiang Li 0067, Dongsheng Li 0002
ICLR6
2024 Resurrecting Label Propagation for Graphs with Heterophily and Label Noise
abstract
Label noise is a common challenge in large datasets, as it can significantly degrade the generalization ability of deep neural networks. Most existing studies focus on noisy labels in computer vision; however, graph models encompass both node features and graph topology as input, and become more susceptible to label noise through message-passing mechanisms. Recently, only a few works have been proposed to tackle the label noise on graphs. One significant limitation is that they operate under the assumption that the graph exhibits homophily and that the labels are distributed smoothly. However, real-world graphs can exhibit varying degrees of heterophily, or even be dominated by heterophily, which results in the inadequacy of the current methods.
Yao Cheng 0009, Yifei Shen 0004, Xiang Li 0067, Siqiang Luo, Dongsheng Li 0002
KDD4
2024 Unified Dual-Intent Translation for Joint Modeling of Search and Recommendation
abstract
Recommendation systems, which assist users in discovering their preferred items among numerous options, have served billions of users across various online platforms. Intuitively, users' interactions with items are highly driven by their unchanging inherent intents (e.g., always preferring high-quality items) and changing demand intents (e.g., wanting a T-shirt in summer but a down jacket in winter). However, both types of intents are implicitly expressed in recommendation scenario, posing challenges in leveraging them for accurate intent-aware recommendations. Fortunately, in search scenario, often found alongside recommendation on the same online platform, users express their demand intents explicitly through their query words. Intuitively, in both scenarios, a user shares the same inherent intent and the interactions may be influenced by the same demand intent. It is therefore feasible to utilize the interaction data from both scenarios to reinforce the dual intents for joint intent-aware modeling. But the joint modeling should deal with two problems: 1) accurately modeling users' implicit demand intents in recommendation; 2) modeling the relation between the dual intents and the interactive items. To address these problems, we propose a novel model named Unified Dual-Intents Translation for joint modeling of Search and Recommendation (UDITSR). To accurately simulate users' demand intents in recommendation, we utilize real queries from search data as supervision information to guide its generation. To explicitly model the relation among the triplet , we propose a dual-intent translation propagation mechanism to learn the triplet in the same semantic space via embedding translations. Extensive experiments demonstrate that UDITSR outperforms SOTA baselines both in search and recommendation tasks.
Yuting Zhang 0010, Yiqing Wu, Ruidong Han, Ying Sun 0006, Yongchun Zhu, Xiang Li 0067, Wei Lin 0022, Fuzhen Zhuang, Zhulin An, Yongjun Xu 0001
KDD6
2024 Mobile Foundation Model as Firmware
abstract
In the current AI era, mobile devices such as smartphones are tasked with executing a myriad of deep neural networks (DNNs) locally. It presents a complex landscape, as these models are highly fragmented in terms of architecture, operators, and implementations. Such fragmentation poses significant challenges to the co-optimization of hardware, systems, and algorithms for efficient and scalable mobile AI.
Jinliang Yuan, Chen Yang 0043, Dongqi Cai 0001, Shihe Wang, Zeling Zhang, Xiang Li 0067, Dingge Zhang, Hanzi Mei, Xianqing Jia, Shangguang Wang, Mengwei Xu 0001
MobiCom7
2024 Cross-model Control: Improving Multiple Large Language Models in One-time Training
abstract
The number of large language models (LLMs) with varying parameter scales and vocabularies is increasing. While they deliver powerful performance, they also face a set of common optimization needs to meet specific requirements or standards, such as instruction following or avoiding the output of sensitive information from the real world. However, how to reuse the fine-tuning outcomes of one model to other models to reduce training costs remains a challenge. To bridge this gap, we introduce Cross-model Control (CMC), a method that improves multiple LLMs in one-time training with a portable tiny language model. Specifically, we have observed that the logit shift before and after fine-tuning is remarkably similar across different models. Based on this insight, we incorporate a tiny language model with a minimal number of parameters. By training alongside a frozen template LLM, the tiny model gains the capability to alter the logits output by the LLMs. To make this tiny language model applicable to models with different vocabularies, we propose a novel token mapping strategy named PM-MinED. We have conducted extensive experiments on instruction tuning and unlearning tasks, demonstrating the effectiveness of CMC. Our code is available at https://github.com/wujwyi/CMC
Jiayi Wu 0001, Hao Sun 0015, Hengyi Cai, Lixin Su, Shuaiqiang Wang, Dawei Yin 0001, Xiang Li 0067, Ming Gao 0001
NeurIPS7
2024 MegaScale: Scaling Large Language Model Training to More Than 10, 000 GPUs
Ziheng Jiang, Haibin Lin, Yinmin Zhong, Qi Huang 0001, Yangrui Chen, Zhi Zhang 0005, Yanghua Peng, Xiang Li 0067, Shibiao Nong, Yulu Jia, Sun He, Hongmin Chen, Zhihao Bai, Qi Hou, Shipeng Yan, Yiyao Sheng, Zhuo Jiang, Haohan Xu, Zhang Zhang 0003, Pengfei Nie, Leqi Zou, Sida Zhao, Zherui Liu, Xiaoying Jia 0001, Jianxi Ye, Xin Jin 0008, Xin Liu 0086
NSDI8
2024 PSP: Pre-training and Structure Prompt Tuning for Graph Neural Networks
Qingqing Ge, Zeyuan Zhao, Anfeng Cheng, Xiang Li 0067, Shuaiqiang Wang, Dawei Yin 0001
ECML/PKDD (5)5
2024 Self-pro: A Self-prompt and Tuning Framework for Graph Neural Networks
Chenghua Gong, Xiang Li 0067, Jianxiang Yu 0001, Yao Cheng 0009, Jiaqi Tan 0006, Chengcheng Yu
ECML/PKDD (2)2
2024 HetCAN: A Heterogeneous Graph Cascade Attention Network with Dual-Level Awareness
Zeyuan Zhao, Qingqing Ge, Anfeng Cheng, Xiang Li 0067, Shuaiqiang Wang
ECML/PKDD (6)5
2024 LARR: Large Language Model Aided Real-time Scene Recommendation with Semantic Understanding
abstract
Click-Through Rate (CTR) prediction is crucial for Recommendation System(RS), aiming to provide personalized recommendation services for users in many aspects such as food delivery, e-commerce and so on. However, traditional RS relies on collaborative signals, which lacks semantic understanding to real-time scenes. We also noticed that a major challenge in utilizing Large Language Models (LLMs) for practical recommendation purposes is their efficiency in dealing with long text input. To break through the problems above, we propose Large Language Model Aided Real-time Scene Recommendation(LARR), adopt LLMs for semantic understanding, utilizing real-time scene information in RS without requiring LLM to process the entire real-time scene text directly, thereby enhancing the efficiency of LLM-based CTR modeling. Specifically, recommendation domain-specific knowledge is injected into LLM and then RS employs an aggregation encoder to build real-time scene information from separate LLM’s outputs. Firstly, a LLM is continual pretrained on corpus built from recommendation data with the aid of special tokens. Subsequently, the LLM is fine-tuned via contrastive learning on three kinds of sample construction strategies. Through this step, LLM is transformed into a text embedding model. Finally, LLM’s separate outputs for different scene features are aggregated by an encoder, aligning to collaborative signals in RS, enhancing the performance of recommendation model.
Zhizhong Wan, Bin Yin 0004, Fei Jiang 0009, Xiang Li 0067, Wei Lin 0022
RecSys5
2024 UPFL: Unsupervised Personalized Federated Learning towards New Clients
abstract
Personalized federated learning (pFL) has gained significant attention as a promising approach to address the challenge of data heterogeneity. In this paper, we address a relatively unexplored problem in federated learning. When a federated model has been trained and deployed, and an unla-beled new client joins, providing a personalized model for the new client becomes a highly challenging task. To address this challenge, we extend the adaptive risk minimization technique into the unsupervised pFL setting and propose our method, FedTTA. We further improve FedTTA with two simple yet highly effective optimization strategies: enhancing the training of the adaptation model with proxy regularization and early-stopping the adaptation through entropy. Moreover, we propose a knowledge distillation loss specifically designed for FedTTA to address the device heterogeneity. Extensive experiments on five datasets against eleven baselines demonstrate the effectiveness of our proposed FedTTA and its variants. The code is available at: https://github.com/anonymous-federated-learning/code.
Tiandi Ye, Cen Chen 0001, Yinggui Wang, Xiang Li 0067, Ming Gao 0001
SDM4
2024 FwdLLM: Efficient Federated Finetuning of Large Language Models with Perturbed Inferences
Mengwei Xu 0001, Dongqi Cai 0001, Yaozong Wu, Xiang Li 0067, Shangguang Wang
USENIX ATC4
2024 More is Different: Prototyping and Analyzing a New Form of Edge Server with Massive Mobile SoCs
Li Zhang 0133, Zhe Fu 0005, Boqing Shi, Xiang Li 0067, Rujin Lai, Chenyang Yang 0004, Ao Zhou 0001, Xiao Ma 0009, Shangguang Wang, Mengwei Xu 0001
USENIX ATC4
2024 MuxFlow: efficient GPU sharing in production-level clusters with more than 10000 GPUs
Xuanzhe Liu, Shufan Liu, Xiang Li 0067, Yibo Zhu 0001, Xin Liu 0086, Xin Jin 0008
Sci. China Inf. Sci.4
2024 Predicting stock market trends with self-supervised learning
Zelin Ying, Dawei Cheng, Cen Chen 0001, Xiang Li 0067, Peng Zhu 0002, Yifeng Luo
Neurocomputing4
2024 SMEC: Scene Mining for E-Commerce
Xiang Li 0067, Dawei Yin 0001, Shuai Ma 0001
J. Comput. Sci. Technol.2
2024 BapFL: You can Backdoor Personalized Federated Learning
abstract
In federated learning (FL), malicious clients could manipulate the predictions of the trained model through backdoor attacks, posing a significant threat to the security of FL systems. Existing research primarily focuses on backdoor attacks and defenses within the generic federated learning scenario, where all clients collaborate to train a single global model. A recent study conducted by Qin et al. [ 24 ] marks the initial exploration of backdoor attacks within the personalized federated learning (pFL) scenario, where each client constructs a personalized model based on its local data. Notably, the study demonstrates that pFL methods with parameter decoupling can significantly enhance robustness against backdoor attacks. However, in this article, we whistleblow that pFL methods with parameter decoupling are still vulnerable to backdoor attacks. The resistance of pFL methods with parameter decoupling is attributed to the heterogeneous classifiers between malicious clients and benign counterparts. We analyze two direct causes of the heterogeneous classifiers: (1) data heterogeneity inherently exists among clients and (2) poisoning by malicious clients further exacerbates the data heterogeneity. To address these issues, we propose a two-pronged attack method, BapFL, which comprises two simple yet effective strategies: (1) poisoning only the feature encoder while keeping the classifier fixed and (2) diversifying the classifier through noise introduction to simulate that of the benign clients. Extensive experiments on three benchmark datasets under varying conditions demonstrate the effectiveness of our proposed attack. Additionally, we evaluate the effectiveness of six widely used defense methods and find that BapFL still poses a significant threat even in the presence of the best defense, Multi-Krum. We hope to inspire further research on attack and defense strategies in pFL scenarios. The code is available at: https://github.com/BapFL/code
Tiandi Ye, Cen Chen 0001, Yinggui Wang, Xiang Li 0067, Ming Gao 0001
ACM Trans. Knowl. Discov. Data4
2024 Learning Prioritized Node-Wise Message Propagation in Graph Neural Networks
abstract
Graph neural networks (GNNs) have recently received significant attention. Learning node-wise message propagation in GNNs aims to set personalized propagation steps for different nodes in the graph. Despite the success, existing methods ignore node priority that can be reflected by node influence and heterophily. In this paper, we propose a versatile framework PriPro, which can be integrated with most existing GNN models and aim to learn prioritized node-wise message propagation in GNNs. Specifically, the framework consists of three components: a backbone GNN model, a propagation controller to determine the optimal propagation steps for nodes, and a weight controller to compute the priority scores for nodes. We design a mutually enhanced mechanism to compute node priority, optimal propagation step and label prediction. We also propose an alternative optimization strategy to learn the parameters in the backbone GNN model and two parametric controllers. We conduct extensive experiments to compare our framework with other 12 state-of-the-art competitors on 10 benchmark datasets. Experimental results show that our framework can lead to superior performance in terms of propagation strategies and node representations.
Yao Cheng 0009, Xiang Li 0067
IEEE Trans. Knowl. Data Eng.4
2024 Heterogeneous Graph Contrastive Learning With Meta-Path Contexts and Adaptively Weighted Negative Samples
abstract
Heterogeneous graph contrastive learning has received wide attention recently. Some existing methods use meta-paths, which are sequences of object types that capture semantic relationships between objects, to construct contrastive views. However, most of them ignore the rich meta-path context information that describes how two objects are connected by meta-paths. Further, they fail to distinguish negative samples, which could adversely affect the model performance. To address the problems, we propose MEOW, which considers both meta-path contexts and weighted negative samples. Specifically, MEOW constructs a coarse view and a fine-grained view for contrast. The former reflects which objects are connected by meta-paths, while the latter uses meta-path contexts and characterizes details on how the objects are connected. Then, we theoretically analyze the InfoNCE loss and recognize its limitations for computing gradients of negative samples. To better distinguish negative samples, we learn hard-valued weights for them based on node clustering and use prototypical contrastive learning to pull close embeddings of nodes in the same cluster. In addition, we propose a variant model AdaMEOW that adaptively learns soft-valued weights of negative samples to further improve node representation. Finally, we conduct extensive experiments to show the superiority of MEOW and AdaMEOW against other state-of-the-art methods.
Jianxiang Yu 0001, Qingqing Ge, Xiang Li 0067, Aoying Zhou
IEEE Trans. Knowl. Data Eng.3
2024 Scalable decoupling graph neural network with feature-oriented optimization
Ningyi Liao, Dingheng Mo, Siqiang Luo, Xiang Li 0067
VLDB J.4
2023 MUSE: Multi-view Contrastive Learning for Heterophilic Graphs via Information Reconstruction
abstract
In recent years, self-supervised learning has emerged as a promising approach in addressing the issues of label dependency and poor generalization performance in traditional GNNs. However, existing self-supervised methods have limited effectiveness on heterophilic graphs, due to the homophily assumption that results in similar node representations for connected nodes. In this work, we propose a multi-view contrastive learning model for heterophilic graphs, namely, MUSE. Specifically, we construct two views to capture the information of the ego node and its neighborhood by GNNs enhanced with contrastive learning, respectively. Then we integrate the information from these two views to fuse the node representations. Fusion contrast is utilized to enhance the effectiveness of fused node representations. Further, considering that the influence of neighboring contextual information on information fusion may vary across different ego nodes, we employ an information fusion controller to model the diversity of node-neighborhood similarity at both the local and global levels. Finally, an alternating training scheme is adopted to ensure that unsupervised node representation learning and information fusion controller can mutually reinforce each other. We conduct extensive experiments to evaluate the performance of MUSE on 9 benchmark datasets. Our results show the effectiveness of MUSE on both node classification and clustering tasks. We provide our data and codes at https://github.com/dcxr969/MUSE.
Mengyi Yuan, Xiang Li 0067
CIKM3
2023 GradMA: A Gradient-Memory-based Accelerated Federated Learning with Alleviated Catastrophic Forgetting
abstract
Federated Learning (FL) has emerged as a de facto machine learning area and received rapid increasing research interests from the community. However, catastrophic forgetting caused by data heterogeneity and partial participation poses distinctive challenges for FL, which are detrimental to the performance. To tackle the problems, we propose a new FL approach (namely GradMA), which takes inspiration from continual learning to simultaneously correct the server-side and worker-side update directions as well as take full advantage of server's rich computing and memory resources. Furthermore, we elaborate a memory reduction strategy to enable GradMA to accommodate FL with a large scale of workers. We then analyze convergence of GradMA theoretically under the smooth non-convex setting and show that its convergence rate achieves a linear speed up w.r.t the increasing number of sampled active workers. At last, our extensive experiments on various image classification tasks show that GradMA achieves significant performance gains in accuracy and communication efficiency compared to SOTA baselines. We provide our code here: https://github.com/lkyddd/GradMA.
Kangyang Luo, Xiang Li 0067, Yunshi Lan, Ming Gao 0001
CVPR2
2023 Meta-learning Siamese Network for Few-Shot Text Classification
Chengcheng Han 0004, Yingnan Fu, Xiang Li 0067, Minghui Qiu, Ming Gao 0001, Aoying Zhou
DASFAA (3)4
2023 Decentralized Local Updates with Dual-Slow Estimation and Momentum-Based Variance-Reduction for Non-Convex Optimization
abstract
Decentralized learning (DL) has recently employed local updates to reduce the communication cost for general non-convex optimization problems. Specifically, local updates require each node to perform multiple update steps on the parameters of the local model before communicating with others. However, most existing methods could be highly sensitive to data heterogeneity (i.e., non-iid data distribution) and adversely affected by the stochastic gradient noise. In this paper, we propose DSE-MVR to address these problems. Specifically, DSE-MVR introduces a dual-slow estimation strategy that utilizes the gradient tracking technique to estimate the global accumulated update direction for handling the data heterogeneity problem; also for stochastic noise, the method uses the mini-batch momentum-based variance-reduction technique. We theoretically prove that DSE-MVR can achieve optimal convergence results for general non-convex optimization in both iid and non-iid data distribution settings. In particular, the leading terms in the convergence rates derived by DSE-MVR are independent of the stochastic noise for large-batches or large partial average intervals (i.e., the number of local update steps). Further, we put forward DSE-SGD and theoretically justify the importance of the dual-slow estimation strategy in the data heterogeneity setting. Finally, we conduct extensive experiments to show the superiority of DSE-MVR against other state-of-the-art approaches. We provide our code here: https://anonymous.4open.science/r/DSE-MVR-32B8/.
Kangyang Luo, Kunkun Zhang, Shengbo Zhang, Xiang Li 0067, Ming Gao 0001
ECAI4
2023 DialCoT Meets PPO: Decomposing and Exploring Reasoning Paths in Smaller Language Models
abstract
Chain-of-Thought (CoT) prompting has proven to be effective in enhancing the reasoning capabilities of Large Language Models (LLMs) with at least 100 billion parameters.However, it is ineffective or even detrimental when applied to reasoning tasks in Smaller Language Models (SLMs) with less than 10 billion parameters.To address this limitation, we introduce Dialogue-guided Chain-of-Thought (DialCoT) which employs a dialogue format to generate intermediate reasoning steps, guiding the model toward the final answer.Additionally, we optimize the model's reasoning path selection using the Proximal Policy Optimization (PPO) algorithm, further enhancing its reasoning capabilities.Our method offers several advantages compared to previous approaches.Firstly, we transform the process of solving complex reasoning questions by breaking them down into a series of simpler sub-questions, significantly reducing the task difficulty and making it more suitable for SLMs.Secondly, we optimize the model's reasoning path selection through the PPO algorithm.We conduct comprehensive experiments on four arithmetic reasoning datasets, demonstrating that our method achieves significant performance improvements compared to state-of-the-art competitors.1
Chengcheng Han 0004, Xiaowei Du, Che Zhang, Yixin Lian, Xiang Li 0067, Ming Gao 0001, Baoyuan Wang
EMNLP5
2023 Graph Self-Contrast Representation Learning
abstract
Graph contrastive learning (GCL) has recently emerged as a promising approach for graph representation learning. Some existing methods adopt the 1-vs-K scheme to construct one positive and K negative samples for each graph, but it is difficult to set K. For those methods that do not use negative samples, it is often necessary to add additional strategies to avoid model collapse, which could only alleviate the problem to some extent. All these drawbacks will undoubtedly have an adverse impact on the generalizability and efficiency of the model. In this paper, to address these issues, we propose a novel graph self-contrast framework GraphSC, which only uses one positive and one negative sample, and chooses triplet loss as the objective. Specifically, self-contrast has two implications. First, GraphSC generates both positive and negative views of a graph sample from the graph itself via graph augmentation functions of various intensities, and use them for self-contrast. Second, GraphSC uses Hilbert-Schmidt Independence Criterion (HSIC) to factorize the representations into multiple factors and proposes a masked self-contrast mechanism to better separate positive and negative samples. Further, Since the triplet loss only optimizes the relative distance between the anchor and its positive/negative samples, it is difficult to ensure the absolute distance between the anchor and positive sample. Therefore, we explicitly reduced the absolute distance between the anchor and positive sample to accelerate convergence. Finally, we conduct extensive experiments to evaluate the performance of GraphSC against 19 other state-of-the-art methods in both unsupervised and transfer learning settings.
Yao Cheng 0009, Xiang Li 0067, Ming Gao 0001
ICDM4
2023 Explaining Temporal Graph Models through an Explorer-Navigator Framework
Wenwen Xia, Mincai Lai, Yao Zhang 0009, Xinnan Dai, Xiang Li 0067, Dongsheng Li 0002
ICLR6
2023 Modeling Dual Period-Varying Preferences for Takeaway Recommendation
abstract
Takeaway recommender systems, which aim to accurately provide stores that offer foods meeting users' interests, have served billions of users in our daily life. Different from traditional recommendation, takeaway recommendation faces two main challenges: (1) Dual Interaction-Aware Preference Modeling. Traditional recommendation commonly focuses on users' single preferences for items while takeaway recommendation needs to comprehensively consider users' dual preferences for stores and foods. (2) Period-Varying Preference Modeling. Conventional recommendation generally models continuous changes in users' preferences from a session-level or day-level perspective. However, in practical takeaway systems, users' preferences vary significantly during the morning, noon, night, and late night periods of the day. To address these challenges, we propose a Dual Period-Varying Preference modeling (DPVP) for takeaway recommendation. Specifically, we design a dual interaction-aware module, aiming to capture users' dual preferences based on their interactions with stores and foods. Moreover, to model various preferences in different time periods of the day, we propose a time-based decomposition module as well as a time-aware gating mechanism. Extensive offline and online experiments demonstrate that our model outperforms state-of-the-art methods on real-world datasets and it is capable of modeling the dual period-varying preferences. Moreover, our model has been deployed online on Meituan Takeaway platform, leading to an average improvement in GMV (Gross Merchandise Value) of 0.70%.
Yuting Zhang 0010, Yiqing Wu, Ran Le, Yongchun Zhu, Fuzhen Zhuang, Ruidong Han, Xiang Li 0067, Wei Lin 0022, Zhulin An, Yongjun Xu 0001
KDD7
2023 Improving Zero-shot Visual Question Answering via Large Language Models with Reasoning Question Prompts
abstract
Zero-shot Visual Question Answering (VQA) is a prominent vision-language task that examines both the visual and textual understanding capability of systems in the absence of training data. Recently, by converting the images into captions, information across multi-modalities is bridged and Large Language Models (LLMs) can apply their strong zero-shot generalization capability to unseen questions. To design ideal prompts for solving VQA via LLMs, several studies have explored different strategies to select or generate question-answer pairs as the exemplar prompts, which guide LLMs to answer the current questions effectively. However, they totally ignore the role of question prompts. The original questions in VQA tasks usually encounter ellipses and ambiguity which require intermediate reasoning. To this end, we present Reasoning Question Prompts for VQA tasks, which can further activate the potential of LLMs in zero-shot scenarios. Specifically, for each question, we first generate self-contained questions as reasoning question prompts via an unsupervised question edition module considering sentence fluency, semantic integrity and syntactic invariance. Each reasoning question prompt clearly indicates the intent of the original question. This results in a set of candidate answers. Then, the candidate answers associated with their confidence scores acting as answer heuristics are fed into LLMs and produce the final answer. We evaluate reasoning question prompts on three VQA challenges, experimental results demonstrate that they can significantly improve the results of LLMs on zero-shot setting and outperform existing state-of-the-art zero-shot methods on three out of four data sets. Our source code is publicly released at https://github.com/ECNU-DASE-NLP/RQP.
Yunshi Lan, Xiang Li 0067, Xin Liu 0086, Yang Li 0218, Weining Qian
ACM Multimedia2
2023 LD2: Scalable Heterophilous Graph Neural Network with Decoupled Embeddings
abstract
Heterophilous Graph Neural Network (GNN) is a family of GNNs that specializes in learning graphs under heterophily, where connected nodes tend to have different labels. Most existing heterophilous models incorporate iterative non-local computations to capture node relationships. However, these approaches have limited application to large-scale graphs due to their high computational costs and challenges in adopting minibatch schemes. In this work, we study the scalability issues of heterophilous GNN and propose a scalable model, LD2, which simplifies the learning process by decoupling graph propagation and generating expressive embeddings prior to training. Theoretical analysis demonstrates that LD2 achieves optimal time complexity in training, as well as a memory footprint that remains independent of the graph scale. We conduct extensive experiments to showcase that our model is capable of lightweight minibatch training on large-scale heterophilous graphs, with up to $15\times$ speed improvement and efficient memory utilization, while maintaining comparable or better performance than the baselines.
Ningyi Liao, Siqiang Luo, Xiang Li 0067, Jieming Shi 0001
NeurIPS3
2023 DFRD: Data-Free Robustness Distillation for Heterogeneous Federated Learning
abstract
Federated Learning (FL) is a privacy-constrained decentralized machine learning paradigm in which clients enable collaborative training without compromising private data. However, how to learn a robust global model in the data-heterogeneous and model-heterogeneous FL scenarios is challenging. To address it, we resort to data-free knowledge distillation to propose a new FL method (namely DFRD). DFRD equips a conditional generator on the server to approximate the training space of the local models uploaded by clients, and systematically investigates its training in terms of fidelity, transferability and diversity. To overcome the catastrophic forgetting of the global model caused by the distribution shifts of the generator across communication rounds, we maintain an exponential moving average copy of the generator on the server. Additionally, we propose dynamic weighting and label sampling to accurately extract knowledge from local models. Finally, our extensive experiments on various image classification tasks illustrate that DFRD achieves significant performance gains compared to SOTA baselines.
Kangyang Luo, Yexuan Fu, Xiang Li 0067, Yunshi Lan, Ming Gao 0001
NeurIPS4
2023 Towards Robust Chinese Spelling Check Systems: Multi-round Error Correction with Ensemble Enhancement
Xiang Li 0067, Hanyue Du, Yike Zhao, Yunshi Lan
NLPCC (3)1
2023 Heterogeneous Knowledge Fusion: A Novel Approach for Personalized Recommendation via LLM
abstract
The analysis and mining of user heterogeneous behavior are of paramount importance in recommendation systems. However, the conventional approach of incorporating various types of heterogeneous behavior into recommendation models leads to feature sparsity and knowledge fragmentation issues. To address this challenge, we propose a novel approach for personalized recommendation via Large Language Model (LLM), by extracting and fusing heterogeneous knowledge from user heterogeneous behavior information. In addition, by combining heterogeneous knowledge and recommendation tasks, instruction tuning is performed on LLM for personalized recommendations. The experimental results demonstrate that our method can effectively integrate user heterogeneous behavior and significantly improve recommendation performance.
Bin Yin 0004, Zixiang Ding, Zhichao Feng, Xiang Li 0067, Wei Lin 0022
RecSys6
2023 Heterogeneous Graph Contrastive Learning with Meta-path Contexts and Weighted Negative Samples
abstract
Heterogeneous graph contrastive learning has received wide attention recently. Some existing methods use meta-paths, which are sequences of object types that capture semantic relationships between objects, to construct contrastive views. However, most of them ignore the rich meta-path context information that describes how two objects are connected by meta-paths. On the other hand, they fail to distinguish hard negatives from false negatives, which could adversely affect the model performance. To address the problems, we propose MEOW, a heterogeneous graph contrastive learning model that considers both meta-path contexts and weighted negative samples. Specifically, MEOW constructs a coarse view and a fine-grained view for contrast. The former reflects which objects are connected by meta-paths, while the latter uses meta-path contexts and characterizes the details on how the objects are connected. We take node embeddings in the coarse view as anchors, and construct positive and negative samples from the fine-grained view. Further, to distinguish hard negatives from false negatives, we learn weights of negative samples based on node clustering. We also use prototypical contrastive learning to pull close embeddings of nodes in the same cluster. Finally, we conduct extensive experiments to show the superiority of MEOW against other state-of-the-art methods.
Jianxiang Yu 0001, Xiang Li 0067
SDM2
2023 Interactive Recommendation System for Meituan Waimai
abstract
As the largest local retail & instant delivery platform in China, Meituan Waimai has deployed a personalized recommender system on server and recommend nearby stores to users through APP homepage. To capture real-time intention of users and flexibly adjust the recommendation results on the homepage, we further add an interactive recommender system. The existing interactive recommender systems in the industry mainly capture intention of users based on their feedback on a specific UI of questions. However, we find that it will undermine use fluency and increase use complexity by rashly inserting a new question UI when users browse the homepage. Therefore, we develop an Embedded Interactive Recommender System (EIRS) that directly infers users' intention according to their click behaviors on the homepage and dynamically inserts a new recommendation result into the homepage1. To demonstrate the effectiveness of EIRS, we conduct systematic online A/B Tests, where click-through & conversion rate of the inserted EIRS result is 132% higher than that of the initial result on the homepage, and the overall gross merchandise volume is effectively enhanced by 0.43%.
Rui Li 0044, Fei Jiang 0009, Xiang Li 0067, Wei Lin 0022, Wei Wang 0468
SIGIR5
2023 CEC: Towards Learning Global Optimized Recommendation through Causality Enhanced Conversion Model
abstract
Most e-commerce platforms consist of multiple entries (e.g., recommendation, search, shopping cart and etc.) for users to purchase their liked items. Among the research on the recommendation entry, most of them focus on improving the conversion volumes merely in the recommendation entry. However, such way could not ensure an increase in the global conversion volumes of the e-commerce platform. To achieve this goal by optimizing the recommendation entry only, in this paper, we focus on modeling the causality between the recommendation-entry-impression and the conversion by proposing the two-stage Causality Enhanced Conversion (CEC) model. In the first stage, we define the recommendation-entry-impression as treatment, then we estimate the conversion rate conditioned on the inclusion or exclusion of treatment respectively and calculate the corresponding individual treatment effect (ITE). In the second stage, we propose a propensity-normalization (PN) based method to transform the learned ITE to a weight term for instance weighting in the conversion loss. Extensive offline and online experiments on a large-scale food e-commerce scenario demonstrate that the CEC model could focus more on those conversed instances that can improve the global conversion volumes of the platform.
Ran Le, Guoqing Jiang, Xiufeng Shu, Ruidong Han, Qianzhong Li, Xiang Li 0067, Wei Lin 0022
SIGIR7
2023 Dual Intent Enhanced Graph Neural Network for Session-based New Item Recommendation
abstract
Recommender systems are essential to various fields, e.g., e-commerce, e-learning, and streaming media. At present, graph neural networks (GNNs) for session-based recommendations normally can only recommend items existing in users’ historical sessions. As a result, these GNNs have difficulty recommending items that users have never interacted with (new items), which leads to a phenomenon of information cocoon. Therefore, it is necessary to recommend new items to users. As there is no interaction between new items and users, we cannot include new items when building session graphs for GNN session-based recommender systems. Thus, it is challenging to recommend new items for users when using GNN-based methods. We regard this challenge as “GNN Session-based New Item Recommendation (GSNIR)”. To solve this problem, we propose a dual-intent enhanced graph neural network for it. Due to the fact that new items are not tied to historical sessions, the users’ intent is difficult to predict. We design a dual-intent network to learn user intent from an attention mechanism and the distribution of historical data respectively, which can simulate users’ decision-making process in interacting with a new item. To solve the challenge that new items cannot be learned by GNNs, inspired by zero-shot learning (ZSL), we infer the new item representation in GNN space by using their attributes. By outputting new item probabilities, which contain recommendation scores of the corresponding items, the new items with higher scores are recommended to users. Experiments on two representative real-world datasets show the superiority of our proposed method. The case study from the real-world verifies interpretability benefits brought by the dual-intent module and the new item reasoning module.
Di Jin 0001, Luzhi Wang, Yizhen Zheng, Guojie Song, Fei Jiang 0009, Xiang Li 0067, Wei Lin 0022, Shirui Pan
WWW6
2023 SeeGera: Self-supervised Semi-implicit Graph Variational Auto-encoders with Masking
abstract
Generative graph self-supervised learning (SSL) aims to learn node representations by reconstructing the input graph data. However, most existing methods focus on unsupervised learning tasks only and very few work has shown its superiority over the state-of-the-art graph contrastive learning (GCL) models, especially on the classification task. While a very recent model has been proposed to bridge the gap, its performance on unsupervised learning tasks is still unknown. In this paper, to comprehensively enhance the performance of generative graph SSL against other GCL models on both unsupervised and supervised learning tasks, we propose the SeeGera model, which is based on the family of self-supervised variational graph auto-encoder (VGAE). Specifically, SeeGera adopts the semi-implicit variational inference framework, a hierarchical variational framework, and mainly focuses on feature reconstruction and structure/feature masking. On the one hand, SeeGera co-embeds both nodes and features in the encoder and reconstructs both links and features in the decoder. Since feature embeddings contain rich semantic information on features, they can be combined with node embeddings to provide fine-grained knowledge for feature reconstruction. On the other hand, SeeGera adds an additional layer for structure/feature masking to the hierarchical variational framework, which boosts the model generalizability. We conduct extensive experiments comparing SeeGera with 9 other state-of-the-art competitors. Our results show that SeeGera can compare favorably against other state-of-the-art GCL methods in a variety of unsupervised and supervised learning tasks.
Xiang Li 0067, Tiandi Ye, Dongsheng Li 0002, Ming Gao 0001
WWW1
2022 Lexical Knowledge Internalization for Neural Dialog Generation
abstract
We propose knowledge internalization (KI), which aims to complement the lexical knowledge into neural dialog models.Instead of further conditioning the knowledge-grounded dialog (KGD) models on externally retrieved knowledge, we seek to integrate knowledge about each input token internally into the model's parameters.To tackle the challenge due to the large scale of lexical knowledge, we adopt the contrastive learning approach and create an effective token-level lexical knowledge retriever that requires only weak supervision mined from Wikipedia.We demonstrate the effectiveness and general applicability of our approach on various datasets and diversified model structures.
Zhiyong Wu 0003, Wei Bi, Xiang Li 0067, Lingpeng Kong, Ben Kao
ACL (1)3
2022 A Neural Network Architecture for Program Understanding Inspired by Human Behaviors
abstract
Program understanding is a fundamental task in program language processing.Despite the success, existing works fail to take human behaviors as reference in understanding programs.In this paper, we consider human behaviors and propose the PGNN-EK model that consists of two main components.On the one hand, inspired by the "divide-and-conquer" reading behaviors of humans, we present a partitioningbased graph neural network model PGNN on the upgraded AST of codes.On the other hand, to characterize human behaviors of resorting to other resources to help code comprehension, we transform raw codes with external knowledge and apply pre-training techniques for information extraction.Finally, we combine the two embeddings generated from the two components to output code embeddings.We conduct extensive experiments to show the superior performance of PGNN-EK on the code summarization and code clone detection tasks.In particular, to show the generalization ability of our model, we release a new dataset that is more challenging for code clone detection and could advance the development of the community.
Renyu Zhu, Xiang Li 0067, Ming Gao 0001, Wenyuan Cai
ACL (1)3
2022 Knowledge Prompting in Pre-trained Language Model for Natural Language Understanding
abstract
Knowledge-enhanced Pre-trained Language Model (PLM) has recently received significant attention, which aims to incorporate factual knowledge into PLMs.However, most existing methods modify the internal structures of fixed types of PLMs by stacking complicated modules, and introduce redundant and irrelevant factual knowledge from knowledge bases (KBs).In this paper, to address these problems, we introduce a seminal knowledge prompting paradigm and further propose a knowledge-prompting-based PLM framework KP-PLM.This framework can be flexibly combined with existing mainstream PLMs.Specifically, we first construct a knowledge sub-graph from KBs for each context.Then we design multiple continuous prompts rules and transform the knowledge sub-graph into natural language prompts.To further leverage the factual knowledge from these prompts, we propose two novel knowledge-aware self-supervised tasks including prompt relevance inspection and masked prompt modeling.Extensive experiments on multiple natural language understanding (NLU) tasks show the superiority of KP-PLM over other state-of-the-art methods in both full-resource and low-resource settings 1 .
Jianing Wang 0002, Wenkang Huang, Minghui Qiu, Qiuhui Shi, Xiang Li 0067, Ming Gao 0001
EMNLP6
2022 An End-to-End Chinese Text Normalization Model Based on Rule-Guided Flat-Lattice Transformer
abstract
Text normalization, defined as a procedure transforming nonstandard words to spoken-form words, is crucial to the intelligibility of synthesized speech in text-to-speech system. Rule-based methods without considering context can not eliminate ambiguation, whereas sequence-to-sequence neural network based methods suffer from the unexpected and uninterpretable errors problem. Recently proposed hybrid system treats rule-based model and neural model as two cascaded sub-modules, where limited interaction capability makes neural network model cannot fully utilize expert knowledge contained in the rules. Inspired by Flat-LAttice Transformer (FLAT), we propose an end-to-end Chinese text normalization model, which accepts Chinese characters as direct input and integrates expert knowledge contained in rules into the neural network, both contribute to the superior performance of proposed model for the text normalization task. We also release a first publicly accessible large-scale dataset for Chinese text normalization. Our proposed model has achieved excellent results on this dataset.
Wenlin Dai, Changhe Song, Xiang Li 0067, Zhiyong Wu 0001, Huashan Pan, Xiulin Li, Helen M. Meng
ICASSP3
2022 Finding Global Homophily in Graph Neural Networks When Meeting Heterophily
abstract
We investigate graph neural networks on graphs with heterophily. Some existing methods amplify a node’s neighborhood with multi-hop neighbors to include more nodes with homophily. However, it is a significant challenge to set personalized neighborhood sizes for different nodes. Further, for other homophilous nodes excluded in the neighborhood, they are ignored for information aggregation. To address these problems, we propose two models GloGNN and GloGNN++, which generate a node’s embedding by aggregating information from global nodes in the graph. In each layer, both models learn a coefficient matrix to capture the correlations between nodes, based on which neighborhood aggregation is performed. The coefficient matrix allows signed values and is derived from an optimization problem that has a closed-form solution. We further accelerate neighborhood aggregation and derive a linear time complexity. We theoretically explain the models’ effectiveness by proving that both the coefficient matrix and the generated node embedding matrix have the desired grouping effect. We conduct extensive experiments to compare our models against 11 other competitors on 15 benchmark datasets in a wide range of domains, scales and graph heterophilies. Experimental results show that our methods achieve superior performance and are also very efficient.
Xiang Li 0067, Renyu Zhu, Yao Cheng 0009, Siqiang Luo, Dongsheng Li 0002, Weining Qian
ICML1
2022 RAW-GNN: RAndom Walk Aggregation based Graph Neural Network
abstract
Graph-Convolution-based methods have been successfully applied to representation learning on homophily graphs where nodes with the same label or similar attributes tend to connect with one another. Due to the homophily assumption of Graph Convolutional Networks (GCNs) that these methods use, they are not suitable for heterophily graphs where nodes with different labels or dissimilar attributes tend to be adjacent. Several methods have attempted to address this heterophily problem, but they do not change the fundamental aggregation mechanism of GCNs because they rely on summation operators to aggregate information from neighboring nodes, which is implicitly subject to the homophily assumption. Here, we introduce a novel aggregation mechanism and develop a RAndom Walk Aggregation-based Graph Neural Network (called RAW-GNN) method. The proposed approach integrates the random walk strategy with graph neural networks. The new method utilizes breadth-first random walk search to capture homophily information and depth-first search to collect heterophily information. It replaces the conventional neighborhoods with path-based neighborhoods and introduces a new path-based aggregator based on Recurrent Neural Networks. These designs make RAW-GNN suitable for both homophily and heterophily graphs. Extensive experimental results showed that the new method achieved state-of-the-art performance on a variety of homophily and heterophily graphs.
Di Jin 0001, Rui Wang 0102, Meng Ge, Dongxiao He, Xiang Li 0067, Wei Lin 0022, Weixiong Zhang
IJCAI5
2022 CGMN: A Contrastive Graph Matching Network for Self-Supervised Graph Similarity Learning
abstract
Graph similarity learning refers to calculating the similarity score between two graphs, which is required in many realistic applications, such as visual tracking, graph classification, and collaborative filtering. As most of the existing graph neural networks yield effective graph representations of a single graph, little effort has been made for jointly learning two graph representations and calculating their similarity score. In addition, existing unsupervised graph similarity learning methods are mainly clustering-based, which ignores the valuable information embodied in graph pairs. To this end, we propose a contrastive graph matching network (CGMN) for self-supervised graph similarity learning in order to calculate the similarity between any two input graph objects. Specifically, we generate two augmented views for each graph in a pair respectively. Then, we employ two strategies, namely cross-view interaction and cross-graph interaction, for effective node representation learning. The former is resorted to strengthen the consistency of node representations in two views. The latter is utilized to identify node differences between different graphs. Finally, we transform node representations into graph-level representations via pooling operations for graph similarity computation. We have evaluated CGMN on eight real-world datasets, and the experiment results show that the proposed new approach is superior to the state-of-the-art methods in graph similarity learning downstream tasks.
Di Jin 0001, Luzhi Wang, Yizhen Zheng, Xiang Li 0067, Fei Jiang 0009, Wei Lin 0022, Shirui Pan
IJCAI4
2022 A Comprehensive Benchmark of Deep Learning Libraries on Mobile Devices
abstract
Deploying deep learning (DL) on mobile devices has been a notable trend in recent years. To support fast inference of on-device DL, DL libraries play a critical role as algorithms and hardware do. Unfortunately, no prior work ever dives deep into the ecosystem of modern DL libs and provides quantitative results on their performance. In this paper, we first build a comprehensive benchmark that includes 6 representative DL libs and 15 diversified DL models. We then perform extensive experiments on 10 mobile devices, which help reveal a complete landscape of the current mobile DL libs ecosystem. For example, we find that the best-performing DL lib is severely fragmented across different models and hardware, and the gap between those DL libs can be rather huge. In fact, the impacts of DL libs can overwhelm the optimizations from algorithms or hardware, e.g., model quantization and GPU/DSP-based heterogeneous computing. Finally, atop the observations, we summarize practical implications to different roles in the DL lib ecosystem.
Qiyang Zhang 0001, Xiang Li 0067, Xiangying Che, Xiao Ma 0009, Ao Zhou 0001, Mengwei Xu 0001, Shangguang Wang, Yun Ma 0002, Xuanzhe Liu
WWW2
2022 SCARA: Scalable Graph Neural Networks with Feature-Oriented Optimization
abstract
Recent advances in data processing have stimulated the demand for learning graphs of very large scales. Graph Neural Networks (GNNs), being an emerging and powerful approach in solving graph learning tasks, are known to be difficult to scale up. Most scalable models apply node-based techniques in simplifying the expensive graph message-passing propagation procedure of GNN. However, we find such acceleration insufficient when applied to million- or even billion-scale graphs. In this work, we propose SCARA, a scalable GNN with feature-oriented optimization for graph computation. SCARA efficiently computes graph embedding from node features, and further selects and reuses feature computation results to reduce overhead. Theoretical analysis indicates that our model achieves sub-linear time complexity with a guaranteed precision in propagation process as well as GNN training and inference. We conduct extensive experiments on various datasets to evaluate the efficacy and efficiency of SCARA. Performance comparison with baselines shows that SCARA can reach up to 100x graph propagation acceleration than current state-of-the-art methods with fast convergence and comparable accuracy. Most notably, it is efficient to process precomputation on the largest available billion-scale GNN dataset Papers100M (111M nodes, 1.6B edges) in 100 seconds.
Ningyi Liao, Dingheng Mo, Siqiang Luo, Xiang Li 0067
Proc. VLDB Endow.4
2022 SCHAIN-IRAM: An Efficient and Effective Semi-Supervised Clustering Algorithm for Attributed Heterogeneous Information Networks
abstract
A heterogeneous information network (HIN) is one whose nodes model objects of different types and whose links model objects’ relationships. To enrich its information, objects in an HIN are typically associated with additional attributes. We call such an HIN anAttributed HINor AHIN. We study the problem of clustering objects in an AHIN, taking into account objects’ similarities with respect to both object attribute values and their structural connectedness in the network. We show how supervision signal, expressed in the form of amust-link setand acannot-link set, can be leveraged to improve clustering results. We put forward the SCHAIN algorithm to solve the clustering problem, and two highly efficient variants, SCHAIN-PI and SCHAIN-IRAM, which employ thepower iteration based methodand theimplicitly restarted Arnoldi methodrespectively to compute eigenvectors of a matrix. We conduct extensive experiments comparing SCHAIN-based algorithms with other state-of-the-art clustering algorithms. Our results show that SCHAIN-IRAM outperforms other competitors in terms of clustering effectiveness and is highly efficient.
Xiang Li 0067, Martin Ester, Ben Kao, Xin Wang 0019, Yudian Zheng
IEEE Trans. Knowl. Data Eng.1
2021 Good for Misconceived Reasons: An Empirical Revisiting on the Need for Visual Context in Multimodal Machine Translation
abstract
Zhiyong Wu, Lingpeng Kong, Wei Bi, Xiang Li, Ben Kao. Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (Volume 1: Long Papers). 2021.
Zhiyong Wu 0003, Lingpeng Kong, Wei Bi, Xiang Li 0067, Ben Kao
ACL/IJCNLP (1)4
2021 SceneRec: Scene-Based Graph Neural Networks for Recommender Systems
Xiang Li 0067, Dawei Yin 0001, Shuai Ma 0001
EDBT3
2021 Leveraging Meta-path Contexts for Classification in Heterogeneous Information Networks
abstract
A heterogeneous information network (HIN) has as vertices objects of different types and as edges the relations between objects, which are also of various types. We study the problem of classifying objects in HINs. Most existing methods perform poorly when given scarce labeled objects as training sets, and methods that improve classification accuracy under such scenarios are often computationally expensive. To address these problems, we propose ConCH, a graph neural network model. ConCH formulates the classification problem as a multitask learning problem that combines semi-supervised learning with self-supervised learning to learn from both labeled and unlabeled data. ConCH employs meta-paths, which are sequences of object types that capture semantic relationships between objects. ConCH co-derives object embeddings and context embeddings via graph convolution. It also uses the attention mechanism to fuse such embeddings. We conduct extensive experiments to evaluate the performance of ConCH against other 15 classification methods. Our results show that ConCH is an effective and efficient method for HIN classification.
Xiang Li 0067, Danhao Ding, Ben Kao, Yizhou Sun, Nikos Mamoulis
ICDE1
2021 Reinforcement Learning Enhanced Explainer for Graph Neural Networks
abstract
Graph neural networks (GNNs) have recently emerged as revolutionary technologies for machine learning tasks on graphs. In GNNs, the graph structure is generally incorporated with node representation via the message passing scheme, making the explanation much more challenging. Given a trained GNN model, a GNN explainer aims to identify a most influential subgraph to interpret the prediction of an instance (e.g., a node or a graph), which is essentially a combinatorial optimization problem over graph. The existing works solve this problem by continuous relaxation or search-based heuristics. But they suffer from key issues such as violation of message passing and hand-crafted heuristics, leading to inferior interpretability. To address these issues, we propose a RL-enhanced GNN explainer, RG-Explainer, which consists of three main components: starting point selection, iterative graph generation and stopping criteria learning. RG-Explainer could construct a connected explanatory subgraph by sequentially adding nodes from the boundary of the current generated graph, which is consistent with the message passing scheme. Further, we design an effective seed locator to select the starting point, and learn stopping criteria to generate superior explanations. Extensive experiments on both synthetic and real datasets show that RG-Explainer outperforms state-of-the-art GNN explainers. Moreover, RG-Explainer can be applied in the inductive setting, demonstrating its better generalization ability.
Yifei Shen 0004, Yao Zhang 0009, Xiang Li 0067, Dongsheng Li 0002
NeurIPS4
2021 Disentangling User Interest and Conformity for Recommendation with Causal Embedding
abstract
Recommendation models are usually trained on observational interaction data. However, observational interaction data could result from users’ conformity towards popular items, which entangles users’ real interest. Existing methods tracks this problem as eliminating popularity bias, e.g., by re-weighting training samples or leveraging a small fraction of unbiased data. However, the variety of user conformity is ignored by these approaches, and different causes of an interaction are bundled together as unified representations, hence robustness and interpretability are not guaranteed when underlying causes are changing. In this paper, we present DICE, a general framework that learns representations where interest and conformity are structurally disentangled, and various backbone recommendation models could be smoothly integrated. We assign users and items with separate embeddings for interest and conformity, and make each embedding capture only one cause by training with cause-specific data which is obtained according to the colliding effect of causal inference. Our proposed methodology outperforms state-of-the-art baselines with remarkable improvements on two real-world datasets on top of various backbone models. We further demonstrate that the learned embeddings successfully capture the desired causes, and show that DICE guarantees the robustness and interpretability of recommendation.
Yu Zheng 0010, Chen Gao 0001, Xiang Li 0067, Xiangnan He 0001, Yong Li 0008, Depeng Jin
WWW3
2020 A General Early-Stopping Module for Crowdsourced Ranking
Leong Hou U, Nikos Mamoulis, Reynold Cheng, Xiang Li 0067
DASFAA (2)5
2020 An End-to-End Deep RL Framework for Task Arrangement in Crowdsourcing Platforms
abstract
In this paper, we propose a Deep Reinforcement Learning (RL) framework for task arrangement, which is a critical problem for the success of crowdsourcing platforms. Previous works conduct the personalized recommendation of tasks to workers via supervised learning methods. However, the majority of them only consider the benefit of either workers or requesters independently. In addition, they do not consider the real dynamic environments (e.g., dynamic tasks, dynamic workers), so they may produce sub-optimal results. To address these issues, we utilize Deep Q-Network (DQN), an RL-based method combined with a neural network to estimate the expected long-term return of recommending a task. DQN inherently considers the immediate and the future rewards and can be updated quickly to deal with evolving data and dynamic changes. Furthermore, we design two DQNs that capture the benefit of both workers and requesters and maximize the profit of the platform. To learn value functions in DQN effectively, we also propose novel state representations, carefully design the computation of Q values, and predict transition probabilities and future states. Experiments on synthetic and real datasets demonstrate the superior performance of our framework.
Nikos Mamoulis, Reynold Cheng, Guoliang Li 0001, Xiang Li 0067, Yuqiu Qian
ICDE5
2020 CAST: A Correlation-based Adaptive Spectral Clustering Algorithm on Multi-scale Data
abstract
We study the problem of applying spectral clustering to cluster multi-scale data, which is data whose clusters are of various sizes and densities. Traditional spectral clustering techniques discover clusters by processing a similarity matrix that reflects the proximity of objects. For multi-scale data, distance-based similarity is not effective because objects of a sparse cluster could be far apart while those of a dense cluster have to be sufficiently close. Following [16], we solve the problem of spectral clustering on multi-scale data by integrating the concept of objects' "reachability similarity" with a given distance-based similarity to derive an objects' coefficient matrix. We propose the algorithm CAST that applies trace Lasso to regularize the coefficient matrix. We prove that the resulting coefficient matrix has the "grouping effect" and that it exhibits "sparsity". We show that these two characteristics imply very effective spectral clustering. We evaluate CAST and 10 other clustering methods on a wide range of datasets w.r.t. various measures. Experimental results show that CAST provides excellent performance and is highly robust across test cases of multi-scale data.
Xiang Li 0067, Ben Kao, Dawei Yin 0001, Martin Ester
KDD1
2019 Spectral Clustering in Heterogeneous Information Networks
abstract
A heterogeneous information network (HIN) is one whose objects are of different types and links between objects could model different object relations. We study how spectral clustering can be effectively applied to HINs. In particular, we focus on how meta-path relations are used to construct an effective similarity matrix based on which spectral clustering is done. We formulate the similarity matrix construction as an optimization problem and propose the SClump algorithm for solving the problem. We conduct extensive experiments comparing SClump with other state-of-the-art clustering algorithms on HINs. Our results show that SClump outperforms the competitors over a range of datasets w.r.t. different clustering quality measures.
Xiang Li 0067, Ben Kao, Zhaochun Ren, Dawei Yin 0001
AAAI1
2018 ROSC: Robust Spectral Clustering on Multi-scale Data
abstract
We investigate the effectiveness of spectral methods in clustering multi-scale data, which is data whose clusters are of various sizes and densities. We review existing spectral methods that are designed to handle multi-scale data and propose an alternative approach that is orthogonal to existing methods. We put forward the algorithm ROSC, which computes an affinity matrix that takes into account both objects' feature similarity and reachability similarity. We perform extensive experiments comparing ROSC against 9 other methods on both real and synthetic datasets. Our results show that ROSC performs very well against the competitors. In particular, it is very robust in that it consistently performs well over all the datasets tested. Also, it outperforms others by wide margins for datasets that are highly multi-scale.
Xiang Li 0067, Ben Kao, Siqiang Luo, Martin Ester
WWW1
2017 Semi-supervised Clustering in Attributed Heterogeneous Information Networks
abstract
A heterogeneous information network (HIN) is one whose nodes model objects of different types and whose links model objects' relationships. In many applications, such as social networks and RDF-based knowledge bases, information can be modeled as HINs. To enrich its information content, objects (as represented by nodes) in an HIN are typically associated with additional attributes. We call such an HIN an Attributed HIN or AHIN. We study the problem of clustering objects in an AHIN, taking into account objects' similarities with respect to both object attribute values and their structural connectedness in the network. We show how supervision signal, expressed in the form of a must-link set and a cannot-link set, can be leveraged to improve clustering results. We put forward the SCHAIN algorithm to solve the clustering problem. We conduct extensive experiments comparing SCHAIN with other state-of-the-art clustering algorithms and show that SCHAIN outperforms the others in clustering quality.
Xiang Li 0067, Martin Ester, Ben Kao, Xin Wang 0019, Yudian Zheng
WWW1
2016 On Transductive Classification in Heterogeneous Information Networks
abstract
A heterogeneous information network (HIN) is used to model objects of different types and their relationships. Objects are often associated with properties such as labels. In many applications, such as curated knowledge bases for which object labels are manually given, only a small fraction of the objects are labeled. Studies have shown that transductive classification is an effective way to classify and to deduce labels of objects, and a number of transductive classifiers have been put forward to classify objects in an HIN. We study the performance of a few representative transductive classification algorithms on HINs. We identify two fundamental properties, namely, cohesiveness and connectedness, of an HIN that greatly influence the effectiveness of transductive classifiers. We define metrics that measure the two properties. Through experiments, we show that the two properties serve as very effective indicators that predict the accuracy of transductive classifiers. Based on cohesiveness and connectedness we derive (1) a black-box tester that evaluates whether transductive classifiers should be applied for a given classification task and (2) an active learning algorithm that identifies the objects in an HIN whose labels should be sought in order to improve classification accuracy.
Xiang Li 0067, Ben Kao, Yudian Zheng, Zhipeng Huang 0001
CIKM1
2016 Meta Structure: Computing Relevance in Large Heterogeneous Information Networks
abstract
A heterogeneous information network (HIN) is a graph model in which objects and edges are annotated with types. Large and complex databases, such as YAGO and DBLP, can be modeled as HINs. A fundamental problem in HINs is the computation of closeness, or relevance, between two HIN objects. Relevance measures can be used in various applications, including entity resolution, recommendation, and information retrieval. Several studies have investigated the use of HIN information for relevance computation, however, most of them only utilize simple structure, such as path, to measure the similarity between objects. In this paper, we propose to use meta structure, which is a directed acyclic graph of object types with edge types connecting in between, to measure the proximity between objects. The strength of meta structure is that it can describe complex relationship between two HIN objects (e.g., two papers in DBLP share the same authors and topics). We develop three relevance measures based on meta structure. Due to the computational complexity of these measures, we further design an algorithm with data structures proposed to support their evaluation. Our extensive experiments on YAGO and DBLP show that meta structure-based relevance is more effective than state-of-the-art approaches, and can be efficiently computed.
Zhipeng Huang 0001, Yudian Zheng, Reynold Cheng, Yizhou Sun, Nikos Mamoulis, Xiang Li 0067
KDD6
2016 T-DesP: Destination Prediction Based on Big Trajectory Data
abstract
Destination prediction is very important in location-based services such as recommendation of targeted advertising location. Most current approaches always predict destination according to existing trip based on history trajectories. However, no existing work has considered the difference between the effects of passing-by locations and the destination in history trajectories, which seriously impacts the accuracy of predicted results as the destination can indicate the purpose of traveling. Meanwhile, the temporal information of history trajectories in destination prediction plays an important role. On one hand, the history trajectories in different periods also differ in the influence, e.g., the history trajectories from last week can reflect the status quo more accurately than the history trajectories two years ago. On the other hand, the history trajectories in different time slots reflect different facts of traffic and moving habits of people, e.g., visiting a restaurant in the daytime and visiting a bar at night. Although a huge amount of history trajectories can be achieved in the era of big data, it is still far from covering all the query trajectories since a road network is widely distributed and trajectory data is sparse. The temporal sensitivity of history trajectories highlights the sparsity problem even more. Therefore, we propose a novel model T-DesP to solve the aforementioned problems. The model is comprised of two modules: trajectory learning and destination prediction. In the module of trajectory learning, a novel method called the mirror absorbing Markov chain model is proposed for modeling the trajectories for isolating the destination. We build a transition tensor to deduce the transition probability between each location pair in a particular time slot. To address the data sparsity problem, we fill the missing values in transition tensor through a context-aware tensor decomposition approach. In the module of destination prediction, an absorbing tensor is derived from the filled transition tensor, and the theoretical model is established for destination prediction. The experiments prove the effectiveness and efficiency of T-DesP.
Xiang Li 0067, Yue-Jiao Gong, Xinglin Zhang 0001, Jian Yin 0001
IEEE Trans. Intell. Transp. Syst.1
2016 TORD Problem and Its Solution Based on Big Trajectories Data
abstract
Identifying a preferable route is an important problem in location-based service. A user may want to find “an optimal route such that it passes by mall, restaurant, and bank, meanwhile go through bank before mall, and arrive at the restaurant at mealtime.” However, none of the algorithms in the existing work on route planning can be used to answer such queries that exist in real life. Motivated by this, we define the problem of temporal-sensitive optimal route discovery (TORD), which is an NP-hard problem. Since the rise of GPS-equipped mobile devices has led to the emergence of big trajectory data, we propose a hybrid evaluation indicator of routes called preference combined by the distance from the road network and popularity from trajectories. To achieve this goal, we first develop an improved compact minimum bound rectangle algorithm to get the association between trajectory data and road network, which can update the association conveniently for new trajectories. Then, we construct a two-layer preference network that is used for mining the preference and travel time. Finally, we devise three approximation algorithms to answer TORD queries. The results of empirical studies show that all the proposed algorithms are capable of answering TORD queries efficiently.
Xiang Li 0067, Jian Yin 0001
IEEE Trans. Intell. Transp. Syst.2
2015 Classification with Active Learning and Meta-Paths in Heterogeneous Information Networks
abstract
A heterogeneous information network (HIN) is used to model objects of different types and their relationships. Meta-paths are sequences of object types. They are used to represent complex relationships between objects beyond what links in a homogeneous network capture. We study the problem of classifying objects in an HIN. We propose class-level meta-paths and study how they can be used to (1) build more accurate classifiers and (2) improve active learning in identifying objects for which training labels should be obtained. We show that class-level meta-paths and object classification exhibit interesting synergy. Our experimental results show that the use of class-level meta-paths results in very effective active learning and good classification performance in HINs.
Chang Wan, Xiang Li 0067, Ben Kao, Xiao Yu 0007, Quanquan Gu, David Wai-Lok Cheung, Jiawei Han 0001
CIKM2
2013 Novel user influence measurement based on user interaction in microblog
abstract
With the development of science and technology, various social networks have emerged in recent years and microblog is a prevailing one. This paper focuses on how to identify the most influential users quantitatively in microblog and proposes a new ranking method which employs the fact that a follower's contribution to the influences of his/her followees varies and depends greatly on the interactions between them. We consider bidirectional interactions from perspectives of followees and followers, and measure the interactive degree by four factors comprised of retweeting strength, commenting intensity, mentioning density and a special indicator to the potential interactions called keyword similarity. The experimental results show that our method based on user interaction is better in calculating the user influence.
Xiang Li 0067, Shaoyin Cheng, Fan Jiang 0005
ASONAM1
2013 Cleaning uncertain data for top-k queries
abstract
The information managed in emerging applications, such as sensor networks, location-based services, and data integration, is inherently imprecise. To handle data uncertainty, probabilistic databases have been recently developed. In this paper, we study how to quantify the ambiguity of answers returned by a probabilistic top-k query. We develop efficient algorithms to compute the quality of this query under the possible world semantics. We further address the cleaning of a probabilistic database, in order to improve top-k query quality. Cleaning involves the reduction of ambiguity associated with the database entities. For example, the uncertainty of a temperature value acquired from a sensor can be reduced, or cleaned, by requesting its newest value from the sensor. While this “cleaning operation” may produce a better query result, it may involve a cost and fail. We investigate the problem of selecting entities to be cleaned under a limited budget. Particularly, we propose an optimal solution and several heuristics. Experiments show that the greedy algorithm is efficient and close to optimal.
Luyi Mo, Reynold Cheng, Xiang Li 0067, David Wai-Lok Cheung, Xuan S. Yang
ICDE3
2013 Predicting Users' Age Range in Micro-blog Network
Chengyu Wang 0001, Xiang Li 0067, Rong Zhang 0002
WISE (1)3
2011 On Link-based Similarity Join
Liwen Sun, Reynold Cheng, Xiang Li 0067, David Wai-Lok Cheung, Jiawei Han 0001
Proc. VLDB Endow.3
2010 Explore or Exploit? Effective Strategies for Disambiguating Large Databases
abstract
Data ambiguity is inherent in applications such as data integration, location-based services, and sensor monitoring. In many situations, it is possible to "clean", or remove, ambiguities from these databases. For example, the GPS location of a user is inexact due to measurement errors, but context information (e.g., what a user is doing) can be used to reduce the imprecision of the location value. In order to obtain a database with a higher quality, we study how to disambiguate a database by appropriately selecting candidates to clean. This problem is challenging because cleaning involves a cost, is limited by a budget, may fail, and may not remove all ambiguities. Moreover, the statistical information about how likely database objects can be cleaned may not be precisely known. We tackle these challenges by proposing two types of algorithms. The first type makes use of greedy heuristics to make sensible decisions; however, these algorithms do not make use of cleaning information and require user input for parameters to achieve high cleaning effectiveness. We propose the Explore-Exploit (or EE ) algorithm, which gathers valuable information during the cleaning process to determine how the remaining cleaning budget should be invested. We also study how to fine-tune the parameters of EE in order to achieve optimal cleaning effectiveness. Experimental evaluations on real and synthetic datasets validate the effectiveness and efficiency of our approaches.
Reynold Cheng, Eric Lo 0001, Xuan S. Yang, Ming-Hay Luk, Xiang Li 0067, Xike Xie
Proc. VLDB Endow.5