Tianxin Wei
I am a third-year Ph.D. student advised by Prof. Jingrui He at the University of Illinois Urbana-Champaign.
Prior to joining UIUC in 2021, I earned my B.S. degree in Computer Science from University of Science and Technology of China, School of the Gifted Young.
My research primarily centers on enhancing the transferability and safety of machine learning algorithms across various modalities and disciplines, with the ultimate goal of making ML models more accessible and inclusive. My research interests covers a wide range of topics:
- Trustworthy: bias, fairness, robustness, and transferability;
- Efficiency: sampling and model (Vanilla and MoE Transformer) efficiency;
- Multi-modality: modal interaction and fusion;
- Knowledge-enhanced LLM: content/KG retrieval and knowledge fusion;
- LLM Governance/Policy: technical strategies for implementing regulatory principles effectively;
- Applications: agriculture, recsys, and looking forward to more science applications.
Feel free to drop me an e-mail, if you are interested in my research and want to discuss relevant research topic or potential collaborations!
CV  / 
Email  / 
Google Scholar  / 
Github  / 
Twitter  / 
Linkedin  
|
|
Education
Ph.D.          2021 - 2026 (expected)
                       University of Illinois Urbana-Champaign (UIUC)
                       Advisor: Prof. Jingrui He
|
|
B.S.              2016 - 2020
                       University of Science and Technology (USTC)
                       School of the Gifted Young
                       Advisor: Prof. Xiangnan He
|
|
|
Work Experience
Amazon Search, Palo Alto, CA
Research Intern • Aug.-Dec. 2023
Multi-modal Large Langauge Models for Personalization
Two papers published at ICLR'24 and, WWW'24 (Oral). Two in submission.
Main Advisors: Dr. Xianfeng Tang at Amazon; Prof. Suhang Wang at PSU
|
|
|
News
Nov, 2024 |
Two (co-)first-author papers accepted @ KDD’25.
|
May, 2024 |
One papers accepted @ KDD’24.
|
May, 2024 |
Two papers accepted @ ICML’24.
|
Jan, 2024 |
One paper RIPOR (Scalable and effective generative retrieval) accepted @ WWW’24.
|
Jan, 2024 |
One paper UniMP (multi-modal personalization including recommendation and search, etc.) accepted @ ICLR’24.
|
Dec, 2023 |
One paper accepted @ AAAI’24. Will attend NeurIPS'23 between Dec. 9-16. See you there!
|
Oct, 2023 |
Awarded the NeurIPS’23 Scholar Award. Thanks to NeurIPS!
|
Sep, 2023 |
Two papers (Test-time personalization and bandit scheduler for meta-learning) accepted @ NeurIPS’23.
|
Aug, 2023 |
One paper BNCL accepted @ CIKM’23.
|
May, 2023 |
Received the ICML’23 Grant Award. Thanks to ICML!
|
Apr, 2023 |
MLP Fusion (NTK-approximating MLP Fusion for efficient learning) accepted @ ICML’23.
|
Mar, 2023 |
Will join Amazon Search team as an applied scientist intern this summer. See you in Palo Alto!
|
Dec, 2022 |
Collect a curated list of papers on the distribution shift. Check out at awesome-distribution-shift!
|
Oct, 2022 |
Awarded the NeurIPS’22 Scholar Award. Thanks to NeurIPS!
|
Oct, 2022 |
HyperGCL (contrastive learning on hypergraphs) accepted @ NeurIPS’22.
|
May, 2022 |
CLOVER (comprehensive fairness of cold-start recsys) accepted @ KDD’22.
|
Jul, 2021 |
Awarded the SIGIR 2021 Best Paper Honorable Mention. Thanks to SIGIR!
|
May, 2021 |
MACR (popularity debias with causal counterfactual reasoning) accepted @ KDD’21.
|
Apr, 2021 |
PDA (popularity adjusted deconfounded training with causal intervention) accepted @ SIGIR’21.
|
Aug, 2020 |
MetaCF (graph meta-learning for cold-start recsys) accepted @ ICDM’20. Internship work at UCLA.
|
|
Publications (*equal contribution ^mentorship)
       Preprint
|
-
Robust Watermarking for Diffusion Models
preprint 2024. Code
-
Automatic Optimizer for Black-box LLMs
preprint 2024. Code
-
WAPITI: A Watermark for Finetuned Open-Source LLMs
Lingjie Chen, Ruizhong Qiu, Siyu Yuan, Zhining Liu, Tianxin Wei, Hyunsik Yoo, Zhichen Zeng, Deqing Yang, Hanghang Tong
preprint 2024. Code
-
Unleashing the Power of LLMs as Multi-Modal Encoders for Text and Graph-Structured Data
Jiacheng Lin, Kun Qian, Haoyu Han, Nurendra Choudhary, Tianxin Wei, Zhongruo Wang, Sahika Genc, Edward W Huang, Sheng Wang, Karthik Subbian, Danai Koutra, Jimeng Sun
preprint 2024. Code
              Conference
