mlsanityData to 7 Oct 2026

LLM-Guided Strategy Synthesis for Scalable Equality Saturation

arXiv 2604.17364 ↗First posted 19 Apr 2026cs.AI cs.MA cs.PLcited by 0

Not on the map. Why?

Abstract (arXiv)

Equality saturation (EqSat) is a powerful optimization paradigm that compactly represents many equivalent programs in an e-graph and delays commitment until extraction selects a lowest-cost program. Making EqSat effective, therefore, requires not only domain-specific rewrite rules but also domain-specific strategies. Today, much of this strategy design is still manual, making it a major obstacle to automating e-graph-based compilers. Recent rule-synthesis frameworks can automatically infer large rewrite vocabularies from semantic specifications, but they also enlarge the rewrite space and further exacerbate e-graph explosion. Although large language models (LLMs) make automated strategy synthesis plausible, directly evolving backend code remains ineffective in practice. The search lacks reusable strategy abstractions and actionable feedback, and can easily trigger e-graph explosion or converge to poor designs. We present EggMind, an LLM-guided, end-to-end framework for synthesizing reusable EqSat strategies. At its core, EggMind introduces a domain-specific language, EqSatL, to represent EqSat strategies as explicit and inspectable artifacts. It then proposes an LLM-guided agentic workflow, equipped with novel techniques including proof-derived rewrite motif caching and tractability guidance, to search efficiently for high-quality strategies while keeping synthesis stable under e-graph growth. Evaluation shows that EggMind substantially improves the resource-quality trade-off on vectorization benchmarks, reducing final cost by 45.1% and peak RAM by 69.1% relative to full EqSat. We further show that the same methodology transfers effectively to an XLA-based tensor compiler, and demonstrate its practical potential in a logic-synthesis case study with augmented rewrite spaces.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 18 of its 42 references that are in the map. First posted on arXiv 19 Apr 2026.

Builds on

18 of 42 references.

References · 18

  1. 2024Meta Large Language Model Compiler: Foundation Models of Compiler Optimizationcited by 39
  2. 2021Equality Saturation for Tensor Graph Superoptimizationcited by 27
  3. 2022babble: Learning Better Abstractions with E-Graphs and Anti-Unificationcited by 22
  4. 2021Rewrite Rule Inference Using Equality Saturationcited by 10
  5. 2023MCTS-GEB: Monte Carlo Tree Search is a Good E-graph Buildercited by 5
  6. 2026Magellan: Autonomous Discovery of Novel Compiler Optimization Heuristics with AlphaEvolvecited by 5
  7. 2022Small Proofs from Congruence Closurecited by 3
  8. 2026Agentic Code Optimization via Compiler-LLM Cooperationcited by 3
  9. 2023Implementation and Synthesis of Math Library Functionscited by 2
  10. 2024Target-Aware Implementation of Real Expressionscited by 2
  11. 2025e-boost: Boosted E-Graph Extraction with Adaptive Heuristics and Exact Solvingcited by 2
  12. 2025TensorRight: Automated Verification of Tensor Graph Rewritescited by 2
6 more
  1. 2025Pushing Tensor Accelerators Beyond MatMul in a User-Schedulable Languagecited by 2
  2. 2026Beyond Pass-by-Pass Optimization: Intent-Driven IR Optimization with Large Language Modelscited by 2
  3. 2021Sketch-Guided Equality Saturation: Scaling Equality Saturation to Complex Optimizations of Functional Programscited by 1
  4. 2024The Maude strategy languagecited by 1
  5. 2025SkyEgg: Heterogeneity-Aware Hardware Synthesis via Equality Saturationcited by 1
  6. 2026E-Graphs as a Persistent Compiler Abstractioncited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025ORGEval: Graph-Theoretic Evaluation of LLMs in Optimization Modeling
  2. 2026EffiSkill: Agent Skill Based Automated Code Efficiency Optimization
  3. 2025Autonomous Code Evolution Meets NP-Completeness
  4. 2026Rethinking LLM-Driven Heuristic Design: Generating Efficient and Specialized Solvers via Dynamics-Aware Optimization
  5. 2026SKILL: Self-correcting Knowledge-guided Iterative Large Language Model Agent for Logic Optimization
  6. 2026ECCO: Evidence-Driven Causal Reasoning for Compiler Optimization
  7. 2026SynAct: A Reasoning-Acting Large Language Model Agent for Adaptive Synthesis Optimization
  8. 2026From Agent Loops to Structured Graphs:A Scheduler-Theoretic Framework for LLM Agent Execution