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

Tianxin Wei

dblp:277/5800 · DBLP profile ↗
← Back
29ranked-venue papers
7as first author
28since 2021 · last 2026
0000-0003-4450-2005ORCID · verified

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

Artificial intelligence and machine learning · 25 · 7 first-author · 24 since 2021Databases, data management, data science and information retrieval · 12 · 4 first-author · 11 since 2021Graphics, computer vision, multimedia, augmented reality and games · 3 · 3 since 2021Applied, interdisciplinary, general and emerging computing · 2 · 2 since 2021

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

Artificial intelligence
18 papers
Graph learning · 20% Transfer learning and domain adaptation · 17% Efficient and distributed learning · 14%
Databases, data mining, and information retrieval
12 papers
Recommender systems · 59% Information retrieval · 29% Data mining · 12%

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

TopicWeightPapersLastEvidence papers
Machine learning › Transfer learning and domain adaptation
test-time adaptation
1.932026
Panda: Test-Time Adaptation with Negative Data Augmentation · AAAI 2026
Adaptive Test-Time Personalization for Federated Learning · NeurIPS 2023
Latte: Collaborative Test-Time Adaptation of Vision-Language Models in Federated Learning · ICCV 2025
Machine learning › Graph learning
graph neural network
1.622025
Graph4MM: Weaving Multimodal Learning with Structural Information · ICML 2025
Graph Mixup on Approximate Gromov-Wasserstein Geodesics · ICML 2024
Machine learning › Efficient and distributed learning
federated learning
1.522025
Latte: Collaborative Test-Time Adaptation of Vision-Language Models in Federated Learning · ICCV 2025
Adaptive Test-Time Personalization for Federated Learning · NeurIPS 2023
Machine learning › Efficient and distributed learning
model compression
1.522025
ResMoE: Space-efficient Compression of Mixture of Experts LLMs via Residual Restoration · KDD (1) 2025
NTK-approximating MLP Fusion for Efficient Language Model Fine-tuning · ICML 2023
Machine learning › Representation and self-supervised learning
contrastive learning
1.422025
Connecting Domains and Contrasting Samples: A Ladder for Domain Generalization · KDD (1) 2025
Augmentations in Hypergraph Contrastive Learning: Fabricated and Generative · NeurIPS 2022
Machine learning › Reinforcement learning › bandit
contextual bandit
1.422024
Meta Clustering of Neural Bandits · KDD 2024
Meta-Learning with Neural Bandit Scheduler · NeurIPS 2023
Machine learning › Deep learning architectures and training
data augmentation
1.322026
Panda: Test-Time Adaptation with Negative Data Augmentation · AAAI 2026
Connecting Domains and Contrasting Samples: A Ladder for Domain Generalization · KDD (1) 2025
Machine learning › Transfer learning and domain adaptation
meta-learning
1.122023
Meta-Learning with Neural Bandit Scheduler · NeurIPS 2023
Fast Adaptation for Cold-start Collaborative Filtering with Meta-learning · ICDM 2020
Recommender systems
cold-start recommendation
1.022022
Comprehensive Fair Meta-learned Recommender System · KDD 2022
Fast Adaptation for Cold-start Collaborative Filtering with Meta-learning · ICDM 2020
Natural language and speech › Language models and text generation
decoding
1.012026
AdaFuse: Adaptive Ensemble Decoding for Large Language Models · ACL (1) 2026
Computer vision › Vision and language
multimodal dialogue
1.012026
Mem-Gallery: Benchmarking Multimodal Long-Term Conversational Memory for MLLM Agents · ACL (1) 2026
Machine learning › Deep learning architectures and training › data augmentation
negative data augmentation
1.012026
Panda: Test-Time Adaptation with Negative Data Augmentation · AAAI 2026
Machine learning › Trustworthy machine learning
robustness
1.012026
Panda: Test-Time Adaptation with Negative Data Augmentation · AAAI 2026
Recommender systems
cross-domain recommendation
1.012026
FeDecider: An LLM-Based Framework for Federated Cross-Domain Recommendation · WWW 2026
Recommender systems
large language model-based recommendation
1.012026
FeDecider: An LLM-Based Framework for Federated Cross-Domain Recommendation · WWW 2026
Recommender systems
collaborative filtering
0.922021
Causal Intervention for Leveraging Popularity Bias in Recommendation · SIGIR 2021
Fast Adaptation for Cold-start Collaborative Filtering with Meta-learning · ICDM 2020
Machine learning › Transfer learning and domain adaptation
domain generalization
0.912025
Connecting Domains and Contrasting Samples: A Ladder for Domain Generalization · KDD (1) 2025
Machine learning › Kernel, tree and ensemble methods
ensemble learning
0.912025
Breaking Silos: Adaptive Model Fusion Unlocks Better Time Series Forecasting · ICML 2025
Natural language and speech › Question answering and dialogue systems › knowledge-intensive question answering › knowledge-grounded question answering
evidence-based question answering
0.912025
SelfElicit: Your Language Model Secretly Knows Where is the Relevant Evidence · ACL (1) 2025
Natural language and speech › Language models and text generation
large language model inference
0.912025
ResMoE: Space-efficient Compression of Mixture of Experts LLMs via Residual Restoration · KDD (1) 2025
Machine learning › Efficient and distributed learning › model compression › parameter compression
mixture-of-experts compression
0.912025
ResMoE: Space-efficient Compression of Mixture of Experts LLMs via Residual Restoration · KDD (1) 2025
Machine learning › Graph learning › graph fusion
multimodal graph fusion
0.912025
Graph4MM: Weaving Multimodal Learning with Structural Information · ICML 2025
Machine learning › Graph learning
multimodal graph learning
0.912025
Graph4MM: Weaving Multimodal Learning with Structural Information · ICML 2025
Natural language and speech › Language models and text generation
retrieval-augmented generation
0.912025
SelfElicit: Your Language Model Secretly Knows Where is the Relevant Evidence · ACL (1) 2025
Information retrieval › evaluation
benchmark
0.912025
CLIMB: Class-imbalanced Learning Benchmark on Tabular Data · NeurIPS 2025
Data mining › predictive modeling › classification
imbalanced classification
0.912025
CLIMB: Class-imbalanced Learning Benchmark on Tabular Data · NeurIPS 2025
Data mining
tabular data
0.912025
CLIMB: Class-imbalanced Learning Benchmark on Tabular Data · NeurIPS 2025
Data mining › time series analysis
time series forecasting
0.912025
Breaking Silos: Adaptive Model Fusion Unlocks Better Time Series Forecasting · ICML 2025
Machine learning › Graph learning › graph neural network
graph data augmentation
0.812024
Graph Mixup on Approximate Gromov-Wasserstein Geodesics · ICML 2024
Machine learning › Graph learning › graph neural network › graph data augmentation
graph mixup
0.812024
Graph Mixup on Approximate Gromov-Wasserstein Geodesics · ICML 2024

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

