Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Ruan, Jingqing, Wang, Kaishen, Zhang, Qingyang, Xing, Dengpeng, Xu, Bo
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:https://arxiv.org/abs/2312.11027
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913355517657088
author Ruan, Jingqing
Wang, Kaishen
Zhang, Qingyang
Xing, Dengpeng
Xu, Bo
author_facet Ruan, Jingqing
Wang, Kaishen
Zhang, Qingyang
Xing, Dengpeng
Xu, Bo
contents Many complicated real-world tasks can be broken down into smaller, more manageable parts, and planning with prior knowledge extracted from these simplified pieces is crucial for humans to make accurate decisions. However, replicating this process remains a challenge for AI agents and naturally raises two questions: How to extract discriminative knowledge representation from priors? How to develop a rational plan to decompose complex problems? Most existing representation learning methods employing a single encoder structure are fragile and sensitive to complex and diverse dynamics. To address this issue, we introduce a multiple-encoder and individual-predictor regime to learn task-essential representations from sufficient data for simple subtasks. Multiple encoders can extract adequate task-relevant dynamics without confusion, and the shared predictor can discriminate the task characteristics. We also use the attention mechanism to generate a top-k subtask planning tree, which customizes subtask execution plans in guiding complex decisions on unseen tasks. This process enables forward-looking and globality by flexibly adjusting the depth and width of the planning tree. Empirical results on a challenging platform composed of some basic simple tasks and combinatorially rich synthetic tasks consistently outperform some competitive baselines and demonstrate the benefits of our design.
format Preprint
id arxiv_https___arxiv_org_abs_2312_11027
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Learning Top-k Subtask Planning Tree based on Discriminative Representation Pre-training for Decision Making
Ruan, Jingqing
Wang, Kaishen
Zhang, Qingyang
Xing, Dengpeng
Xu, Bo
Artificial Intelligence
Many complicated real-world tasks can be broken down into smaller, more manageable parts, and planning with prior knowledge extracted from these simplified pieces is crucial for humans to make accurate decisions. However, replicating this process remains a challenge for AI agents and naturally raises two questions: How to extract discriminative knowledge representation from priors? How to develop a rational plan to decompose complex problems? Most existing representation learning methods employing a single encoder structure are fragile and sensitive to complex and diverse dynamics. To address this issue, we introduce a multiple-encoder and individual-predictor regime to learn task-essential representations from sufficient data for simple subtasks. Multiple encoders can extract adequate task-relevant dynamics without confusion, and the shared predictor can discriminate the task characteristics. We also use the attention mechanism to generate a top-k subtask planning tree, which customizes subtask execution plans in guiding complex decisions on unseen tasks. This process enables forward-looking and globality by flexibly adjusting the depth and width of the planning tree. Empirical results on a challenging platform composed of some basic simple tasks and combinatorially rich synthetic tasks consistently outperform some competitive baselines and demonstrate the benefits of our design.
title Learning Top-k Subtask Planning Tree based on Discriminative Representation Pre-training for Decision Making
topic Artificial Intelligence
url https://arxiv.org/abs/2312.11027