-
Multi-View Domain Generalization: Connecting Domains by Integrating Data and Model Views
Tianxin Wei*, Yifan Chen*, Xinrui He, Wenxuan Bao, Jingrui He
KDD 2025. (Full Research, AR: 19%). Code
-
ResMoE: Residual Approximation for Compressing Mixture-of-Experts in Large Language Models with Space Efficiency
Mengting Ai*, Tianxin Wei*^, Yifan Chen*, Zhichen Zeng, Ritchie Zhao, Girish Varatkar, Bita Darvish Rouhani, Xianfeng Tang, Hanghang Tong, Jingrui He
KDD 2025. (Full Research, AR: 19%). Code
-
Meta Clustering of Neural Bandits
Yikun Ban, Yunzhe Qi, Tianxin Wei, Lihui Liu, Jingrui He
KDD 2024. (Full Research, AR: 20%). Code
-
Graph Mixup on Approximate Gromov–Wasserstein Geodesics
Zhichen Zeng, Ruizhong Qiu, Zhe Xu, Zhining Liu, Yuchen Yan, Tianxin Wei, Lei Ying, Jingrui He, Hanghang Tong
ICML 2024 (Full Research, AR: 27.5%). Code
-
Language Models as Semantic Indexers
Bowen Jin, Hansi Zeng, Guoyin Wang, Xiusi Chen, Tianxin Wei, Ruirui Li, Zhengyang Wang, Zheng Li, Yang Li, Hanqing Lu, Suhang Wang, Jiawei Han, Xianfeng Tang
ICML 2024 (Full Research, AR: 27.5%). Code
- Towards Unified Multi-Modal Personalization: Large Vision-Language Models for Generative Recommendation and Beyond
Tianxin Wei, Bowen Jin, Ruirui Li, Hansi Zeng, Zhengyang Wang, Jianhui Sun, Qingyu Yin, Hanqing Lu, Suhang Wang, Jingrui He, Xianfeng Tang
ICLR 2024 (Full Research, AR: 31%). Code
- Scalable and Effective Generative Information Retrieval
Hansi Zeng, Chen Luo, Bowen Jin, Sheikh Muhammad Sarwar, Tianxin Wei, Hamed Zamani
WWW 2024 (Oral, Full Research, AR: 20.2%). Code
- TAU: Trajectory Data Augmentation with Uncertainty for Next POI Recommendation
Zhuang Zhuang, Tianxin Wei^, Lingbo Liu, Heng Qi, Yanming Shen, Baocai Yin
AAAI 2024 (Full Research, AR: 24%). Code
- Meta-Learning with Neural Bandit Scheduler
Yunzhe Qi, Yikun Ban, Tianxin Wei, Jiaru Zou, Huaxiu Yao, Jingrui He
NeurIPS 2023 (Full Research, AR: 26.1%). Code
- Adaptive Test-Time Personalization for Federated Learning
Wenxuan Bao*, Tianxin Wei*, Haohan Wang, Jingrui He
NeurIPS 2023 (Full Research, AR: 26.1%). Code
- Robust Basket Recommendation via Noise-tolerated Graph Contrastive Learning
Xinrui He*, Tianxin Wei*^, Jingrui He
CIKM 2023 (Full Research, AR: 24.0%). Code
- NTK-approximating MLP Fusion for Efficient Language Model Fine-tuning
Tianxin Wei*^, Zeming Guo*, Yifan Chen*, Jingrui He
ICML 2023 (Full Research, AR: 27.9%). Code
- Augmentations in Hypergraph Contrastive Learning: Fabricated and Generative
Tianxin Wei*, Yuning You*, Tianlong Chen, Yang Shen, Jingrui He, Zhangyang Wang
NeurIPS 2022 (Full Research, AR: 25.6%). Code Appendix
- Comprehensive Fair Meta-learned Recommender System
Tianxin Wei, Jingrui He
KDD 2022 (Full Research, AR: 15.0%). Code
- Model-Agnostic Counterfactual Reasoning for Eliminating Popularity Bias in Recommender System
Tianxin Wei, Fuli Feng, Jiawei Chen, Ziwei Wu, Jinfeng Yi, Xiangnan He
KDD 2021 (Full Research, AR: 15.4%). Code
- Causal Intervention for Leveraging Popularity Bias in Recommendation
Yang Zhang, Fuli Feng, Xiangnan He, Tianxin Wei, Chonggang Song, Guohui Ling and Yongdong Zhang
SIGIR 2021 (Best Paper Honorable Mention, 1 out of All, Full Research, AR: 21%). Code
- Fast Adaptation for Cold-start Collaborative Filtering with Meta-learning
Tianxin Wei, Ziwei Wu, Ruirui Li, Ziniu Hu, Fuli Feng, Xiangnan He, Yizhou Sun, and Wei Wang
ICDM 2020 (Full Research, AR: 9.8%). Code
- Unpaired Multimodal Neural Machine Translation via Reinforcement Learning
Yijun Wang*, Tianxin Wei*, Qi Liu, Enhong Chen
DASFAA 2021 (Full Research, AR: 20%).
- AR-Stock: Deep Augmented Relational Stock Prediction
Tianxin Wei, Yuning You, Tianlong Chen
AAAI 2021 Workshop on Knowledge Discovery from Unstructured Data (Oral). Code
              Journal
-
Graph Contrastive Learning: An Odyssey towards Generalizable, Scalable and Principled Representation Learning on Graphs
Yan Han, Yuning You, Wenqing Zheng, Scott Hoang, Tianxin Wei, Majdi Hassan, Tianlong Chen, Ying Ding, Yang Shen, Zhangyang Wang
IEEE Data Engineering Bulletin.
Services & Awards
University Nomination (Top3) for Apple Scholar in AI/ML 2024
Amazon Internship Fellowship in 2024
Conference Presentation Award, UIUC 2023
SIGIR 2021 Best Paper Honorable Mention
NeurIPS 2023 Scholar Award
NeurIPS 2022 Scholar Award
ICML 2023 Grant Award
Program Committee/Reviewer: ICML (2022-2025), NeurIPS (2022-2024), ICLR (2023-2025), KDD (2023-2025), AAAI (2023-2024), CIKM (2021-2023), WSDM 2023, ACL 2023, EMNLP 2023, LOG 2022
Journal Reviewer: TOIS, TKDE, DMKD, Machine Learning, TMLR
|
Pets
|
                                        
|
|