activity
20232026
most citedEfficient Prompt Optimization Through the Lens of Best Arm Identification

1 citations · 1 across the 5 of their papers we have counts for

collaborators

5 papers

eess.SY2026

Hybrid Offline-Online Multi-Agent Decision Transformers for Wireless Resource Management

Yiming Zhang, Kun Yang, Cong Shen +1

This paper develops a hybrid offline-online multi-agent reinforcement learning framework based on decision transformers. The policy is first pretrained offline via supervised seque…

cs.IT2025

Average Reward Reinforcement Learning for Wireless Radio Resource Management

Kun Yang, Jing Yang, Cong Shen

In this paper, we address a crucial but often overlooked issue in applying reinforcement learning (RL) to radio resource management (RRM) in wireless communications: the mismatch b…

stat.ML2024

Transformers as Game Players: Provable In-context Game-playing Capabilities of Pre-trained Models

Chengshuai Shi, Kun Yang, Jing Yang +1

The in-context learning (ICL) capability of pre-trained models based on the transformer architecture has received growing interest in recent years. While theoretical understanding…

stat.ML2024★ 1 cited

Efficient Prompt Optimization Through the Lens of Best Arm Identification

Chengshuai Shi, Kun Yang, Zihan Chen +3

The remarkable instruction-following capability of large language models (LLMs) has sparked a growing interest in automatically finding good prompts, i.e., prompt optimization. Mos…

stat.ML2023

Harnessing the Power of Federated Learning in Federated Contextual Bandits

Chengshuai Shi, Ruida Zhou, Kun Yang +1

Federated learning (FL) has demonstrated great potential in revolutionizing distributed machine learning, and tremendous efforts have been made to extend it beyond the original foc…