Yang, Kai

27 publications

ICML 2025 Contextures: Representations from Contexts Runtian Zhai, Kai Yang, Burak Varıcı, Che-Ping Tsai, J Zico Kolter, Pradeep Kumar Ravikumar

ICML 2025 DTZO: Distributed Trilevel Zeroth Order Learning with Provable Non-Asymptotic Convergence Yang Jiao, Kai Yang, Chengtao Jian

NeurIPS 2025 From Sequence to Structure: Uncovering Substructure Reasoning in Transformers Xinnan Dai, Kai Yang, Jay Revolinsky, Kai Guo, Aoran Wang, Bohang Zhang, Jiliang Tang

AAAI 2025 GPU-Accelerated Parallel Bilevel Optimization for Roubst 6g ISAC Xingdi Chen, Kai Yang

AAAI 2025 Novelty-Guided Data Reuse for Efficient and Diversified Multi-Agent Reinforcement Learning Yangkun Chen, Kai Yang, Jian Tao, Jiafei Lyu

IJCAI 2024 BATON: Aligning Text-to-Audio Model Using Human Preference Feedback Huan Liao, Haonan Han, Kai Yang, Tianjiao Du, Rui Yang, Qinmei Xu, Zunnan Xu, Jingquan Liu, Jiasheng Lu, Xiu Li

ICML 2024 Do Efficient Transformers Really Save Computation? Kai Yang, Jan Ackermann, Zhenyu He, Guhao Feng, Bohang Zhang, Yunzhen Feng, Qiwei Ye, Di He, Liwei Wang

ICML 2024 Exploration and Anti-Exploration with Distributional Random Network Distillation Kai Yang, Jian Tao, Jiafei Lyu, Xiu Li

AAAI 2024 Improving IP Geolocation with Target-Centric IP Graph (Student Abstract) Kai Yang, Jiayang Li, Wenxin Tai, Zhenhui Li, Ting Zhong, Guangqiang Yin, Yong Wang

AAAI 2024 Interpreting Temporal Knowledge Graph Reasoning (Student Abstract) Bin Chen, Kai Yang, Wenxin Tai, Zhangtao Cheng, Leyuan Liu, Ting Zhong, Fan Zhou

AAAI 2024 Provably Convergent Federated Trilevel Learning Yang Jiao, Kai Yang, Tiancheng Wu, Chengtao Jian, Jianwei Huang

AAAI 2024 Robust Beamforming for Downlink Multi-Cell Systems: A Bilevel Optimization Perspective Xingdi Chen, Yu Xiong, Kai Yang

NeurIPS 2024 Tri-Level Navigator: LLM-Empowered Tri-Level Learning for Time Series OOD Generalization Chengtao Jian, Kai Yang, Yang Jiao

ICML 2024 Triadic-OCD: Asynchronous Online Change Detection with Provable Robustness, Optimality, and Convergence Yancheng Huang, Kai Yang, Zelin Zhu, Leian Chen

ICML 2024 Two Stones Hit One Bird: Bilevel Positional Encoding for Better Length Extrapolation Zhenyu He, Guhao Feng, Shengjie Luo, Kai Yang, Liwei Wang, Jingjing Xu, Zhi Zhang, Hongxia Yang, Di He

CVPR 2024 Using Human Feedback to Fine-Tune Diffusion Models Without Any Reward Model Kai Yang, Jian Tao, Jiafei Lyu, Chunjiang Ge, Jiaxin Chen, Weihan Shen, Xiaolong Zhu, Xiu Li

ICCV 2023 ASM: Adaptive Skinning Model for High-Quality 3D Face Modeling Kai Yang, Hong Shang, Tianyang Shi, Xinghan Chen, Jingkai Zhou, Zhongqian Sun, Wei Yang

ICLR 2023 Asynchronous Distributed Bilevel Optimization Yang Jiao, Kai Yang, Tiancheng Wu, Dongjin Song, Chengtao Jian

WACV 2023 Few-Shot Object Counting with Similarity-Aware Feature Enhancement Zhiyuan You, Kai Yang, Wenhan Luo, Xin Lu, Lei Cui, Xinyi Le

AAAI 2023 KerPrint: Local-Global Knowledge Graph Enhanced Diagnosis Prediction for Retrospective and Prospective Interpretations Kai Yang, Yongxin Xu, Peinie Zou, Hongxin Ding, Junfeng Zhao, Yasha Wang, Bing Xie

IJCAI 2023 VecoCare: Visit Sequences-Clinical Notes Joint Learning for Diagnosis Prediction in Healthcare Data Yongxin Xu, Kai Yang, Chaohe Zhang, Peinie Zou, Zhiyuan Wang, Hongxin Ding, Junfeng Zhao, Yasha Wang, Bing Xie

NeurIPS 2022 A Unified Model for Multi-Class Anomaly Detection Zhiyuan You, Lei Cui, Yujun Shen, Kai Yang, Xin Lu, Yu Zheng, Xinyi Le

NeurIPS 2022 Distributed Distributionally Robust Optimization with Non-Convex Objectives Yang Jiao, Kai Yang, Dongjin Song

WACV 2021 DualSANet: Dual Spatial Attention Network for Iris Recognition Kai Yang, Zihao Xu, Jingjing Fei

AAAI 2020 COTSAE: CO-Training of Structure and Attribute Embeddings for Entity Alignment Kai Yang, Shaoqin Liu, Junfeng Zhao, Yasha Wang, Bing Xie

IJCAI 2017 Autoencoder Regularized Network for Driving Style Representation Learning Weishan Dong, Ting Yuan, Kai Yang, Changsheng Li, Shilei Zhang

AAAI 2017 TaGiTeD: Predictive Task Guided Tensor Decomposition for Representation Learning from Electronic Health Records Kai Yang, Xiang Li, Haifeng Liu, Jing Mei, Guo Tong Xie, Junfeng Zhao, Bing Xie, Fei Wang