atlas

Research map / Reinforcement learning / Policy gradient and actor-critic

Coprocessor Actor Critic: A Model-Based Reinforcement Learning Approach For Adaptive Brain Stimulation

ICML 2024 · Reinforcement learning › Policy gradient and actor-critic · approach: Stabilizing deep actor-critic · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes Coprocessor Actor Critic, a model-based RL method that learns a Q-function for world actions in a biomechanical simulator, then learns a brain stimulation-to-action model online, guided by and recalibrating that Q-function, for adaptive brain stimulation in simulated stroke.

Problem

Adaptive brain stimulation policies for stroke patients must be learned with very few costly patient interactions, while model-free RL needs many environment samples.

Approach

Decouple coprocessor policy learning into learning optimal world actions via a biomechanical simulator (Q-function), then learning a stimulation-to-world-action brain model online, sampling stimulations that maximize the simulator Q-function and iteratively updating the Q-function over the learned brain model's realizable actions.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use CopAC instead of SACAdaptive brain stimulation control tasks · Control environmentssample efficiency: 10xHelped
Use CopAC instead of SACAdaptive brain stimulation control tasks · Control environmentsevaluation rewardHelped
Use CopAC instead of MBPOAdaptive brain stimulation control tasks · MyoSim Poseevaluation rewardHelped
Use CopAC instead of MBPOAdaptive brain stimulation control tasks · MyoSim Reach and Pendulumevaluation rewardNo clear effect
Add online Q-function updates (ablation)CopAC · Swimmerevaluation rewardHelped
Add online Q-function updates (ablation)CopAC · Reacherevaluation rewardHelped
Add online Q-function updates (ablation)CopAC · Several other control domainsevaluation rewardHelped
Use Q-maximizing stimulation sampling instead of random sampling (ablation)CopAC without Q updates · MyoSim Poseevaluation rewardHelped
Use Q-maximizing stimulation sampling instead of random sampling (ablation)CopAC without Q updates · Swimmer, Reacher, Pendulum, and Inverted Double Pendulumevaluation rewardHelped
Use Q-maximizing stimulation sampling instead of random sampling (ablation)CopAC · All evaluated environmentstraining rewardHelped
Learn the world-action Q-function using offline RL instead of simulationCopAC · Most environmentstraining rewardHelped
Learn the world-action Q-function using offline RL instead of simulationCopAC · Evaluated environmentstraining rewardHurt

Surprising findings

Limitations

Experimental setup

Datasets: MyoSuite simulated reaching and pose tasks OpenAI Gym continuous control tasks

Benchmarks: MyoSim Arm Reach MyoSim Hand Pose OpenAI Gym Swimmer OpenAI Gym Reacher OpenAI Gym Pendulum OpenAI Gym Inverted Double Pendulum LunarLander

Metrics: evaluation reward training reward sample efficiency

Baselines: Soft Actor-Critic Model-Based Policy Optimization CopAC without Q update CopAC without Q update and Q max inverse brain model coprocessor

Builds on: Soft Actor-Critic Model-Based Policy Optimization MyoSuite Conservative Soft Actor Critic

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy