Reinforcement learning under temporal logic constraints as a sequence modeling problem

Daiying Tian,Hao Fang,Qingkai Yang,Haoyong Yu,Wenyu Liang,Yan Wu
DOI: https://doi.org/10.1016/j.robot.2022.104351
IF: 3.7
2023-01-01
Robotics and Autonomous Systems
Abstract:Reinforcement learning (RL) under temporal logic typically suffers from slow propagation for credit assignment. Inspired by recent advancements called trajectory transformer in machine learning, the reinforcement learning under Temporal Logic (TL) is modeled as a sequence modeling problem in this paper, where an agent utilizes the transformer to fit the optimal policy satisfying the Finite Linear Temporal Logic (LTLf) tasks. To combat the sparse reward issue, dense reward functions for LTLf are designed. For the sake of reducing the computational complexity, a sparse transformer with local and global attention is constructed to automatically conduct credit assignment, which removes the time-consuming value iteration process. The optimal action is found by the beam search performed in transformers. The proposed method generates a series of policies fitted by sparse transformers, which has sustainably high accuracy in fitting the demonstrations. At last, the effectiveness of the proposed method is demonstrated by simulations in Mini-Grid environments.(c) 2022 Elsevier B.V. All rights reserved.
What problem does this paper attempt to address?