Date
Publisher
arXiv
Large Language Models (LLMs) hold significant promise for mathematics education, yet they often struggle with complex mathematical reasoning. While Retrieval-Augmented Generation (RAG) mitigates these issues by grounding LLMs in external knowledge, its effectiveness remains unstable, heavily dependent on the choice of a single embedding model. Moving beyond static RAG workflows, we draw on agentic workflow patterns, a paradigm that introduces structured task decomposition and collaboration to enhance system performance. We propose and examine two novel approaches that combine the benefits of multiple embedding models. While our Mixture-Embedding RAG approach (fusing retrieved documents) shows limited gains, our Confident RAG method (generating multiple answers and selecting the one with the highest confidence score) demonstrates significant improvement. Experimental results show that Confident RAG achieved average accuracy improvements of approximately 10% over vanilla LLMs and 5% over vanilla RAG. The consistent results across different LLMs and embedding models indicate that Confident RAG is an efficient plug-and-play solution for trustworthy mathematical AI assistants. Finally, we discuss how this work lays the groundwork for deploying Agentic RAG systems in educational settings, where autonomous planning and iterative refinement can be built upon our robust retrieval foundation.
Study at a glance
- Question: Can combining multiple embedding models with confidence-based answer selection improve language models' accuracy on math questions?
- Setting and data: First 500 test questions and training corpus from GSM8K math dataset, combined with OpenStax math textbooks.
- Method: Benchmark evaluation comparing standard language models, standard retrieval-augmented generation, and two multi-embedding retrieval-augmented generation methods across multiple confidence scoring metrics.
Why it matters
Large language models frequently hallucinate in mathematics education where precision is crucial. While retrieval-augmented generation grounds answers in external knowledge, its performance remains unstable depending heavily on the single embedding model selected.
Key findings
- Confident RAG improved average accuracy by 9.9% over vanilla language models and by 4.9% over vanilla retrieval-augmented generation when using the best confidence metric.
- Self-certainty and distributional perplexity were the most effective confidence metrics, achieving average accuracy gains of 9.9% and 9.7% over vanilla language models.
- Mixture-Embedding RAG showed limited improvement over vanilla retrieval-augmented generation and actually decreased accuracy by 5.5% on the specialized Qwen2.5-Math-7B model.
Summary drafted with AI from the full paper and may contain errors. See the full paper for details.
What is the application?
Who is the user?
Who age?
Why use AI?
Study design
