Document Retrieval Augmented Fine-Tuning (DRAFT) for safety-critical software assessments
This addresses the challenge of automating regulatory compliance assessments for safety-critical software, though it appears incremental as it builds on existing RAG techniques.
The paper tackled the problem of safety-critical software compliance assessment by introducing Document Retrieval-Augmented Fine-Tuning (DRAFT), which improved correctness by 7% over a baseline model and enhanced evidence handling and reasoning.
Safety critical software assessment requires robust assessment against complex regulatory frameworks, a process traditionally limited by manual evaluation. This paper presents Document Retrieval-Augmented Fine-Tuning (DRAFT), a novel approach that enhances the capabilities of a large language model (LLM) for safety-critical compliance assessment. DRAFT builds upon existing Retrieval-Augmented Generation (RAG) techniques by introducing a novel fine-tuning framework that accommodates our dual-retrieval architecture, which simultaneously accesses both software documentation and applicable reference standards. To fine-tune DRAFT, we develop a semi-automated dataset generation methodology that incorporates variable numbers of relevant documents with meaningful distractors, closely mirroring real-world assessment scenarios. Experiments with GPT-4o-mini demonstrate a 7% improvement in correctness over the baseline model, with qualitative improvements in evidence handling, response structure, and domain-specific reasoning. DRAFT represents a practical approach to improving compliance assessment systems while maintaining the transparency and evidence-based reasoning essential in regulatory domains.