Benchmarking Fine-tuning and Retrieval Strategies for a Multimodal Language Model on the NRC Reactor Operator Licensing Examination

· Source: cs.CL updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

A study evaluated the 31-billion-parameter open-weight multimodal model Gemma 4 31B-IT on its ability to apply nuclear knowledge, using the U.S. Nuclear Regulatory Commission (NRC) Reactor Operator licensing examination. Researchers benchmarked eight model-retrieval configurations, including supervised fine-tuning (SFT) on Gemini-distilled chain-of-thought rationales, retrieval-augmented generation (RAG) with BM25 sparse retrieval, and retrieval-augmented fine-tuning (RAFT). The evaluation used 14 Generic Fundamentals Examinations from 2015-2021, applying an 80% human passing criterion. The SFT configuration combined with fixed-size chunking RAG met the passing criterion on 8 of the 14 examinations, achieving an aggregate accuracy of 79.7% and 80.2% on PWR items. Notably, no configuration without fine-tuning passed any exam. The study also found that the preferred chunking strategy depends on the model's training state, and RAFT underperformed standard SFT in search environments.

Key takeaway

For AI Scientists and Machine Learning Engineers developing LLMs for high-stakes technical domains like nuclear power, you should prioritize supervised fine-tuning (SFT) combined with retrieval-augmented generation (RAG) using fixed-size chunking. This approach enabled Gemma 4 31B-IT to meet the 80% passing criterion on 8 of 14 NRC Reactor Operator exams, demonstrating operator-level capabilities. Avoid relying solely on RAG without fine-tuning, as it failed to pass any examinations.

Key insights

Fine-tuning is crucial for multimodal LLMs to achieve domain-specific, operator-level capabilities in nuclear knowledge applications.

Principles

Method

The study benchmarked Gemma 4 31B-IT using SFT, RAG with BM25 sparse retrieval, and RAFT, evaluating performance on 14 NRC Reactor Operator licensing examinations with an 80% passing threshold.

In practice

Topics

Best for: AI Engineer, AI Scientist, Machine Learning Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.