Publications
2026
Belief Context Graph (BCG)
Open-source belief-aware context infrastructure for long-horizon agents. Paper forthcoming.
NeurIPS Workshop on Resource-Aware Agentic AI 2026
CircuitLens: Reasoning Circuits as Data Selection Signals for Reinforcement Learning with Verifiable Rewards
Zhuofan Chen, Ziqian Jiao, Yikai Cui, Zhixin Cai, Jun Bai, Wenge Rong
EMNLP Findings 2026
Xetrieval: Mechanistically Explaining Dense Retrieval
Zhixin Cai*, Jun Bai*, Yang Liu*, Jiaqi Li, Yichi Zhang, Taichuan Li, Zhuofan Chen, Zixia Jia, Zilong Zheng, Wenge Rong
Preprint 2026
* equal contribution
OneMillion-Bench: How Far are Language Agents from Human Experts?
Qianyu Yang*, Yang Liu*, Jiaqi Li*, Jun Bai*, Hao Chen, Kaiyuan Chen, Tiliang Duan, Jiayun Dong, Xiaobo Hu, Zixia Jia, Yang Liu, Tao Peng, Yixin Ren, Ran Tian, Zaiyuan Wang, Yanglihong Xiao, Gang Yao, Lingyue Yin, Ge Zhang, Chun Zhang, Jianpeng Jiao, Zilong Zheng, Yuan Gong
NeurIPS 2026Evaluations and Datasets Track
* equal contribution
2025
Understanding and Leveraging the Expert Specialization of Context Faithfulness in Mixture-of-Experts LLMs
Jun Bai, Minghao Tong, Yang Liu, Zixia Jia, Zilong Zheng
EMNLP 2025Top 2%
Reinforced Query Reasoners for Reasoning-intensive Retrieval Tasks
Xubo Qin, Jun Bai, Jiaqi Li, Zixia Jia, Zilong Zheng
EMNLP 2025
CogAtom: From Cognitive Atoms to Olympiad-level Mathematical Reasoning in Large Language Models
Zhuofan Chen, Jiyuan He, Yichi Zhang, Xing Hu, Haoxing Wen, Jun Bai, Wenge Rong
EMNLP Findings 2025
Selecting Text Classification Model through Maximizing Posterior Evidence over Informative Sub-space
Zhiwei Sun, Jun Bai#, Zhuofan Chen, Chen Li, Wenge Rong, Zhang Xiong
Frontiers of Computer Science 2025
# corresponding author
Selecting Demonstrations for Many-Shot In-Context Learning via Gradient Matching
Jianfei Zhang, Bei Li, Jun Bai, Rumei Li, Yanmeng Wang, Chenghua Lin, Wenge Rong
ACL Findings 2025
Rectifying and Discriminating Hard Negatives for Biomedical Retrieval Question Answering
Jun Bai, Zhenzi Li, Bo Zhao, Chen Li, Chenghua Lin, Wenge Rong
IEEE Transactions on Computational Biology and Bioinformatics 2025
Disentangling Preference Representation and Text Generation for Efficient Individual Preference Alignment
Jianfei Zhang, Jun Bai, Bei Li, Yanmeng Wang, Rumei Li, Chenghua Lin, Wenge Rong
COLING 2025
2024
Leveraging Estimated Transferability over Human Intuition for Model Selection in Text Ranking
Jun Bai, Zhuofan Chen, Zhenzi Li, Hanhua Hong, Jianfei Zhang, Chen Li, Chenghua Lin, Wenge Rong
EMNLP 2024
2023
How to Determine the Most Powerful Pre-trained Language Model without Brute Force Fine-tuning? An Empirical Survey
Jun Bai, Xiaofeng Zhang, Chen Li, Hanhua Hong, Xi Xu, Chenghua Lin, Wenge Rong
EMNLP Findings 2023
Multisource Soft Labeling and Hard Negative Sampling for Retrieval Distractor Ranking
Jiayun Wang, Wenge Rong, Jun Bai, Zhiwei Sun, Yuanxin Ouyang, Zhang Xiong
IEEE Transactions on Learning Technologies 2023
Permutation Invariant Training for Paraphrase Identification
Jun Bai, Chuantao Yin, Hanhua Hong, Jianfei Zhang, Chen Li, Yanmeng Wang, Wenge Rong
ICASSP 2023Oral
Enhancing Biomedical Reqa with Adversarial Hard In-batch Negative Samples
Bo Zhao, Jun Bai, Chen Li, Jianfei Zhang, Wenge Rong, Yuanxin Ouyang, Zhang Xiong
IEEE/ACM Transactions on Computational Biology and Bioinformatics 2023
2022
Improving Variational Autoencoders with Density Gap-based Regularization
Jianfei Zhang, Jun Bai, Chenghua Lin, Yanmeng Wang, Wenge Rong
NeurIPS 2022
Goal-oriented Conditional Variational Autoencoders for Proactive and Knowledge-aware Conversational Recommender System
Cen Yan, Jun Bai, Yanmeng Wang, Wenge Rong, Yuanxin Ouyang, Zhang Xiong
Computer Speech & Language 2022
Improving Biomedical ReQA with Consistent NLI-transfer and Post-whitening
Jun Bai, Chuantao Yin, Zimeng Wu, Jianfei Zhang, Yanmeng Wang, Guanyi Jia, Wenge Rong, Zhang Xiong
IEEE/ACM Transactions on Computational Biology and Bioinformatics 2022
Adversarial Knowledge Distillation based Biomedical Factoid Question Answering
Jun Bai, Chuantao Yin, Jianfei Zhang, Yanmeng Wang, Yi Dong, Wenge Rong, Zhang Xiong
IEEE/ACM Transactions on Computational Biology and Bioinformatics 2022
2021
Enhancing Dual-encoders with Question and Answer Cross-embeddings for Answer Retrieval
Yanmeng Wang, Jun Bai, Ye Wang, Jianfei Zhang, Wenge Rong, Zongcheng Ji, Shaojun Wang, Jing Xiao
EMNLP Findings 2021
Paragraph Level Multi-perspective Context Modeling for Question Generation
Jun Bai, Wenge Rong, Feiyu Xia, Yanmeng Wang, Yuanxin Ouyang, Zhang Xiong
ICASSP 2021
