Saved in:
Bibliographic Details
Main Authors: Yang, Tianbo, Yan, Mingqi, Zhao, Hongyi, Yang, Tianshuo
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2501.15797
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913684497891328
author Yang, Tianbo
Yan, Mingqi
Zhao, Hongyi
Yang, Tianshuo
author_facet Yang, Tianbo
Yan, Mingqi
Zhao, Hongyi
Yang, Tianshuo
contents Developing the logic necessary to solve mathematical problems or write mathematical proofs is one of the more difficult objectives for large language models (LLMS). Currently, the most popular methods in literature consists of fine-tuning the model on written mathematical content such as academic publications and textbooks, so that the model can learn to emulate the style of mathematical writing. In this project, we explore the effectiveness of using retrieval augmented generation (RAG) to address gaps in the mathematical reasoning of LLMs. We develop LemmaHead, a RAG knowledge base that supplements queries to the model with relevant mathematical context, with particular focus on context from published textbooks. To measure our model's performance in mathematical reasoning, our testing paradigm focuses on the task of automated theorem proving via generating proofs to a given mathematical claim in the Lean formal language.
format Preprint
id arxiv_https___arxiv_org_abs_2501_15797
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle LemmaHead: RAG Assisted Proof Generation Using Large Language Models
Yang, Tianbo
Yan, Mingqi
Zhao, Hongyi
Yang, Tianshuo
Machine Learning
Computation and Language
Information Retrieval
Developing the logic necessary to solve mathematical problems or write mathematical proofs is one of the more difficult objectives for large language models (LLMS). Currently, the most popular methods in literature consists of fine-tuning the model on written mathematical content such as academic publications and textbooks, so that the model can learn to emulate the style of mathematical writing. In this project, we explore the effectiveness of using retrieval augmented generation (RAG) to address gaps in the mathematical reasoning of LLMs. We develop LemmaHead, a RAG knowledge base that supplements queries to the model with relevant mathematical context, with particular focus on context from published textbooks. To measure our model's performance in mathematical reasoning, our testing paradigm focuses on the task of automated theorem proving via generating proofs to a given mathematical claim in the Lean formal language.
title LemmaHead: RAG Assisted Proof Generation Using Large Language Models
topic Machine Learning
Computation and Language
Information Retrieval
url https://arxiv.org/abs/2501.15797