Modeling Heterogeneous Hierarchies with Relation-specific Hyperbolic Cones. - Yushi Bai; Zhitao Ying; Hongyu Ren; Jure Leskovec (NeurIPS, 2021)

Envy-Free and Pareto-Optimal Allocations for Asymmetric Agents. - Yushi Bai; Paul Gölz (CoRR, 2021)

SQUIRE: A Sequence-to-sequence Framework for Multi-hop Knowledge Graph Reasoning. - Yushi Bai; Xin Lv; Juanzi Li; Lei Hou 001; Yincen Qu; Zelin Dai; Feiyu Xiong (EMNLP, 2022)

Envy-Free and Pareto-Optimal Allocations for Agents with Asymmetric Random Valuations. - Yushi Bai; Paul Gölz (IJCAI, 2022)

Fair Allocations for Smoothed Utilities. - Yushi Bai; Uriel Feige; Paul Gölz; Ariel D. Procaccia (EC, 2022)

Answering Complex Logical Queries on Knowledge Graphs via Query Computation Tree Optimization. - Yushi Bai; Xin Lv; Juanzi Li; Lei Hou 001 (ICML, 2023)

Benchmarking Foundation Models with Language-Model-as-an-Examiner. - Yushi Bai; Jiahao Ying; Yixin Cao 002; Xin Lv; Yuze He; Xiaozhi Wang; Jifan Yu; Kaisheng Zeng; Yijia Xiao; Haozhe Lyu; Jiayin Zhang; Juanzi Li; et al. (NeurIPS, 2023)

Large Language Models Can Be Good Privacy Protection Learners. - Yijia Xiao; Yiqiao Jin; Yushi Bai; Yue Wu; Xianjun Yang; Xiao Luo 001; Wenchao Yu; Xujiang Zhao; Yanchi Liu; Haifeng Chen; Wei Wang 010; Wei Cheng 002 (CoRR, 2023)

T3Bench: Benchmarking Current Progress in Text-to-3D Generation. - Yuze He; Yushi Bai; Matthieu Lin; Wang Zhao 001; Yubin Hu 001; Jenny Sheng; Ran Yi; Juanzi Li; Yong-Jin Liu 001 (CoRR, 2023)

Know2BIO: A Comprehensive Dual-View Benchmark for Evolving Biomedical Knowledge Graphs. - Yijia Xiao; Dylan Steinecke; Alexander Russell Pelletier; Yushi Bai; Peipei Ping; Wei Wang 010 (CoRR, 2023)

LongBench: A Bilingual, Multitask Benchmark for Long Context Understanding. - Yushi Bai; Xin Lv; Jiajie Zhang; Hongchang Lyu; Jiankai Tang; Zhidian Huang; Zhengxiao Du; Xiao Liu 036; Aohan Zeng; Lei Hou 001; Yuxiao Dong; Jie Tang 001; et al. (ACL (1), 2024)

WaterBench: Towards Holistic Evaluation of Watermarks for Large Language Models. - Shangqing Tu; Yuliang Sun; Yushi Bai; Jifan Yu; Lei Hou 001; Juanzi Li (ACL (1), 2024)

LongAlign: A Recipe for Long Context Alignment of Large Language Models. - Yushi Bai; Xin Lv; Jiajie Zhang; Yuze He; Ji Qi 003; Lei Hou 001; Jie Tang 001; Yuxiao Dong; Juanzi Li (EMNLP (Findings), 2024)

MM-MATH: Advancing Multimodal Math Evaluation with Process Evaluation and Fine-grained Classification. - Kai Sun; Yushi Bai; Ji Qi 003; Lei Hou 001; Juan-Zi Li (EMNLP (Findings), 2024)

Large Language Models Can Be Contextual Privacy Protection Learners. - Yijia Xiao; Yiqiao Jin; Yushi Bai; Yue Wu; Xianjun Yang; Xiao Luo 001; Wenchao Yu; Xujiang Zhao; Yanchi Liu; Quanquan Gu; Haifeng Chen; Wei Wang 010; et al. (EMNLP, 2024)

KoLA: Carefully Benchmarking World Knowledge of Large Language Models. - Jifan Yu; Xiaozhi Wang; Shangqing Tu; Shulin Cao; Daniel Zhang-Li; Xin Lv; Hao Peng 015; Zijun Yao 002; Xiaohan Zhang; Hanming Li; Chunyang Li; Zheyuan Zhang; et al. (ICLR, 2024)

AlphaTablets: A Generic Plane Representation for 3D Planar Reconstruction from Monocular Videos. - Yuze He; Wang Zhao 001; Shaohui Liu; Yubin Hu 001; Yushi Bai; Yu-Hui Wen; Yongjin Liu 001 (NeurIPS, 2024)

Automating Dataset Updates Towards Reliable and Timely Evaluation of Large Language Models. - Jiahao Ying; Yixin Cao 002; Yushi Bai; Qianru Sun; Bo Wang; Wei Tang 015; Zhaojun Ding; Yizhe Yang; Xuanjing Huang 001; Shuicheng Yan (NeurIPS, 2024)

Text-image conditioned diffusion for consistent text-to-3D generation. - Yuze He; Yushi Bai; Matthieu Lin; Jenny Sheng; Yubin Hu 001; Qi Wang 079; Yu-Hui Wen; Yong-Jin Liu 001 (Comput. Aided Geom. Des., 2024)

CogCoM: Train Large Vision-Language Models Diving into Details through Chain of Manipulations. - Ji Qi 003; Ming Ding 004; Weihan Wang; Yushi Bai; Qingsong Lv; Wenyi Hong; Bin Xu 001; Lei Hou 001; Juanzi Li; Yuxiao Dong; Jie Tang 001 (CoRR, 2024)

Advancing Geometric Problem Solving: A Comprehensive Benchmark for Multimodal Model Evaluation. - Kai Sun; Yushi Bai; Nianyi Lin (CoRR, 2024)

DICE: Detecting In-distribution Contamination in LLM's Fine-tuning Phase for Math Reasoning. - Shangqing Tu; Kejian Zhu; Yushi Bai; Zijun Yao 002; Lei Hou 001; Juanzi Li (CoRR, 2024)

ChatGLM: A Family of Large Language Models from GLM-130B to GLM-4 All Tools. - Aohan Zeng; Bin Xu 001; Bowen Wang; Chenhui Zhang; Da Yin; Diego Rojas; Guanyu Feng; Hanlin Zhao; Hanyu Lai; Hao Yu; Hongning Wang; Jiadai Sun; et al. (CoRR, 2024)

Finding Safety Neurons in Large Language Models. - Jianhui Chen; Xiaozhi Wang; Zijun Yao 002; Yushi Bai; Lei Hou 001; Juanzi Li (CoRR, 2024)

LongWriter: Unleashing 10,000+ Word Generation from Long Context LLMs. - Yushi Bai; Jiajie Zhang; Xin Lv; Linzhi Zheng; Siqi Zhu; Lei Hou 001; Yuxiao Dong; Jie Tang 001; Juanzi Li (CoRR, 2024)

Pre-training Distillation for Large Language Models: A Design Space Exploration. - Hao Peng 015; Xin Lv; Yushi Bai; Zijun Yao 002; Jiajie Zhang; Lei Hou 001; Juanzi Li (ACL (1), 2025)

LongBench v2: Towards Deeper Understanding and Reasoning on Realistic Long-context Multitasks. - Yushi Bai; Shangqing Tu; Jiajie Zhang; Hao Peng 015; Xiaozhi Wang; Xin Lv; Shulin Cao; Jiazheng Xu; Lei Hou 001; Yuxiao Dong; Jie Tang 001; Juanzi Li (ACL (1), 2025)

LongCite: Enabling LLMs to Generate Fine-grained Citations in Long-Context QA. - Jiajie Zhang; Yushi Bai; Xin Lv; Wanjun Gu; Danqing Liu; Minhao Zou; Shulin Cao; Lei Hou 001; Yuxiao Dong; Ling Feng; Juanzi Li (ACL (Findings), 2025)

LongWriter: Unleashing 10, 000+ Word Generation from Long Context LLMs. - Yushi Bai; Jiajie Zhang; Xin Lv; Linzhi Zheng; Siqi Zhu; Lei Hou 001; Yuxiao Dong; Jie Tang 001; Juanzi Li (ICLR, 2025)

CogCoM: A Visual Language Model with Chain-of-Manipulations Reasoning. - Ji Qi 003; Ming Ding 004; Weihan Wang; Yushi Bai; Qingsong Lv; Wenyi Hong; Bin Xu 001; Lei Hou 001; Juanzi Li; Yuxiao Dong; Jie Tang 001 (ICLR, 2025)

Long Context vs. RAG: Strategies for Processing Long Documents in LLMs. - Xinze Li; Yushi Bai; Bowen Jin; Fengbin Zhu; Liangming Pan; Yixin Cao 002 (SIGIR, 2025)

LongWriter-V: Enabling Ultra-Long and High-Fidelity Generation in Vision-Language Models. - Shangqing Tu; Yucheng Wang; Daniel Zhang-Li; Yushi Bai; Jifan Yu; Yuhao Wu; Lei Hou 001; Huiqin Liu; Zhiyuan Liu 001; Bin Xu 001; Juanzi Li (CoRR, 2025)

Shifting Long-Context LLMs Research from Input to Output. - Yuhao Wu; Yushi Bai; Zhiqing Hu; Shangqing Tu; Ming Shan Hee; Juanzi Li; Roy Ka-Wei Lee (CoRR, 2025)

An LMM for Efficient Video Understanding via Reinforced Compression of Video Cubes. - Ji Qi 003; Yuan Yao 011; Yushi Bai; Bin Xu 001; Juanzi Li; Zhiyuan Liu 001; Tat-Seng Chua (CoRR, 2025)

Hard Negative Contrastive Learning for Fine-Grained Geometric Understanding in Large Multimodal Models. - Kai Sun; Yushi Bai; Zhen Yang 034; Jiajie Zhang; Ji Qi 003; Lei Hou 001; Juanzi Li (CoRR, 2025)

How does Transformer Learn Implicit Reasoning? - Jiaran Ye; Zijun Yao 002; Zhidian Huang; Liangming Pan; Jinxin Liu 002; Yushi Bai; Amy Xin; Weichuan Liu; Xiaoyin Che; Lei Hou 001; Juanzi Li (CoRR, 2025)

SuperWriter: Reflection-Driven Long-Form Generation with Large Language Models. - Yuhao Wu; Yushi Bai; Zhiqiang Hu; Juanzi Li; Roy Ka-Wei Lee (CoRR, 2025)

LongWriter-Zero: Mastering Ultra-Long Text Generation via Reinforcement Learning. - Yuhao Wu; Yushi Bai; Zhiqiang Hu; Roy Ka-Wei Lee; Juanzi Li (CoRR, 2025)

GLM-4.5: Agentic, Reasoning, and Coding (ARC) Foundation Models. - Aohan Zeng; Xin Lv; Qinkai Zheng; Zhenyu Hou; Bin Chen; Chengxing Xie; Cunxiang Wang; Da Yin; Hao Zeng; Jiajie Zhang; Kedong Wang; Lucen Zhong; et al. (CoRR, 2025)