# Symmetry-Aware Transformer Training for Automated Planning

Source: [Symmetry-Aware Transformer Training for Automated Planning](<https://www.sota2.com/research/paper/68fa7a821e209131cea8398c>)

Collection index: [llms.txt](<https://www.sota2.com/research/llms.txt>)

Data updated: 2026-06-30T05:00:08.405Z

Authors: Markus Fritzsche, Elliot Gestrin, Jendrik Seipp

Published: 2025-08-11T08:23:34.000Z

## Abstract

While transformers excel in many settings, their application in the field of automated planning is limited. Prior work like PlanGPT, a state-of-the-art decoder-only transformer, struggles with extrapolation from easy to hard planning problems. This in turn stems from problem symmetries: planning tasks can be represented with arbitrary variable names that carry no meaning beyond being identifiers. This causes a combinatorial explosion of equivalent representations that pure transformers cannot efficiently learn from. We propose a novel contrastive learning objective to make transformers symmetry-aware and thereby compensate for their lack of inductive bias. Combining this with architectural improvements, we show that transformers can be efficiently trained for either plan-generation or heuristic-prediction. Our results across multiple planning domains demonstrate that our symmetry-aware training effectively and efficiently addresses the limitations of PlanGPT.

## Recorded benchmark results

A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage.

- [Generalized Planning on Gripper (Extrapolation)](<https://www.sota2.com/research/sota/generalized-planning-on-gripper-extrapolation.md>): Generalized Planning; Gripper (Extrapolation); Coverage: 79.

- [Automated Planning on Gripper (val)](<https://www.sota2.com/research/sota/automated-planning-on-gripper-val.md>): Automated Planning; Gripper (val); QSS: 1.

- [Automated Planning on Blocks (interpolation)](<https://www.sota2.com/research/sota/automated-planning-on-blocks-interpolation.md>): Automated Planning; Blocks (Interpolation); Normalized Coverage Score: 100.

- [Plan Generation on Blocks 1.0 (val)](<https://www.sota2.com/research/sota/plan-generation-on-blocks-1-0-val.md>): Plan Generation; Blocks 1.0 (val); QS: 1.

- [Plan Generation on Blocks 1.0 (interpolation)](<https://www.sota2.com/research/sota/plan-generation-on-blocks-1-0-interpolation.md>): Plan Generation; Blocks 1.0 (interpolation); QS Score: 100.

- [Plan Generation on Blocks 1.0 (extrapolation)](<https://www.sota2.com/research/sota/plan-generation-on-blocks-1-0-extrapolation.md>): Plan Generation; Blocks 1.0 (extrapolation); QS: 0.13.

- [Plan Generation on Gripper 1.0 (val)](<https://www.sota2.com/research/sota/plan-generation-on-gripper-1-0-val.md>): Plan Generation; Gripper 1.0 (val); Quality Score (QS): 0.99.

- [Plan Generation on Gripper 1.0 (interpolation)](<https://www.sota2.com/research/sota/plan-generation-on-gripper-1-0-interpolation.md>): Plan Generation; Gripper 1.0 (interpolation); QS: 1.

- [Plan Generation on Gripper 1.0 (extrapolation)](<https://www.sota2.com/research/sota/plan-generation-on-gripper-1-0-extrapolation.md>): Plan Generation; Gripper 1.0 (extrapolation); QS: 0.77.

- [Plan Generation on Visitall 1.0 (val)](<https://www.sota2.com/research/sota/plan-generation-on-visitall-1-0-val.md>): Plan Generation; Visitall 1.0 (val); QS Score: 100.

- [Plan Generation on Visitall 1.0 (interpolation)](<https://www.sota2.com/research/sota/plan-generation-on-visitall-1-0-interpolation.md>): Plan Generation; Visitall 1.0 (interpolation); QS: 1.

- [Plan Generation on Visitall 1.0 (extrapolation)](<https://www.sota2.com/research/sota/plan-generation-on-visitall-1-0-extrapolation.md>): Plan Generation; Visitall 1.0 (extrapolation); QS: 60.

- [Automated Planning on Gripper (interpolation)](<https://www.sota2.com/research/sota/automated-planning-on-gripper-interpolation.md>): Automated Planning; Gripper (Interpolation); QSS: 1.

- [Automated Planning on Visitall (val)](<https://www.sota2.com/research/sota/automated-planning-on-visitall-val.md>): Automated Planning; VisitAll (val); QSS: 1.

- [Automated Planning on Visitall (extrapolation)](<https://www.sota2.com/research/sota/automated-planning-on-visitall-extrapolation.md>): Automated Planning; VisitAll (Extrapolation); QSS: 98.

- [Automated Planning on Blocks (val)](<https://www.sota2.com/research/sota/automated-planning-on-blocks-val.md>): Automated Planning; Blocks (val); QSS: 1.

- [Automated Planning on Blocks (extrapolation)](<https://www.sota2.com/research/sota/automated-planning-on-blocks-extrapolation.md>): Automated Planning; Blocks (Extrapolation); QSS: 1.

- [Automated Planning on Gripper (extrapolation)](<https://www.sota2.com/research/sota/automated-planning-on-gripper-extrapolation.md>): Automated Planning; Gripper (Extrapolation); QSS: 97.

- [Generalized Planning on Gripper (val)](<https://www.sota2.com/research/sota/generalized-planning-on-gripper-val.md>): Generalized Planning; Gripper (val); Coverage: 100.

- [Generalized Planning on Gripper (Interpolation)](<https://www.sota2.com/research/sota/generalized-planning-on-gripper-interpolation.md>): Generalized Planning; Gripper (Interpolation); Coverage: 100.

- [Generalized Planning on VisitAll (val)](<https://www.sota2.com/research/sota/generalized-planning-on-visitall-val.md>): Generalized Planning; VisitAll (val); Coverage: 100.

- [Generalized Planning on VisitAll (Interpolation)](<https://www.sota2.com/research/sota/generalized-planning-on-visitall-interpolation.md>): Generalized Planning; VisitAll (Interpolation); Coverage: 100.

- [Generalized Planning on VisitAll (Extrapolation)](<https://www.sota2.com/research/sota/generalized-planning-on-visitall-extrapolation.md>): Generalized Planning; VisitAll (Extrapolation); Coverage: 64.

- [Automated Planning on Visitall (interpolation)](<https://www.sota2.com/research/sota/automated-planning-on-visitall-interpolation.md>): Automated Planning; VisitAll (Interpolation); QSS: 100.

- [Plan Generation on Logistics 1.0 (interpolation)](<https://www.sota2.com/research/sota/plan-generation-on-logistics-1-0-interpolation.md>): Plan Generation; Logistics 1.0 (interpolation); QS Score: 24.

- [Generalized Planning on Blocks (val)](<https://www.sota2.com/research/sota/generalized-planning-on-blocks-val.md>): Generalized Planning; Blocks (val); Coverage: 100.

- [Generalized Planning on Blocks (Interpolation)](<https://www.sota2.com/research/sota/generalized-planning-on-blocks-interpolation.md>): Generalized Planning; Blocks (Interpolation); Coverage: 100.

- [Automated Planning on Logistics (interpolation)](<https://www.sota2.com/research/sota/automated-planning-on-logistics-interpolation.md>): Automated Planning; Logistics (Interpolation); QSS: 100.

- [Plan Generation on Logistics 1.0 (val)](<https://www.sota2.com/research/sota/plan-generation-on-logistics-1-0-val.md>): Plan Generation; Logistics 1.0 (val); Quality Score (QS): 0.

- [Plan Generation on Logistics 1.0 (extrapolation)](<https://www.sota2.com/research/sota/plan-generation-on-logistics-1-0-extrapolation.md>): Plan Generation; Logistics 1.0 (extrapolation); QS: 0.

- [Generalized Planning on Blocks (Extrapolation)](<https://www.sota2.com/research/sota/generalized-planning-on-blocks-extrapolation.md>): Generalized Planning; Blocks (Extrapolation); Coverage: 13.

- [Generalized Planning on Logistics (Interpolation)](<https://www.sota2.com/research/sota/generalized-planning-on-logistics-interpolation.md>): Generalized Planning; Logistics (Interpolation); Coverage: 26.

- [Generalized Planning on Logistics (Extrapolation)](<https://www.sota2.com/research/sota/generalized-planning-on-logistics-extrapolation.md>): Generalized Planning; Logistics (Extrapolation); Coverage: 0.

- [Generalized Planning on Logistics (val)](<https://www.sota2.com/research/sota/generalized-planning-on-logistics-val.md>): Generalized Planning; Logistics (val); Coverage: 0.

- [Automated Planning on Logistics (val)](<https://www.sota2.com/research/sota/automated-planning-on-logistics-val.md>): Automated Planning; Logistics (val); QSS: not recorded.

## Original sources and implementation

[Original paper](<https://arxiv.org/abs/2508.07743>)

[PDF](<https://arxiv.org/pdf/2508.07743.pdf>)
