Proximal Residual Value Functions for Consistent Planning and Real-Time Execution
Organizations: Amazon SCOT
Abstract
We study two-timescale decision systems in which a planning layer periodically supplies a continuation-value function to a real-time optimizer that allocates arriving resources, with inventory placement as our motivating application. We propose an end-to-end reinforcement learning (RL) method for learning this function using proximal residual value functions, which combine a strictly convex potential of post-decision inventory with a learned convex residual. This general form yields a well-posed optimization layer that supports end-to-end differentiation while preserving an explicit convex objective for real-time execution. We characterize the necessary and sufficient conditions under which a smooth value function yields decisions that are consistent across the planning and execution timescales. In an offline simulation using historical inventory arrival and demand patterns from a large e-commerce retailer, learned proximal residual value functions reduce total routing and transfer cost relative to a historical-production-system proxy by 5.0%.
Figures & tables
| Policy | In sample | Out of sample | ||
|---|---|---|---|---|
| Historical proxy | 1.000 | 1.000 | ||
| Direct neural RL | ||||
| PWL training solver | State potential | State potential | ||
| With | Without | With | Without | |
| Picard iteration | ||||
| Mirror-prox | ||||
Value Function Estimation
Value function estimation predicts long-term returns or outcomes from states and actions to support reinforcement learning, planning, and policy evaluation.
See all 68 papers