Lei, Tao

11 publications

ICCV 2025 Adaptive Learning of High-Value Regions for Semi-Supervised Medical Image Segmentation Tao Lei, Ziyao Yang, Xingwu Wang, Yi Wang, Xuan Wang, Feiman Sun, Asoke K. Nandi

ICLR 2025 EC-DIT: Scaling Diffusion Transformers with Adaptive Expert-Choice Routing Haotian Sun, Tao Lei, Bowen Zhang, Yanghao Li, Haoshuo Huang, Ruoming Pang, Bo Dai, Nan Du

ICML 2025 Instruction-Following Pruning for Large Language Models Bairu Hou, Qibin Chen, Jianyu Wang, Guoli Yin, Chong Wang, Nan Du, Ruoming Pang, Shiyu Chang, Tao Lei

ECCV 2024 MM1: Methods, Analysis & Insights from Multimodal LLM Pre-Training Brandon McKinzie, Zhe Gan, Jean-Philippe Fauconnier, Samuel Dodge, Bowen Zhang, Philipp Dufter, Dhruti Shah, Futang Peng, Anton Belyi, Max A Schwarzer, Hongyu Hè, Xianzhi Du, Haotian Zhang, Karanjeet Singh, Doug Kang, Tom Gunter, Xiang Kong, Aonan Zhang, Jianyu Wang, Chong Wang, Nan Du, Tao Lei, Sam Wiseman, Mark Lee, Zirui Wang, Ruoming Pang, Peter Grasch, Alexander Toshev, Yinfei Yang

IJCAI 2023 CiT-Net: Convolutional Neural Networks Hand in Hand with Vision Transformers for Medical Image Segmentation Tao Lei, Rui Sun, Xuan Wang, Yingbo Wang, Xi He, Asoke K. Nandi

NeurIPS 2023 Conditional Adapters: Parameter-Efficient Transfer Learning with Fast Inference Tao Lei, Junwen Bai, Siddhartha Brahma, Joshua Ainslie, Kenton Lee, Yanqi Zhou, Nan Du, Vincent Zhao, Yuexin Wu, Bo Li, Yu Zhang, Ming-Wei Chang

NeurIPS 2023 Rethinking the Role of Token Retrieval in Multi-Vector Retrieval Jinhyuk Lee, Zhuyun Dai, Sai Meher Karthik Duddu, Tao Lei, Iftekhar Naim, Ming-Wei Chang, Vincent Zhao

NeurIPS 2022 Mixture-of-Experts with Expert Choice Routing Yanqi Zhou, Tao Lei, Hanxiao Liu, Nan Du, Yanping Huang, Vincent Zhao, Andrew M Dai, Zhifeng Chen, Quoc V Le, James Laudon

AAAI 2021 Nutri-Bullets: Summarizing Health Studies by Composing Segments Darsh J. Shah, Lili Yu, Tao Lei, Regina Barzilay

ICML 2017 Deriving Neural Architectures from Sequence and Graph Kernels Tao Lei, Wengong Jin, Regina Barzilay, Tommi Jaakkola

NeurIPS 2017 Style Transfer from Non-Parallel Text by Cross-Alignment Tianxiao Shen, Tao Lei, Regina Barzilay, Tommi Jaakkola