Graph Neural Network Based Action Ranking for Planning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Mangannavar, Rajesh, Lee, Stefan, Fern, Alan, Tadepalli, Prasad
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911228590292992
author Mangannavar, Rajesh
Lee, Stefan
Fern, Alan
Tadepalli, Prasad
author_facet Mangannavar, Rajesh
Lee, Stefan
Fern, Alan
Tadepalli, Prasad
contents We propose a novel approach to learn relational policies for classical planning based on learning to rank actions. We introduce a new graph representation that explicitly captures action information and propose a Graph Neural Network (GNN) architecture augmented with Gated Recurrent Units (GRUs) to learn action rankings. Unlike value-function based approaches that must learn a globally consistent function, our action ranking method only needs to learn locally consistent ranking. Our model is trained on data generated from small problem instances that are easily solved by planners and is applied to significantly larger instances where planning is computationally prohibitive. Experimental results across standard planning benchmarks demonstrate that our action-ranking approach not only achieves better generalization to larger problems than those used in training but also outperforms multiple baselines (value function and action ranking) methods in terms of success rate and plan quality.
format Preprint
id arxiv_https___arxiv_org_abs_2412_04752
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Graph Neural Network Based Action Ranking for Planning
Mangannavar, Rajesh
Lee, Stefan
Fern, Alan
Tadepalli, Prasad
Machine Learning
I.2.6; I.2.9
We propose a novel approach to learn relational policies for classical planning based on learning to rank actions. We introduce a new graph representation that explicitly captures action information and propose a Graph Neural Network (GNN) architecture augmented with Gated Recurrent Units (GRUs) to learn action rankings. Unlike value-function based approaches that must learn a globally consistent function, our action ranking method only needs to learn locally consistent ranking. Our model is trained on data generated from small problem instances that are easily solved by planners and is applied to significantly larger instances where planning is computationally prohibitive. Experimental results across standard planning benchmarks demonstrate that our action-ranking approach not only achieves better generalization to larger problems than those used in training but also outperforms multiple baselines (value function and action ranking) methods in terms of success rate and plan quality.
title Graph Neural Network Based Action Ranking for Planning
topic Machine Learning
I.2.6; I.2.9
url https://arxiv.org/abs/2412.04752