Adapting While Learning: Grounding LLMs for Scientific Problems with Intelligent Tool Usage Adaptation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lyu, Bohan, Cao, Yadi, Watson-Parris, Duncan, Bergen, Leon, Berg-Kirkpatrick, Taylor, Yu, Rose
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916801568309248
author Lyu, Bohan
Cao, Yadi
Watson-Parris, Duncan
Bergen, Leon
Berg-Kirkpatrick, Taylor
Yu, Rose
author_facet Lyu, Bohan
Cao, Yadi
Watson-Parris, Duncan
Bergen, Leon
Berg-Kirkpatrick, Taylor
Yu, Rose
contents Large Language Models (LLMs) demonstrate promising capabilities in solving scientific problems but often suffer from the issue of hallucination. While integrating LLMs with tools can mitigate this issue, models fine-tuned on tool usage become overreliant on them and incur unnecessary costs. Inspired by how human experts assess problem complexity before selecting solutions, we propose a novel two-component fine-tuning method, Adapting While Learning (AWL). In the first component, World Knowledge Learning (WKL), LLMs internalize scientific knowledge by learning from tool-generated solutions. In the second component, Tool Usage Adaptation (TUA), we categorize problems as easy or hard based on the model's accuracy, and train it to maintain direct reasoning for easy problems while switching to tools for hard ones. We validate our method on six scientific benchmark datasets across climate science, epidemiology, physics, and other domains. Compared to the original instruct model (8B), models post-trained with AWL achieve 29.11% higher answer accuracy and 12.72% better tool usage accuracy, even surpassing state-of-the-art models including GPT-4o and Claude-3.5 on four custom-created datasets. Our code is open-source at https://github.com/Rose-STL-Lab/Adapting-While-Learning.
format Preprint
id arxiv_https___arxiv_org_abs_2411_00412
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Adapting While Learning: Grounding LLMs for Scientific Problems with Intelligent Tool Usage Adaptation
Lyu, Bohan
Cao, Yadi
Watson-Parris, Duncan
Bergen, Leon
Berg-Kirkpatrick, Taylor
Yu, Rose
Machine Learning
Artificial Intelligence
Computation and Language
I.2.6; I.2.7
Large Language Models (LLMs) demonstrate promising capabilities in solving scientific problems but often suffer from the issue of hallucination. While integrating LLMs with tools can mitigate this issue, models fine-tuned on tool usage become overreliant on them and incur unnecessary costs. Inspired by how human experts assess problem complexity before selecting solutions, we propose a novel two-component fine-tuning method, Adapting While Learning (AWL). In the first component, World Knowledge Learning (WKL), LLMs internalize scientific knowledge by learning from tool-generated solutions. In the second component, Tool Usage Adaptation (TUA), we categorize problems as easy or hard based on the model's accuracy, and train it to maintain direct reasoning for easy problems while switching to tools for hard ones. We validate our method on six scientific benchmark datasets across climate science, epidemiology, physics, and other domains. Compared to the original instruct model (8B), models post-trained with AWL achieve 29.11% higher answer accuracy and 12.72% better tool usage accuracy, even surpassing state-of-the-art models including GPT-4o and Claude-3.5 on four custom-created datasets. Our code is open-source at https://github.com/Rose-STL-Lab/Adapting-While-Learning.
title Adapting While Learning: Grounding LLMs for Scientific Problems with Intelligent Tool Usage Adaptation
topic Machine Learning
Artificial Intelligence
Computation and Language
I.2.6; I.2.7
url https://arxiv.org/abs/2411.00412