Document Retrieval Augmented Fine-Tuning (DRAFT) for safety-critical software assessments
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.
Code (0)
등록된 구현이 없습니다.
Tasks
Dataset GenerationLanguage ModelingLanguage ModellingLarge Language ModelRAGRetrievalRetrieval-augmented GenerationSimilar Papers 제목 키워드 기반
LexDrafter: Terminology Drafting for Legislative Documents using Retrieval Augmented Generation
With the increase in legislative documents at the EU, the number of new terms and their definitions is increasing as well. As per the Joint Practical Guide of the European Parliament, the Council and the Commission, term…
ArticlesRAGRetrievalRetrieval-augmented GenerationSpeculative RAG: Enhancing Retrieval Augmented Generation through Drafting
Retrieval augmented generation (RAG) combines the generative abilities of large language models (LLMs) with external knowledge sources to provide more accurate and up-to-date responses. Recent RAG advancements focus on i…
ARCRAGRetrievalRetrieval-augmented Generation+1A Large Language Model-based Framework for Semi-Structured Tender Document Retrieval-Augmented Generation
The drafting of documents in the procurement field has progressively become more complex and diverse, driven by the need to meet legal requirements, adapt to technological advancements, and address stakeholder demands. W…
Language ModelingLanguage ModellingLarge Language ModelRetrieval+1Long-Context Inference with Retrieval-Augmented Speculative Decoding
The emergence of long-context large language models (LLMs) offers a promising alternative to traditional retrieval-augmented generation (RAG) for processing extensive documents. However, the computational overhead of lon…
Computational EfficiencyRAGRetrievalRetrieval-augmented Generation+1Enhancing Judgment Document Generation via Agentic Legal Information Collection and Rubric-Guided Optimization
Automating the drafting of judgment documents is pivotal to judicial efficiency, yet it remains challenging due to the dual requirements of comprehensive retrieval of legal information and rigorous logical reasoning. Exi…
Reinforcement LearningLogical ReasoningLegal Reasoning