Data Quality over Capacity: Internalizing Documents into LoRA Adapters for Closed-Book QA

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Advanced, quick

Summary

A study investigates internalizing documents directly into the weights of a 4-bit Gemma-4-e4b model using LoRA adapters to enable closed-book question answering without retrieval or context window budget. Across approximately 100 training runs, ranging from single documents to a 99-document corpus, researchers found that once adapter capacity is sufficient, training-data quality becomes the primary factor influencing closed-book accuracy, surpassing LoRA rank, learning rate, and alternative architectures. A single data curation pass, involving shortening gold answers to 1-6 word spans and removing trivia, significantly improved accuracy from 57.7% to 85.7% on a 15-document corpus. The internalized adapter achieved 84.2% recall, outperforming a BM25-RAG pipeline (58.9%) and a gold-chunk oracle (65.6%) at lower latency. The research also confirms that LoRA rank must increase with corpus size, noting a complex interaction between rank and learning rate. The full debugging process, including three initial misdiagnoses, is presented as an empirical case study.

Key takeaway

For machine learning engineers developing closed-book QA systems with LoRA adapters, prioritize rigorous data curation over extensive architectural experimentation. Your efforts in shortening gold answers to canonical spans and removing trivia can yield significantly higher accuracy, as demonstrated by an 85.7% accuracy with curation versus 57.7% without. Ensure your adapter capacity, specifically LoRA rank, scales appropriately with corpus size, as this forms a critical performance baseline before data quality improvements can take effect.

Key insights

For closed-book QA with LoRA, data quality is the dominant performance lever once adapter capacity is sufficient.

Principles

Method

Documents are internalized into 4-bit Gemma-4-e4b LoRA adapters for closed-book QA. Curation involves shortening gold answers to 1-6 words and dropping trivia.

In practice

Topics

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

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.