Yu, Dianhai

7 publications

TMLR 2025 Federated Generalized Novel Category Discovery with Prompts Tuning Lei Shen, Nan Pu, Zhun Zhong, Mingming Gong, Dianhai Yu, Chengqi Zhang, Bo Han
ICLR 2025 FlashMask: Efficient and Rich Mask Extension of FlashAttention Guoxia Wang, Jinle Zeng, Xiyuan Xiao, Siming Wu, Jiabin Yang, Lujing Zheng, Zeyu Chen, Jiang Bian, Dianhai Yu, Haifeng Wang
IJCAI 2022 Simple and Effective Relation-Based Embedding Propagation for Knowledge Representation Learning Huijuan Wang, Siming Dai, Weiyue Su, Hui Zhong, Zeyang Fang, Zhengjie Huang, Shikun Feng, Zeyu Chen, Yu Sun, Dianhai Yu
NeurIPS 2022 TA-MoE: Topology-Aware Large Scale Mixture-of-Expert Training Chang Chen, Min Li, Zhihua Wu, Dianhai Yu, Chao Yang
IJCAI 2021 Rethinking Label-Wise Cross-Modal Retrieval from a Semantic Sharing Perspective Yang Yang, Chubing Zhang, Yi-Chu Xu, Dianhai Yu, De-Chuan Zhan, Jian Yang
IJCAI 2019 RLTM: An Efficient Neural IR Framework for Long Documents Chen Zheng, Yu Sun, Shengxian Wan, Dianhai Yu
ICLR 2018 A New Method of Region Embedding for Text Classification Chao Qiao, Bo Huang, Guocheng Niu, Daren Li, Daxiang Dong, Wei He, Dianhai Yu, Hua Wu