Research map / Large language models / Instruction and fine-tuning
Adapting While Learning: Grounding LLMs for Scientific Problems with Tool Usage Adaptation
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes Adapting While Learning (AWL), a two-stage fine-tuning method that internalizes tool-generated scientific knowledge (WKL) and trains the model to answer easy problems directly while switching to tools for hard ones (TUA), evaluated on six scientific benchmarks with Llama-3.1-8B-Instruct.
Problem
LLMs hallucinate on scientific problems, and tool-augmented fine-tuning makes them overreliant on tools, incurring unnecessary cost; they lack adaptive decisions about when to use tools.
Approach
Train the model on tool-generated solutions so knowledge is internalized (WKL), then partition questions into easy/hard by the model's direct-answer accuracy and keep direct-answer targets for easy questions while using tool-trace targets for hard ones (TUA), combined in a single mixed loss to avoid prompt inconsistency.
Key results
- AWL-trained Llama3.1-8B with intelligent tool use (P_i) reaches 66.13% average answer accuracy vs 56.16% for the base model with P_i, a reported 29.11% average improvement over the base model.
- Tool usage accuracy averages 63.93% for AWL vs 51.21% for the base model, a reported 12.72% improvement.
- On open-ended Epidemiology tasks, AWL-RL-P_i achieves 56.25% meeting requirements vs 43.75% for GPT-4o, and reduces tool calls from 2.80 to 0.42 per question.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Apply AWL with intelligent tool-use prompt instead of the base model prompt | Llama3.1-8B · Mujoco | answer accuracy (%): 59.32 -> 64.17 | Helped |
| Apply AWL with intelligent tool-use prompt instead of the base model prompt | Llama3.1-8B · PDE | answer accuracy (%): 61.67 -> 78.33 | Helped |
| Apply AWL with intelligent tool-use prompt instead of the base model prompt | Llama3.1-8B · Climate | answer accuracy (%): 77.50 -> 83.33 | Helped |
| Apply AWL with intelligent tool-use prompt instead of the base model prompt | Llama3.1-8B · Epidemiology | answer accuracy (%): 57.78 -> 74.44 | Helped |
| Apply AWL with intelligent tool-use prompt instead of the base model prompt | Llama3.1-8B · MATH | answer accuracy (%): 57.64 -> 62.35 | Helped |
| Apply AWL with intelligent tool-use prompt instead of the base model prompt | Llama3.1-8B · SciBench | answer accuracy (%): 31.67 -> 34.17 | Helped |
| Apply AWL instead of the base model | Llama3.1-8B · All datasets | tool usage accuracy (%): 51.21 -> 63.93 | Helped |
| Add preference learning (DPO) to AWL | AWL · Climate open-ended | proposals meeting quality threshold (%): 37.50 -> 49.16 | Helped |
| Add preference learning (DPO) to AWL | AWL · Epidemiology open-ended | proposals meeting quality threshold (%): 33.75 -> 56.25 | Helped |
| Remove WKL (ablation) | AWL · Climate and PDE | answer accuracy | Hurt |
| Remove TUA (ablation) | AWL · Climate and PDE | tool-use decision quality | Hurt |
| Use combined WKL and TUA training rather than alternating their losses (ablation) | AWL · easy problems | answer accuracy | Helped |
Surprising findings
- Naive alternation between WKL and TUA loss terms caused significant performance drops on easy problems, attributed to prompt inconsistency.
- The base model without training over-relies on tools (tool usage accuracy near 50%), while closed models such as GPT-4o and Claude-3.5 often answer directly on public datasets, possibly due to pretraining exposure.
Limitations
- Requires domain-specific fine-tuning; no unified training across scientific domains.
- Binary easy/hard partition ignores the spectrum of problem complexity.
- Remaining errors are dominated by agent limitations for multi-step tool interactions with the 8B base model.
Experimental setup
Datasets: MATH SciBench Mujoco (custom) PDE (custom) Climate (custom) Epidemiology (custom)
Benchmarks: MATH SciBench Mujoco PDE Climate Epidemiology
Metrics: answer accuracy tool usage accuracy tool usage rate proportion of proposals meeting threshold
Baselines: Llama3.1-70B GPT-4o GPT-4o-mini Claude-3.5-Sonnet Llama3.1-8B-Instruct (base) in-context learning tool-use baseline Qwen2.5-14B-Instruct
Builds on: direct preference optimization Llama-Factory
Closest papers by meaning
- Automated Creation of Reusable and Diverse Toolsets for Enhancing LLM Reasoning (AAAI 2025 · similarity 0.90)
- SciLitLLM: How to Adapt LLMs for Scientific Literature Understanding (ICLR 2025 · similarity 0.88)
- Large Language Models as Tool Makers (ICLR 2024 · similarity 0.88)
- SciInstruct: a Self-Reflective Instruction Annotated Dataset for Training Scientific Language Models (NeurIPS 2024 · similarity 0.87)
- From Exploration to Mastery: Enabling LLMs to Master Tools via Self-Driven Interactions (ICLR 2025 · similarity 0.87)
- SciBench: Evaluating College-Level Scientific Problem-Solving Abilities of Large Language Models (ICML 2024 · similarity 0.87)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
