publications

Full publication lists sorted in time order

2024

  1. Accelerating Hybrid Federated Learning Convergence Under Partial Participation
    Jieming Bian, Lei Wang, Kun Yang, and 2 more authors
    IEEE Transactions on Signal Processing, 2024
  2. Offline Reinforcement Learning for Wireless Network Optimization with Mixture Datasets
    Kun Yang, Chengshuai Shi, Cong Shen, and 3 more authors
    IEEE Transactions on Wireless Communications, 2024
  3. Harnessing the Power of Federated Learning in Federated Contextual Bandits
    Chengshuai Shi, Ruida Zhou, Kun Yang, and 1 more author
    Transactions on Machine Learning Research, 2024
  4. Efficient Prompt Optimization Through the Lens of Best Arm Identification
    Chengshuai Shi*, Kun Yang*, Zihan Chen, and 3 more authors
    Advances in neural information processing systems, 2024
  5. Transformers as Game Players: Provable In-context Game-playing Capabilities of Pre-trained Models
    Chengshuai Shi, Kun Yang, Jing Yang, and 1 more author
    Advances in neural information processing systems, 2024
  6. Average Reward Reinforcement Learning for Wireless Radio Resource Management
    Kun Yang, Jing Yang, and Cong Shen
    In 2024 58th Asilomar Conference on Signals, Systems, and Computers, 2024

2023

  1. Teaching Reinforcement Learning Agents via Reinforcement Learning
    Kun Yang, Chengshuai Shi, and Cong Shen
    In 2023 57th Annual Conference on Information Sciences and Systems (CISS), 2023
  2. Advancing RAN Slicing with Offline Reinforcement Learning
    Kun Yang, Shu-ping Yeh, Menglei Zhang, and 3 more authors
    2023

2022

  1. On the convergence of hybrid federated learning with server-clients collaborative training
    Kun Yang, and Cong Shen
    In 2022 56th Annual Conference on Information Sciences and Systems (CISS), 2022
  2. Multi-Agent Reinforcement Learning for Wireless User Scheduling: Performance, Scalablility, and Generalization
    Kun Yang, Donghao Li, Cong Shen, and 3 more authors
    In 2022 56th Asilomar Conference on Signals, Systems, and Computers, 2022
  3. On the Convergence of Hybrid Server-Clients Collaborative Training
    Kun Yang, Shengbo Chen, and Cong Shen
    IEEE Journal on Selected Areas in Communications, 2022

2020

  1. Deep reinforcement learning based wireless network optimization: A comparative study
    Kun Yang, Cong Shen, and Tie Liu
    In IEEE INFOCOM 2020-IEEE Conference on Computer Communications Workshops (INFOCOM WKSHPS), 2020