Enhancing LLMs for Physics Problem-Solving using Reinforcement Learning with Human-AI Feedback

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Anand, Avinash, Prasad, Kritarth, Kirtani, Chhavi, Nair, Ashwin R, Gupta, Mohit, Garg, Saloni, Gautam, Anurag, Buldeo, Snehal, Shah, Rajiv Ratn
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915055305490432
author Anand, Avinash
Prasad, Kritarth
Kirtani, Chhavi
Nair, Ashwin R
Gupta, Mohit
Garg, Saloni
Gautam, Anurag
Buldeo, Snehal
Shah, Rajiv Ratn
author_facet Anand, Avinash
Prasad, Kritarth
Kirtani, Chhavi
Nair, Ashwin R
Gupta, Mohit
Garg, Saloni
Gautam, Anurag
Buldeo, Snehal
Shah, Rajiv Ratn
contents Large Language Models (LLMs) have demonstrated strong capabilities in text-based tasks but struggle with the complex reasoning required for physics problems, particularly in advanced arithmetic and conceptual understanding. While some research has explored ways to enhance LLMs in physics education using techniques such as prompt engineering and Retrieval Augmentation Generation (RAG), not enough effort has been made in addressing their limitations in physics reasoning. This paper presents a novel approach to improving LLM performance on physics questions using Reinforcement Learning with Human and Artificial Intelligence Feedback (RLHAIF). We evaluate several reinforcement learning methods, including Proximal Policy Optimization (PPO), Direct Preference Optimization (DPO), and Remax optimization. These methods are chosen to investigate RL policy performance with different settings on the PhyQA dataset, which includes challenging physics problems from high school textbooks. Our RLHAIF model, tested on leading LLMs like LLaMA2 and Mistral, achieved superior results, notably with the MISTRAL-PPO model, demonstrating marked improvements in reasoning and accuracy. It achieved high scores, with a 58.67 METEOR score and a 0.74 Reasoning score, making it a strong example for future physics reasoning research in this area.
format Preprint
id arxiv_https___arxiv_org_abs_2412_06827
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Enhancing LLMs for Physics Problem-Solving using Reinforcement Learning with Human-AI Feedback
Anand, Avinash
Prasad, Kritarth
Kirtani, Chhavi
Nair, Ashwin R
Gupta, Mohit
Garg, Saloni
Gautam, Anurag
Buldeo, Snehal
Shah, Rajiv Ratn
Machine Learning
Artificial Intelligence
Large Language Models (LLMs) have demonstrated strong capabilities in text-based tasks but struggle with the complex reasoning required for physics problems, particularly in advanced arithmetic and conceptual understanding. While some research has explored ways to enhance LLMs in physics education using techniques such as prompt engineering and Retrieval Augmentation Generation (RAG), not enough effort has been made in addressing their limitations in physics reasoning. This paper presents a novel approach to improving LLM performance on physics questions using Reinforcement Learning with Human and Artificial Intelligence Feedback (RLHAIF). We evaluate several reinforcement learning methods, including Proximal Policy Optimization (PPO), Direct Preference Optimization (DPO), and Remax optimization. These methods are chosen to investigate RL policy performance with different settings on the PhyQA dataset, which includes challenging physics problems from high school textbooks. Our RLHAIF model, tested on leading LLMs like LLaMA2 and Mistral, achieved superior results, notably with the MISTRAL-PPO model, demonstrating marked improvements in reasoning and accuracy. It achieved high scores, with a 58.67 METEOR score and a 0.74 Reasoning score, making it a strong example for future physics reasoning research in this area.
title Enhancing LLMs for Physics Problem-Solving using Reinforcement Learning with Human-AI Feedback
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2412.06827