RARE: Retrieval-Augmented Reasoning Enhancement for Large Language Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Tran, Hieu, Yao, Zonghai, Wang, Junda, Zhang, Yifan, Yang, Zhichao, Yu, Hong
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915318442491904
author Tran, Hieu
Yao, Zonghai
Wang, Junda
Zhang, Yifan
Yang, Zhichao
Yu, Hong
author_facet Tran, Hieu
Yao, Zonghai
Wang, Junda
Zhang, Yifan
Yang, Zhichao
Yu, Hong
contents This work introduces RARE (Retrieval-Augmented Reasoning Enhancement), a versatile extension to the mutual reasoning framework (rStar), aimed at enhancing reasoning accuracy and factual integrity across large language models (LLMs) for complex, knowledge-intensive tasks such as commonsense and medical reasoning. RARE incorporates two innovative actions within the Monte Carlo Tree Search (MCTS) framework: A6, which generates search queries based on the initial problem statement, performs information retrieval using those queries, and augments reasoning with the retrieved data to formulate the final answer; and A7, which leverages information retrieval specifically for generated sub-questions and re-answers these sub-questions with the relevant contextual information. Additionally, a Retrieval-Augmented Factuality Scorer is proposed to replace the original discriminator, prioritizing reasoning paths that meet high standards of factuality. Experimental results with LLaMA 3.1 show that RARE enables open-source LLMs to achieve competitive performance with top open-source models like GPT-4 and GPT-4o. This research establishes RARE as a scalable solution for improving LLMs in domains where logical coherence and factual integrity are critical.
format Preprint
id arxiv_https___arxiv_org_abs_2412_02830
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle RARE: Retrieval-Augmented Reasoning Enhancement for Large Language Models
Tran, Hieu
Yao, Zonghai
Wang, Junda
Zhang, Yifan
Yang, Zhichao
Yu, Hong
Computation and Language
This work introduces RARE (Retrieval-Augmented Reasoning Enhancement), a versatile extension to the mutual reasoning framework (rStar), aimed at enhancing reasoning accuracy and factual integrity across large language models (LLMs) for complex, knowledge-intensive tasks such as commonsense and medical reasoning. RARE incorporates two innovative actions within the Monte Carlo Tree Search (MCTS) framework: A6, which generates search queries based on the initial problem statement, performs information retrieval using those queries, and augments reasoning with the retrieved data to formulate the final answer; and A7, which leverages information retrieval specifically for generated sub-questions and re-answers these sub-questions with the relevant contextual information. Additionally, a Retrieval-Augmented Factuality Scorer is proposed to replace the original discriminator, prioritizing reasoning paths that meet high standards of factuality. Experimental results with LLaMA 3.1 show that RARE enables open-source LLMs to achieve competitive performance with top open-source models like GPT-4 and GPT-4o. This research establishes RARE as a scalable solution for improving LLMs in domains where logical coherence and factual integrity are critical.
title RARE: Retrieval-Augmented Reasoning Enhancement for Large Language Models
topic Computation and Language
url https://arxiv.org/abs/2412.02830