← Search

Edwin Zhang

6 accepted papers

2024

A Decision-Language Model (DLM) for Dynamic Restless Multi-Armed Bandit Tasks in Public Health

NeurIPS 2024poster

Restless multi-armed bandits (RMAB) have demonstrated success in optimizing resource allocation for large beneficiary populations in public health settings. Unfortunately, RMAB models lack flexibility to adapt to evolving public health policy priorities. Concurrently, Large Language Models (LLMs) ha…

Cited by 14SourcePDFScholar
2024

Language Control Diffusion: Efficiently Scaling through Space, Time, and Tasks

ICLR 2024poster

Training generalist agents is difficult across several axes, requiring us to deal with high-dimensional inputs (space), long horizons (time), and generalization to novel tasks. Recent advances with architectures have allowed for improved scaling along one or two of these axes, but are still computat…

2024

Position: Social Environment Design Should be Further Developed for AI-based Policy-Making

ICML 2024poster

Artificial Intelligence (AI) holds promise as a technology that can be used to improve government and economic policy-making. This paper proposes a new research agenda towards this end by introducing **Social Environment Design**, a general framework for the use of AI in automated policy-making that…

Cited by 5SourcePDFScholar
2024

Towards a Pretrained Model for Restless Bandits via Multi-arm Generalization

IJCAI 2024poster

Restless multi-arm bandits (RMABs) is a class of resource allocation problems with broad application in areas such as healthcare, online advertising, and anti-poaching. We explore several important question such as how to handle arms opting-in and opting-out over time without frequent retraining fro…

2024

Transcendence: Generative Models Can Outperform The Experts That Train Them

NeurIPS 2024poster

Generative models are trained with the simple objective of imitating the conditional probability distribution induced by the data they are trained on. Therefore, when trained on data generated by humans, we may not expect the artificial model to outperform the humans on their original objectives. In…

Cited by 11SourcePDFScholar
2023

Offline Reinforcement Learning with Closed-Form Policy Improvement Operators

ICML 2023poster

Behavior constrained policy optimization has been demonstrated to be a successful paradigm for tackling Offline Reinforcement Learning. By exploiting historical transitions, a policy is trained to maximize a learned value function while constrained by the behavior policy to avoid a significant distr…