Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Prompted Policy Search: Reinforcement Learning through Linguistic and Numerical Reasoning in LLMs

About

Reinforcement Learning (RL) traditionally relies on scalar reward signals, limiting its ability to leverage the rich semantic knowledge often available in real-world tasks. In contrast, humans learn efficiently by combining numerical feedback with language, prior knowledge, and common sense. We introduce Prompted Policy Search (ProPS), a novel RL method that unifies numerical and linguistic reasoning within a single framework. Unlike prior work that augment existing RL components with language, ProPS places a large language model (LLM) at the center of the policy optimization loop-directly proposing policy updates based on both reward feedback and natural language input. We show that LLMs can perform numerical optimization in-context, and that incorporating semantic signals, such as goals, domain knowledge, and strategy hints can lead to more informed exploration and sample-efficient learning. ProPS is evaluated across fifteen Gymnasium tasks, spanning classic control, Atari games, and MuJoCo environments, and compared to seven widely-adopted RL algorithms (e.g., PPO, SAC, TRPO). It outperforms all baselines on eight out of fifteen tasks and demonstrates substantial gains when provided with domain knowledge. These results highlight the potential of unifying semantics and numerics for transparent, generalizable, and human-aligned RL.

Yifan Zhou, Sachin Grover, Mohamed El Mistiri, Kamalesh Kalirathnam, Pratyush Kerhalkar, Swaroop Mishra, Neelesh Kumar, Sanket Gaurav, Oya Aran, Heni Ben Amor• 2025

Related benchmarks

TaskDatasetResultRank
Reinforcement LearningMountainCarContinuous v0
Average Agent Reward98.7
65
Reinforcement Learningcartpole
Average Reward258.1
29
Reinforcement LearningSwimmer
Average Returns162.1
24
Reinforcement LearningMountainCar
Avg Episode Reward199.3
18
Reinforcement LearningInverted Double Pendulum
Avg Episode Reward128.8
18
Reinforcement LearningMountainCar
Maximum Return126.1
14
Reinforcement LearningFrozenLake
Reward0.37
12
Reinforcement Learningcartpole
Max Return427.7
9
Reinforcement LearningMaze
Mean Reward1.03
8
Reinforcement LearningInvertedPendulum
Mean Reward657.9
8
Showing 10 of 24 rows

Other info

Follow for update