meta-learning · 3.0personalized aggregation · 2.0low-rank adaptation · 2.0test-time adaptation · 1.9negative data augmentation · 1.0multimodal agent evaluation · 1.0ensemble learning · 1.0benchmark construction · 1.0adaptive decoding · 1.0self-guided highlighting · 0.9rebalancing · 0.9meta-features · 0.9learnable fusor · 0.9federated learning · 0.9ensemble methods · 0.9attention scores · 0.9graph-based augmentation · 0.8evidence deep learning · 0.8
YearPublicationVenuePosition
2026 Panda: Test-Time Adaptation with Negative Data Augmentation
abstract
Pretrained vision-language models exhibit strong zero-shot classification capabilities, but their predictions degrade significantly under common image corruptions. To improve robustness, many test-time adaptation (TTA) methods adopt positive data augmentation (PDA), which generates multiple views of each test sample to reduce prediction variance. However, these methods suffer from two key limitations. First, it introduces considerable computational overhead due to the large number of augmentations required per image. Second, it fails to mitigate prediction bias, where the model tends to predict certain classes disproportionately under corruption, as PDA operates on corrupted inputs and typically does not remove the corruption itself. To address these challenges, we propose Panda, a novel TTA method based on negative data augmentation (NDA). Unlike positive augmentations that preserve object semantics, Panda generates negative augmentations by disrupting semantic content. It divides images into patches and randomly assembles them from a shared patch pool. These negatively augmented images retain corruption-specific features while discarding object-relevant signals. We then subtract the mean feature of these negative samples from the original image feature, effectively suppressing corruption-related components while preserving class-relevant information. This mitigates prediction bias under distribution shifts. Importantly, Panda allows augmentation to be shared across samples within a batch, resulting in minimal computational overhead. Panda can be seamlessly integrated into existing test-time adaptation frameworks and substantially improve their robustness. Our experiments indicate that Panda delivers superior performance compared to PDA methods, and a wide range of TTA methods exhibit significantly enhanced performance when integrated with Panda.
Ruxi Deng, Tianxin Wei, Jingrui He
AAAI3
2026 Mem-Gallery: Benchmarking Multimodal Long-Term Conversational Memory for MLLM Agents
abstract
Yuanchen Bei, Tianxin Wei, Xuying Ning, Yanjun Zhao, Zhining Liu, Xiao Lin, Yada Zhu, Hendrik Hamann, Jingrui He, Hanghang Tong. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026.
Yuanchen Bei, Tianxin Wei, Xuying Ning, Zhining Liu 0002, Xiao Lin 0016, Yada Zhu, Hendrik F. Hamann, Jingrui He, Hanghang Tong
ACL (1)2
2026 AdaFuse: Adaptive Ensemble Decoding for Large Language Models
abstract
Chengming Cui, Tianxin Wei, Ziyi Chen, Ruizhong Qiu, Zhichen Zeng, Zhining Liu, Xuying Ning, Duo Zhou, Jingrui He. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026.
Chengming Cui, Tianxin Wei, Ruizhong Qiu, Zhichen Zeng 0001, Zhining Liu 0002, Xuying Ning, Duo Zhou, Jingrui He
ACL (1)2
2026 FeDecider: An LLM-Based Framework for Federated Cross-Domain Recommendation
abstract
Federated cross-domain recommendation (Federated CDR) aims to collaboratively learn personalized recommendation models across heterogeneous domains while preserving data privacy. Recently, large language model (LLM)-based recommendation models have demonstrated impressive performance by leveraging LLMs' strong reasoning capabilities and broad knowledge. However, adopting LLM-based recommendation models in Federated CDR scenarios introduces new challenges. First, there exists a risk of overfitting with domain-specific local adapters. The magnitudes of locally optimized parameter updates often vary across domains, causing biased aggregation and overfitting toward domain-specific distributions. Second, unlike traditional recommendation models (e.g., collaborative filtering, bipartite graph-based methods) that learn explicit and comparable user/item representations, LLMs encode knowledge implicitly through autoregressive text generation training. This poses additional challenges for effectively measuring the cross-domain similarities under heterogeneity. To address these challenges, we propose an LLM-based framework for federated cross-domain recommendation, FeDecider. Specifically, FeDecider tackles the challenge of scale-specific noise by disentangling each client's low-rank updates and sharing only their directional components. To handle the need for flexible and effective integration, each client further learns personalized weights that achieve the data-aware integration of updates from other domains. Extensive experiments across diverse datasets validate the effectiveness of our proposed FeDecider.
Xinrui He, Ting-Wei Li, Tianxin Wei, Xuying Ning, Xinyu He 0003, Hanghang Tong, Jingrui He
WWW3
2025 SelfElicit: Your Language Model Secretly Knows Where is the Relevant Evidence
abstract
Providing Language Models (LMs) with relevant evidence in the context (either via retrieval or user-provided) can significantly improve their ability to provide better-grounded responses. However, recent studies have found that LMs often struggle to fully comprehend and utilize key evidence from the context, especially when it contains noise and irrelevant information—an issue common in real-world scenarios.To address this, we propose SelfElicit, an inference-time approach that helps LMs focus on key contextual evidence through self-guided explicit highlighting.By leveraging the inherent evidence-finding capabilities of LMs using the attention scores of deeper layers, our method automatically identifies and emphasizes key evidence within the input context, facilitating more accurate and grounded responses without additional training or iterative prompting.We demonstrate that SelfElicit brings consistent and significant improvement on multiple evidence-based QA tasks for various LM families while maintaining computational efficiency.Our code and documentation are available at https://github.com/ZhiningLiu1998/SelfElicit.
Zhining Liu 0002, Rana Ali Amjad, Ravinarayana Adkathimar, Tianxin Wei, Hanghang Tong
ACL (1)4
2025 PyG-SSL: A Graph Self-Supervised Learning Toolkit
abstract
Graph Self-Supervised Learning (SSL) has emerged as a pivotal area of research in recent years. By engaging in pretext tasks to learn the intricate topological structures and properties of graphs using unlabeled data, these graph SSL models achieve enhanced performance, improved generalization, and heightened robustness. Despite the remarkable achievements of these graph SSL methods, their current implementation poses significant challenges for beginners and practitioners due to the complex nature of graph structures, inconsistent evaluation metrics, and concerns regarding reproducibility hinder further progress in this field. Recognizing the growing interest within the research community, there is an urgent need for a comprehensive, beginner-friendly, and accessible toolkit consisting of the most representative graph SSL algorithms. To address these challenges, we present a Graph SSL toolkit named PyG-SSL, which is built upon PyTorch and is compatible with various deep learning and scientific computing backends. Within the toolkit, we offer a unified framework encompassing dataset loading, hyper-parameter configuration, model training, and comprehensive performance evaluation for diverse downstream tasks. Moreover, we provide beginner-friendly tutorials and the best hyper-parameters of each graph SSL algorithm on different graph datasets, facilitating the reproduction of results. The GitHub repository of the library is https://github.com/iDEA-iSAIL-Lab-UIUC/pyg-ssl.
Lecheng Zheng, Baoyu Jing, Zihao Li 0006, Zhichen Zeng 0001, Tianxin Wei, Mengting Ai, Xinrui He, Lihui Liu, Dongqi Fu, Jiaxuan You, Hanghang Tong, Jingrui He
CIKM5
2025 Latte: Collaborative Test-Time Adaptation of Vision-Language Models in Federated Learning
Ruxi Deng, Ruizhong Qiu, Tianxin Wei, Hanghang Tong, Jingrui He
ICCV4
2025 Breaking Silos: Adaptive Model Fusion Unlocks Better Time Series Forecasting
abstract
Time-series forecasting plays a critical role in many real-world applications. Although increasingly powerful models have been developed and achieved superior results on benchmark datasets, through a fine-grained sample-level inspection, we find that (i) no single model consistently outperforms others across different test samples, but instead (ii) each model excels in specific cases. These findings prompt us to explore how to adaptively leverage the distinct strengths of various forecasting models for different samples. We introduce TimeFuse, a framework for collective time-series forecasting with sample-level adaptive fusion of heterogeneous models. TimeFuse utilizes meta-features to characterize input time series and trains a learnable fusor to predict optimal model fusion weights for any given input. The fusor can leverage samples from diverse datasets for joint training, allowing it to adapt to a wide variety of temporal patterns and thus generalize to new inputs, even from unseen datasets. Extensive experiments demonstrate the effectiveness of TimeFuse in various long-/short-term forecasting tasks, achieving near-universal improvement over the state-of-the-art individual models. Code is available at https://github.com/ZhiningLiu1998/TimeFuse.
Zhining Liu 0002, Xiao Lin 0016, Ruizhong Qiu, Tianxin Wei, Yada Zhu, Hendrik F. Hamann, Jingrui He, Hanghang Tong
ICML5
2025 Graph4MM: Weaving Multimodal Learning with Structural Information
abstract
Real-world multimodal data usually exhibit complex structural relationships beyond traditional one-to-one mappings like image-caption pairs. Entities across modalities interact in intricate ways, with images and text forming diverse interconnections through contextual dependencies and co-references. Graphs provide powerful structural information for modeling intra-modal and inter-modal relationships. However, previous works fail to distinguish multi-hop neighbors and treat the graph as a standalone modality, which fragments the overall understanding. This limitation presents two key challenges in multimodal learning: (1) integrating structural information from multi-hop neighbors into foundational models, and (2) fusing modality-specific information in a principled manner. To address these challenges, we revisit the role of graphs in multimodal learning within the era of foundation models and propose Graph4MM, a graph-based multimodal learning framework. To be specific, we introduce Hop-Diffused Attention, which integrates multi-hop structural information into self-attention through causal masking and hop diffusion. Furthermore, we design MM-QFormer, a multi-mapping querying transformer for cross-modal fusion. Through theoretical and empirical analysis, we show that leveraging structures to integrate both intra- and inter-modal interactions improves multimodal understanding beyond treating them as a standalone modality. Experiments on both generative and discriminative tasks show that Graph4MM outperforms larger VLMs, LLMs, and multimodal graph baselines, achieving a 6.93% average improvement.
Xuying Ning, Dongqi Fu, Tianxin Wei, Wujiang Xu, Jingrui He
ICML3
2025 ResMoE: Space-efficient Compression of Mixture of Experts LLMs via Residual Restoration
abstract
Mixture-of-Experts (MoE) Transformer, the backbone architecture of multiple phenomenal language models, leverages sparsity by activating only a fraction of model parameters for each input token. The sparse structure, while allowing constant time costs, results in space inefficiency: we still need to load all the model parameters during inference. We introduce ResMoE, an innovative MoE approximation framework that utilizes Wasserstein barycenter to extract a common expert (barycenter expert) and approximate the residuals between this barycenter expert and the original ones. ResMoE enhances the space efficiency for inference of large-scale MoE Transformers in a one-shot and data-agnostic manner without retraining while maintaining minimal accuracy loss, thereby paving the way for broader accessibility to large language models. We demonstrate the effectiveness of ResMoE through extensive experiments on Switch Transformer, Mixtral, and DeepSeekMoE models. The results show that ResMoE can reduce the number of parameters in an expert by up to 75% while maintaining comparable performance. The code is available at https://github.com/iDEA-iSAIL-Lab-UIUC/ResMoE, and the supplementary appendix is available at https://famous-blue-raincoat.github.io/mengtingai/files/ResMoE_Appendix.pdf.
Mengting Ai, Tianxin Wei, Yifan Chen 0004, Zhichen Zeng 0001, Ritchie Zhao, Girish Varatkar, Bita Darvish Rouhani, Xianfeng Tang, Hanghang Tong, Jingrui He
KDD (1)2
2025 Connecting Domains and Contrasting Samples: A Ladder for Domain Generalization
abstract
Distribution shifts between training and testing samples frequently occur in practice and impede model generalization performance. This crucial challenge thereby motivates studies on domain generalization (DG), which aim to predict the label on unseen target domain data by solely using data from source domains. It is intuitive to conceive the class-separated representations learned in contrastive learning (CL) are able to improve DG, while the reality is quite the opposite: users observe directly applying CL deteriorates the performance. We analyze the phenomenon with the insights from CL theory and discover lack of intra-class connectivity in the DG setting causes the deficiency. We thus propose a new paradigm, domain-connecting contrastive learning (DCCL), to enhance the conceptual connectivity across domains and obtain generalizable representations for DG. On the data side, more aggressive data augmentation and cross-domain positive samples are introduced to improve intra-class connectivity. On the model side, to better embed the unseen test domains, we propose model anchoring to exploit the intra-class connectivity in pre-trained representations and complement the anchoring with generative transformation loss. Extensive experiments on five standard DG benchmarks are performed. The results verify that DCCL outperforms state-of-the-art baselines even without domain supervision. The detailed model implementation and the code are provided through https://github.com/weitianxin/DCCL.
Tianxin Wei, Yifan Chen 0004, Xinrui He, Jingrui He
KDD (1)1
2025 CLIMB: Class-imbalanced Learning Benchmark on Tabular Data
abstract
Class-imbalanced learning (CIL) on tabular data is important in many real-world applications where the minority class holds the critical but rare outcomes. In this paper, we present CLIMB, a comprehensive benchmark for class-imbalanced learning on tabular data. CLIMB includes 73 real-world datasets across diverse domains and imbalance levels, along with unified implementations of 29 representative CIL algorithms. Built on a high-quality open-source Python package with unified API designs, detailed documentation, and rigorous code quality controls, CLIMB supports easy implementation and comparison between different CIL algorithms. Through extensive experiments, we provide practical insights on method accuracy and efficiency, highlighting the limitations of naive rebalancing, the effectiveness of ensembles, and the importance of data quality. Our code, documentation, and examples are available at https://github.com/ZhiningLiu1998/imbalanced-ensemble.
Zhining Liu 0002, Zihao Li 0006, Tianxin Wei, Jian Kang 0008, Yada Zhu, Hendrik F. Hamann, Jingrui He, Hanghang Tong
NeurIPS4
2025 i2VAE: Interest Information Augmentation with Variational Regularizers for Cross-Domain Sequential Recommendation
abstract
Cross-Domain Sequential Recommendation (CDSR) leverages user behaviors across multiple domains to mitigate data sparsity and cold-start challenges in Single-Domain Sequential Recommendation. Existing methods primarily rely on shared users (overlapping users) to learn transferable interest representations. However, these approaches have limited information propagation, benefiting mainly overlapping users and those with rich interaction histories while neglecting non-overlapping (cold-start) and long-tailed users, who constitute the majority in real-world scenarios. To address this issue, we propose i$^2$VAE, a novel variational autoencoder (VAE)-based framework that enhances user interest learning with mutual information-based regularizers. i$^2$VAE improves recommendations for cold-start and long-tailed users while maintaining strong performance across all user groups. Specifically, cross-domain and disentangling regularizers extract transferable features for cold-start users, while a pseudo-sequence generator synthesizes interactions for long-tailed users, refined by a denoising regularizer to filter noise and preserve meaningful interest signals. Extensive experiments demonstrate that i$^2$VAE outperforms state-of-the-art methods, underscoring its effectiveness in real-world CDSR applications. Code and datasets are available at https://github.com/WujiangXu/IM-VAE.
Xuying Ning, Wujiang Xu, Tianxin Wei
UAI3
2024 TAU: Trajectory Data Augmentation with Uncertainty for Next POI Recommendation
abstract
Next Point-of-Interest (POI) recommendation has been proven effective at utilizing sparse, intricate spatial-temporal trajectory data to recommend subsequent POIs to users. While existing methods commonly alleviate the problem of data sparsity by integrating spatial-temporal context information, POI category features, and social relationships, they largely overlook the fact that the trajectory sequences collected in the datasets are often incomplete. This oversight limits the model’s potential to fully leverage historical context. In light of this background, we propose Trajectory Data Augmentation with Uncertainty (TAU) for Next POI Recommendation. TAU is a general graph-based trajectory data augmentation method designed to complete user mobility patterns by marrying uncertainty estimation into the next POI recommendation task. More precisely, TAU taps into the global transition pattern graph to identify sets of intermediate nodes located between every pair of locations, effectively leveraging edge weights as transition probabilities. During trajectory sequence construction, TAU selectively prompts intermediate nodes, chosen based on their likelihood of occurrence as pseudo-labels, to establish comprehensive trajectory sequences. Furthermore, to gauge the certainty and impact of pseudo-labels on the target location, we introduce a novel confidence-aware calibration strategy using evidence deep learning (EDL) for improved performance and reliability. The experimental results clearly indicate that our TAU method achieves consistent performance improvements over existing techniques across two real-world datasets, verifying its effectiveness as the state-of-the-art approach to the task.
Zhuang Zhuang, Tianxin Wei, Lingbo Liu, Heng Qi, Yanming Shen
AAAI2
2024 Towards Unified Multi-Modal Personalization: Large Vision-Language Models for Generative Recommendation and Beyond
abstract
Developing a universal model that can effectively harness heterogeneous resources and respond to a wide range of personalized needs has been a longstanding community aspiration. Our daily choices, especially in domains like fashion and retail, are substantially shaped by multi-modal data, such as pictures and textual descriptions. These modalities not only offer intuitive guidance but also cater to personalized user preferences. However, the predominant personalization approaches mainly focus on ID or text-based recommendation problems, failing to comprehend the information spanning various tasks or modalities. In this paper, our goal is to establish a Unified paradigm for Multi-modal Personalization systems (UniMP), which effectively leverages multi-modal data while eliminating the complexities associated with task- and modality-specific customization. We argue that the advancements in foundational generative modeling have provided the flexibility and effectiveness necessary to achieve the objective. In light of this, we develop a generic and extensible personalization generative framework, that can handle a wide range of personalized needs including item recommendation, product search, preference prediction, explanation generation, and further user-guided image generation. Our methodology enhances the capabilities of foundational language models for personalized tasks by seamlessly ingesting interleaved cross-modal user history information, ensuring a more precise and customized experience for users. To train and evaluate the proposed multi-modal personalized tasks, we also introduce a novel and comprehensive benchmark covering a variety of user requirements. Our experiments on the real-world benchmark showcase the model's potential, outperforming competitive methods specialized for each task.
Tianxin Wei, Bowen Jin, Ruirui Li 0002, Hansi Zeng, Jianhui Sun, Qingyu Yin, Hanqing Lu, Suhang Wang, Jingrui He, Xianfeng Tang
ICLR1
2024 Language Models as Semantic Indexers
abstract
Semantic identifier (ID) is an important concept in information retrieval that aims to preserve the semantics of objects such as documents and items inside their IDs. Previous studies typically adopt a two-stage pipeline to learn semantic IDs by first procuring embeddings using off-the-shelf text encoders and then deriving IDs based on the embeddings. However, each step introduces potential information loss, and there is usually an inherent mismatch between the distribution of embeddings within the latent space produced by text encoders and the anticipated distribution required for semantic indexing. It is non-trivial to design a method that can learn the document’s semantic representations and its hierarchical structure simultaneously, given that semantic IDs are discrete and sequentially structured, and the semantic supervision is deficient. In this paper, we introduce LMIndexer, a self-supervised framework to learn semantic IDs with a generative language model. We tackle the challenge of sequential discrete ID by introducing a semantic indexer capable of generating neural sequential discrete representations with progressive training and contrastive learning. In response to the semantic supervision deficiency, we propose to train the model with a self-supervised document reconstruction objective. We show the high quality of the learned IDs and demonstrate their effectiveness on three tasks including recommendation, product search, and document retrieval on five datasets from various domains. Code is available at https://github.com/PeterGriffinJin/LMIndexer.
Bowen Jin, Hansi Zeng, Guoyin Wang 0001, Xiusi Chen, Tianxin Wei, Ruirui Li 0002, Zheng Li 0018, Hanqing Lu, Suhang Wang, Jiawei Han 0001, Xianfeng Tang
ICML5
2024 Graph Mixup on Approximate Gromov-Wasserstein Geodesics
abstract
Mixup, which generates synthetic training samples on the data manifold, has been shown to be highly effective in augmenting Euclidean data. However, finding a proper data manifold for graph data is non-trivial, as graphs are non-Euclidean data in disparate spaces. Though efforts have been made, most of the existing graph mixup methods neglect the intrinsic geodesic guarantee, thereby generating inconsistent sample-label pairs. To address this issue, we propose GeoMix to mixup graphs on the Gromov-Wasserstein (GW) geodesics. A joint space over input graphs is first defined based on the GW distance, and graphs are then transformed into the GW space through equivalence-preserving transformations. We further show that the linear interpolation of the transformed graph pairs defines a geodesic connecting the original pairs on the GW manifold, hence ensuring the consistency between generated samples and labels. An accelerated mixup algorithm on the approximate low-dimensional GW manifold is further proposed. Extensive experiments show that the proposed GeoMix promotes the generalization and robustness of GNN models.
Zhichen Zeng 0001, Ruizhong Qiu, Zhe Xu 0007, Zhining Liu 0002, Tianxin Wei, Lei Ying 0001, Jingrui He, Hanghang Tong
ICML6
2024 Meta Clustering of Neural Bandits
abstract
The contextual bandit has been identified as a powerful framework to formulate the recommendation process as a sequential decision-making process, where each item is regarded as an arm and the objective is to minimize the regret of T rounds. In this paper, we study a new problem, Clustering of Neural Bandits, by extending previous work to the arbitrary reward function, to strike a balance between user heterogeneity and user correlations in the recommender system. To solve this problem, we propose a novel algorithm called M-CNB, which utilizes a meta-learner to represent and rapidly adapt to dynamic clusters, along with an informative Upper Confidence Bound (UCB)-based exploration strategy. We provide an instance-dependent performance guarantee for the proposed algorithm that withstands the adversarial context, and we further prove the guarantee is at least as good as state-of-the-art (SOTA) approaches under the same assumptions. In extensive experiments conducted in both recommendation and online classification scenarios, M-CNB outperforms SOTA baselines. This shows the effectiveness of the proposed approach in improving online recommendation and online classification performance.
Yikun Ban, Yunzhe Qi, Tianxin Wei, Lihui Liu, Jingrui He
KDD3
2024 Scalable and Effective Generative Information Retrieval
abstract
Recent research has shown that transformer networks can be used as differentiable search indexes by representing each document as a sequence of document ID tokens. These generative retrieval models cast the retrieval problem to a document ID generation problem for each query. Despite their elegant design, existing generative retrieval models only perform well on artificially-constructed and small-scale collections. This paper represents an important milestone in generative retrieval research by showing that generative retrieval models can be trained to perform effectively on large-scale standard retrieval benchmarks. In more detail, we propose RIPOR- an optimization framework for generative retrieval that is designed based on two often-overlooked fundamental design considerations. First, RIPOR introduces a novel prefix-oriented ranking optimization algorithm for accurate estimation of relevance score during sequential document ID generation. Second, RIPOR constructs document IDs based on the relevance associations between queries and documents. Evaluation on MSMARCO and TREC Deep Learning Track reveals that RIPOR surpasses state-of-the-art generative retrieval models by a large margin (e.g., 30.5% MRR improvements on MS MARCO Dev Set).
Hansi Zeng, Chen Luo 0003, Bowen Jin, Sheikh Muhammad Sarwar, Tianxin Wei, Hamed Zamani
WWW5
2023 Robust Basket Recommendation via Noise-tolerated Graph Contrastive Learning
abstract
The growth of e-commerce has seen a surge in popularity of platforms like Amazon, eBay, and Taobao. This has given rise to a unique shopping behavior involving baskets - sets of items purchased together. As a less studied interaction mode in the community, the question of how should shopping basket complement personalized recommendation systems remains under-explored. While previous attempts focused on jointly modeling user purchases and baskets, the distinct semantic nature of these elements can introduce noise when directly integrated. This noise negatively impacts the model's performance, further exacerbated by significant noise (e.g., a user is misled to click an item or recognizes it as uninteresting after consuming it) within both user and basket behaviors. In order to cope with the above difficulties, we propose a novel Basket recommendation framework via Noise-tolerated Contrastive Learning, named BNCL, to handle the noise existing in the cross-behavior integration and within-behavior modeling. First, we represent the basket-item interactions as the hypergraph to model the complex basket behavior, where all items appearing in the same basket are treated as a single hyperedge. Second, cross-behavior contrastive learning is designed to suppress the noise during the fusion of diverse behaviors. Next, to further inhibit the within-behavior noise of the user and basket interactions, we propose to exploit invariant properties of the recommenders w.r.t augmentations through within-behavior contrastive learning. A novel consistency-aware augmentation approach is further designed to better identify the noisy interactions with the consideration of the above two types of interactions. Our framework BNCL offers a generic training paradigm that is applicable to different backbones. Extensive experiments on three shopping transaction datasets verify the effectiveness of our proposed method.
Xinrui He, Tianxin Wei, Jingrui He
CIKM2
2023 NTK-approximating MLP Fusion for Efficient Language Model Fine-tuning
abstract
Fine-tuning a pre-trained language model (PLM) emerges as the predominant strategy in many natural language processing applications. However, even fine-tuning the PLMs and doing inference are expensive, especially on edge devices with low computing power. Some general approaches (e.g. quantization and distillation) have been widely studied to reduce the compute/memory of PLM fine-tuning, while very few one-shot compression techniques are explored. In this paper, we investigate the neural tangent kernel (NTK)–which reveals the gradient descent dynamics of neural networks–of the multilayer perceptrons (MLP) modules in a PLM and propose to coin a lightweight PLM through NTK-approximating MLP fusion. To achieve this, we reconsider the MLP as a bundle of sub-MLPs, and cluster them into a given number of centroids, which can then be restored as a compressed MLP and surprisingly shown to well approximate the NTK of the original PLM. Extensive experiments of PLM fine-tuning on both natural language understanding (NLU) and generation (NLG) tasks are provided to verify the effectiveness of the proposed method MLP fusion. Our code is available at https://github.com/weitianxin/MLP_Fusion.
Tianxin Wei, Zeming Guo, Yifan Chen 0004, Jingrui He
ICML1
2023 Adaptive Test-Time Personalization for Federated Learning
abstract
Personalized federated learning algorithms have shown promising results in adapting models to various distribution shifts. However, most of these methods require labeled data on testing clients for personalization, which is usually unavailable in real-world scenarios. In this paper, we introduce a novel setting called test-time personalized federated learning (TTPFL), where clients locally adapt a global model in an unsupervised way without relying on any labeled data during test-time. While traditional test-time adaptation (TTA) can be used in this scenario, most of them inherently assume training data come from a single domain, while they come from multiple clients (source domains) with different distributions. Overlooking these domain interrelationships can result in suboptimal generalization. Moreover, most TTA algorithms are designed for a specific kind of distribution shift and lack the flexibility to handle multiple kinds of distribution shifts in FL. In this paper, we find that this lack of flexibility partially results from their pre-defining which modules to adapt in the model. To tackle this challenge, we propose a novel algorithm called ATP to adaptively learns the adaptation rates for each module in the model from distribution shifts among source domains. Theoretical analysis proves the strong generalization of ATP. Extensive experiments demonstrate its superiority in handling various distribution shifts including label shift, image corruptions, and domain shift, outperforming existing TTA methods across multiple datasets and model architectures. Our code is available at https://github.com/baowenxuan/ATP.
Tianxin Wei, Haohan Wang, Jingrui He
NeurIPS2
2023 Meta-Learning with Neural Bandit Scheduler
abstract
Meta-learning has been proven an effective learning paradigm for training machine learning models with good generalization ability. Apart from the common practice of uniformly sampling the meta-training tasks, existing methods working on task scheduling strategies are mainly based on pre-defined sampling protocols or the assumed task-model correlations, and greedily make scheduling decisions, which can lead to sub-optimal performance bottlenecks of the meta-model. In this paper, we propose a novel task scheduling framework under Contextual Bandits settings, named BASS, which directly optimizes the task scheduling strategy based on the status of the meta-model. By balancing the exploitation and exploration in meta-learning task scheduling, BASS can help tackle the challenge of limited knowledge about the task distribution during the early stage of meta-training, while simultaneously exploring potential benefits for forthcoming meta-training iterations through an adaptive exploration strategy. Theoretical analysis and extensive experiments are presented to show the effectiveness of our proposed framework.
Yunzhe Qi, Yikun Ban, Tianxin Wei, Jiaru Zou, Huaxiu Yao, Jingrui He
NeurIPS3
2022 Comprehensive Fair Meta-learned Recommender System
abstract
In recommender systems, one common challenge is the cold-start problem, where interactions are very limited for fresh users in the systems. To address this challenge, recently, many works introduce the meta-optimization idea into the recommendation scenarios, i.e. learning to learn the user preference by only a few past interaction items. The core idea is to learn global shared meta-initialization parameters for all users and rapidly adapt them into local parameters for each user respectively. They aim at deriving general knowledge across preference learning of various users, so as to rapidly adapt to the future new user with the learned prior and a small amount of training data. However, previous works have shown that recommender systems are generally vulnerable to bias and unfairness. Despite the success of meta-learning at improving the recommendation performance with cold-start, the fairness issues are largely overlooked.
Tianxin Wei, Jingrui He
KDD1
2022 Augmentations in Hypergraph Contrastive Learning: Fabricated and Generative
abstract
This paper targets at improving the generalizability of hypergraph neural networks in the low-label regime, through applying the contrastive learning approach from images/graphs (we refer to it as HyperGCL). We focus on the following question: How to construct contrastive views for hypergraphs via augmentations? We provide the solutions in two folds. First, guided by domain knowledge, we fabricate two schemes to augment hyperedges with higher-order relations encoded, and adopt three vertex augmentation strategies from graph-structured data. Second, in search of more effective views in a data-driven manner, we for the first time propose a hypergraph generative model to generate augmented views, and then an end-to-end differentiable pipeline to jointly learn hypergraph augmentations and model parameters. Our technical innovations are reflected in designing both fabricated and generative augmentations of hypergraphs. The experimental findings include: (i) Among fabricated augmentations in HyperGCL, augmenting hyperedges provides the most numerical gains, implying that higher-order information in structures is usually more downstream-relevant; (ii) Generative augmentations do better in preserving higher-order information to further benefit generalizability; (iii) HyperGCL also boosts robustness and fairness in hypergraph representation learning. Codes are released at https://github.com/weitianxin/HyperGCL.
Tianxin Wei, Yuning You, Tianlong Chen 0001, Yang Shen 0001, Jingrui He, Zhangyang Wang
NeurIPS1
2021 Unpaired Multimodal Neural Machine Translation via Reinforcement Learning
Yijun Wang 0002, Tianxin Wei, Qi Liu 0003, Enhong Chen
DASFAA (2)2
2021 Model-Agnostic Counterfactual Reasoning for Eliminating Popularity Bias in Recommender System
abstract
The general aim of the recommender system is to provide personalized suggestions to users, which is opposed to suggesting popular items. However, the normal training paradigm, i.e., fitting a recommender model to recover the user behavior data with pointwise or pairwise loss, makes the model biased towards popular items. This results in the terrible Matthew effect, making popular items be more frequently recommended and become even more popular. Existing work addresses this issue with Inverse Propensity Weighting (IPW), which decreases the impact of popular items on the training and increases the impact of long-tail items. Although theoretically sound, IPW methods are highly sensitive to the weighting strategy, which is notoriously difficult to tune.
Tianxin Wei, Fuli Feng, Jiawei Chen 0007, Jinfeng Yi, Xiangnan He 0001
KDD1
2021 Causal Intervention for Leveraging Popularity Bias in Recommendation
abstract
Recommender system usually faces popularity bias issues: from the data perspective, items exhibit uneven (usually long-tail) distribution on the interaction frequency; from the method perspective, collaborative filtering methods are prone to amplify the bias by over-recommending popular items. It is undoubtedly critical to consider popularity bias in recommender systems, and existing work mainly eliminates the bias effect with propensity-based unbiased learning or causal embeddings. However, we argue that not all biases in the data are bad, \ie some items demonstrate higher popularity because of their better intrinsic quality. Blindly pursuing unbiased learning may remove the beneficial patterns in the data, degrading the recommendation accuracy and user satisfaction. This work studies an unexplored problem in recommendation --- how to leverage popularity bias to improve the recommendation accuracy. The key lies in two aspects: how to remove the bad impact of popularity bias during training, and how to inject the desired popularity bias in the inference stage that generates top-K recommendations. This questions the causal mechanism of the recommendation generation process. Along this line, we find that item popularity plays the role ofconfounder between the exposed items and the observed interactions, causing the bad effect of bias amplification. To achieve our goal, we propose a new training and inference paradigm for recommendation named Popularity-bias Deconfounding and Adjusting (PDA). It removes the confounding popularity bias in model training and adjusts the recommendation score with desired popularity bias via causal intervention. We demonstrate the new paradigm on the latent factor model and perform extensive experiments on three real-world datasets from Kwai, Douban, and Tencent. Empirical studies validate that the deconfounded training is helpful to discover user real interests and the inference adjustment with popularity bias could further improve the recommendation accuracy. We release our code at https://github.com/zyang1580/PDA.
Yang Zhang 0072, Fuli Feng, Xiangnan He 0001, Tianxin Wei, Chonggang Song, Guohui Ling, Yongdong Zhang 0001
SIGIR4
2020 Fast Adaptation for Cold-start Collaborative Filtering with Meta-learning
abstract
Collaborative Filtering (CF), as one of the most popular approaches, is widely employed in recommender systems but suffers from the cold-start problem, where interactions are very limited for new users in the system. To deal with this issue, previous work has largely focused on utilizing various auxiliary information such as user profiles and social relationships to infer user preferences. However, the auxiliary information is not always available due to reasons such as user privacy concerns, making the CF approaches have to count on the limited interactions. Moreover, real-world situations require both accurate and quick recommendations for newly arrived users dynamically. Therefore, it is of critical importance to enable fast learning for new users during the training time of CF models. In this paper, we present a novel learning paradigm, named MetaCF, to learn an accurate CF model that makes fast adaptation on new users with limited interactions. Inspired by meta-learning, MetaCF treats the fast adaptation on a new user as a task and aims to learn a suitable model for initializing the adaption. To pursue a well-generalized model, MetaCF is equipped with a Dynamic Subgraph Sampling that accounts for the dynamic arrival of new users by dynamically generating representative adaptation tasks for existing users. Moreover, to stabilize the adaption procedure that faces the shortage of training samples, MetaCF further optimizes the learning rates for adaption in a fine-grained manner. MetaCF is applicable to any differentiable CF-based models where we demonstrate it on two representative ones, FISM [1] and NGCF [2]. Extensive experiments on three datasets validate the effectiveness of the proposed framework, which significantly outperforms state-of-the-art baselines by a large margin in the cold-start scenario where user-item interactions are limited.
Tianxin Wei, Ruirui Li 0002, Ziniu Hu, Fuli Feng, Xiangnan He 0001, Yizhou Sun, Wei Wang 0010
ICDM1