Preprint(^*Equal Contribution, ^#Corresponding)

Jiaru Zou, Yikun Ban^#, Zihao Li, Yunzhe Qi, Ruizhong Qiu, Ling Yang, Jingrui He^#
Transformer Copilot: Learning from The Mistake Log in LLM Fine-tuning

Zixuan Huang, Yikun Ban^#, Lean Fu, Xiaojie Li, Zhongxiang Dai, Jianxin Li, Deqing Wang^#
Adaptive Sample Scheduling for Direct Preference Optimization

Publications(^*Equal Contribution, ^#Corresponding)

Xiaodong Lu, Mingzhe Liu, Tongyu Zhu, Leilei Sun, Jibin Wang, Weifeng Lv, Yikun Ban, Deqing Wang
Adaptive Sampling-based Dynamic Graph Learning for Information Diffusion Prediction
ACM Transactions on Information Systems (2025) ( TOIS )

Xinrui He, Yikun Ban^#, Jiaru Zou, Tianxin Wei, Curtiss Cook, Jingrui He^#
LLM-Forest: Ensemble Learning of LLMs with Graph-Augmented Prompts for Data Imputation
The 63rd Annual Meeting of the Association for Computational Linguistics, Findings ( ACL'25 )

Zihao Li, Lecheng Zheng, Bowen Jin, Dongqi Fu, Baoyu Jing, Yikun Ban, Jingrui He, Jiawei Han
Can Graph Neural Networks Learn Language with Extremely Weak Text Supervision?
The 63rd Annual Meeting of the Association for Computational Linguistics, Main( ACL'25 )

Yunzhe Qi, Yikun Ban, Arindam Banerjee, Jingrui He
Robust Neural Contextual Bandit against Adversarial Corruptions
Thirty-eighth Conference on Neural Information Processing Systems ( NeurIPS'24 )

Yikun Ban^*, Jiaru Zou^*, Zihao Li, Yunzhe Qi, Dongqi Fu, Jian Kang, Hanghang Tong, Jingrui He
PageRank Bandits for Link Prediction
Thirty-eighth Conference on Neural Information Processing Systems ( NeurIPS'24 )

Yikun Ban^*, Yunzhe Qi^*, Tianxin Wei, Lihui Liu, Jingrui He
Meta Clustering of Neural Bandits
ACM SIGKDD Conference on Knowledge Discovery and Data Mining ( KDD'24 )

Yikun Ban, Yuchen Yan, Arindam Banerjee, Jingrui He
Neural Exploitation and Exploration of Contextual Bandits

To Appear
Yikun Ban, Yunzhe Qi, Jingrui He
Neural Contextual Bandits for Personalized Recommendation
The Web Conference, Tutorial ( WWW'24)
[ ]   []
Yikun Ban, Ishika Agarwal, Ziwei Wu, Yada Zhu, Kommy Weldemariam, Hanghang Tong, and Jingrui He
Neural Active Learning Beyond Bandits
International Conference on Learning Representations ( ICLR'24 )

Rohan Deb, Yikun Ban, Shiliang Zuo, Jingrui He, Arindam Banerjee
Contextual Bandits with Online Neural Regression
International Conference on Learning Representations ( ICLR'24 )

Yunzhe Qi^*, Yikun Ban^*, Tianxin Wei, Jiaru Zou, Huaxiu Yao, and Jingrui He
Meta-Learning with Neural Bandit Scheduler
Thirty-seventh Conference on Neural Information Processing Systems ( NeurIPS'23 )

Yunzhe Qi^*, Yikun Ban^*, and Jingrui He
Graph Neural Bandits
ACM SIGKDD Conference on Knowledge Discovery and Data Mining ( KDD'23 )

Yikun Ban^*, Yuheng Zhang^*, Hanghang Tong, Arindam Banerjee, and Jingrui He
Improved Algorithms for Neural Active Learning
Thirty-sixth Conference on Neural Information Processing Systems ( NeurIPS'22 )

Dongqi Fu, Yikun Ban, Hanghang Tong, Ross Maciejewski, and Jingrui He
DISCO: Comprehensive and Explainable Disinformation Detection
ACM International Conference on Information and Knowledge Management (Demo Track) ( CIKM'22 )

Yunzhe Qi, Yikun Ban, and Jingrui He
Neural Bandit with Arm Group Graph
ACM SIGKDD Conference on Knowledge Discovery and Data Mining ( KDD'22 )

Yikun Ban, Yuchen Yan, Arindam Banerjee, and Jingrui He
EE-Net: Exploitation-Exploration Neural Networks in Contextual Bandits
International Conference on Learning Representations ( ICLR'22, Spotlight)

Yikun Ban and Jingrui He
Convolutional Neural Bandit for Visual-aware Recommendation
Preprint: ArXiv:2107.07438

Yikun Ban, Jingrui He, and Curtiss B. Cook
Multi-Facet Contextual Bandits: A Neural Network Perspective
ACM SIGKDD Conference on Knowledge Discovery and Data Mining ( KDD'21 )

Yikun Ban and Jingrui He
Local Clustering in Contextual Multi-Armed Bandits
The Web Conference ( WWW'21 )

Yuchen Yan, Lihui Liu, Yikun Ban, Baoyu Jing, and Hanghang Tong
Dynamic Knowledge Graph Alignment
AAAI Conference on Artificial Intelligence ( AAAI'21 )

Yikun Ban and Jingrui He
Generic Outlier Detection in Multi-Armed Bandit
ACM SIGKDD Conference on Knowledge Discovery and Data Mining ( KDD'20 )

Yikun Ban, Xin liu, Ling Huang, Yitao Duan, Xue Liu, and Wei Xu
No Place to Hide: Catching Fraudulent Entities in Tensors
The Web Conference ( WWW ’19 )

Education

Internship

Services