AlphaIntegrator: Transformer Action Search for Symbolic Integration Proofs

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Ünsal, Mert, Gehr, Timon, Vechev, Martin
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912056919195648
author Ünsal, Mert
Gehr, Timon
Vechev, Martin
author_facet Ünsal, Mert
Gehr, Timon
Vechev, Martin
contents We present the first correct-by-construction learning-based system for step-by-step mathematical integration. The key idea is to learn a policy, represented by a GPT transformer model, which guides the search for the right mathematical integration rule, to be carried out by a symbolic solver. Concretely, we introduce a symbolic engine with axiomatically correct actions on mathematical expressions, as well as the first dataset for step-by-step integration. Our GPT-style transformer model, trained on this synthetic data, demonstrates strong generalization by surpassing its own data generator in accuracy and efficiency, using 50% fewer search steps. Our experimental results with SoTA LLMs also demonstrate that the standard approach of fine-tuning LLMs on a set of question-answer pairs is insufficient for solving this mathematical task. This motivates the importance of discovering creative methods for combining LLMs with symbolic reasoning engines, of which our work is an instance.
format Preprint
id arxiv_https___arxiv_org_abs_2410_02666
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle AlphaIntegrator: Transformer Action Search for Symbolic Integration Proofs
Ünsal, Mert
Gehr, Timon
Vechev, Martin
Machine Learning
Artificial Intelligence
Symbolic Computation
We present the first correct-by-construction learning-based system for step-by-step mathematical integration. The key idea is to learn a policy, represented by a GPT transformer model, which guides the search for the right mathematical integration rule, to be carried out by a symbolic solver. Concretely, we introduce a symbolic engine with axiomatically correct actions on mathematical expressions, as well as the first dataset for step-by-step integration. Our GPT-style transformer model, trained on this synthetic data, demonstrates strong generalization by surpassing its own data generator in accuracy and efficiency, using 50% fewer search steps. Our experimental results with SoTA LLMs also demonstrate that the standard approach of fine-tuning LLMs on a set of question-answer pairs is insufficient for solving this mathematical task. This motivates the importance of discovering creative methods for combining LLMs with symbolic reasoning engines, of which our work is an instance.
title AlphaIntegrator: Transformer Action Search for Symbolic Integration Proofs
topic Machine Learning
Artificial Intelligence
Symbolic Computation
url https://arxiv.org/abs/2410.02666