A Fine-tuning Enhanced RAG System with Quantized Influence Measure as AI Judge

Keshav Rangan,Yiqiao Yin
2024-02-27
Abstract:This study presents an innovative enhancement to retrieval-augmented generation (RAG) systems by seamlessly integrating fine-tuned large language models (LLMs) with vector databases. This integration capitalizes on the combined strengths of structured data retrieval and the nuanced comprehension provided by advanced LLMs. Central to our approach are the LoRA and QLoRA methodologies, which stand at the forefront of model refinement through parameter-efficient fine-tuning and memory optimization. A novel feature of our research is the incorporation of user feedback directly into the training process, ensuring the model's continuous adaptation to user expectations and thus, improving its performance and applicability. Additionally, we introduce a Quantized Influence Measure (QIM) as an innovative "AI Judge" mechanism to enhance the precision of result selection, further refining the system's accuracy. Accompanied by an executive diagram and a detailed algorithm for fine-tuning QLoRA, our work provides a comprehensive framework for implementing these advancements within chatbot technologies. This research contributes significant insights into LLM optimization for specific uses and heralds new directions for further development in retrieval-augmented models. Through extensive experimentation and analysis, our findings lay a robust foundation for future advancements in chatbot technology and retrieval systems, marking a significant step forward in the creation of more sophisticated, precise, and user-centric conversational AI systems.
Information Retrieval
What problem does this paper attempt to address?
### Problems the Paper Aims to Solve The paper aims to address the increasingly severe issue of homeless families in the United States, particularly the lack of communication channels and limited critical resources in homeless shelters. Specifically, the paper proposes a web-based chatbot interface based on Large Language Models (LLM) and Generative Artificial Intelligence (GAI) technologies to significantly enhance the resources and communication channels for homeless shelters and their service recipients. ### Main Contributions 1. **Enhanced Retrieval-Augmented Generation (RAG) System**: - Improved communication efficiency by seamlessly integrating fine-tuned Large Language Models (LLM) with traditional vector databases. - Utilized LoRA and QLoRA methods for parameter-efficient fine-tuning and memory optimization. - Introduced Quantitative Impact Measurement (QIM) as an "AI Judge" mechanism to improve the accuracy and relevance of query result selection. 2. **Fine-Tuning Strategies**: - Detailed the fine-tuning methods of LoRA and QLoRA, including parameter adjustments and optimization processes. - Proposed a specific fine-tuning algorithm that optimizes model performance by iteratively adjusting attention dimensions, LoRA scaling factors, and dropout probabilities. 3. **Quantitative Impact Measurement (QIM)**: - Introduced a new quantitative impact measurement method to assess the similarity between queries and references. - Experimentally validated that QIM outperforms cosine similarity in high-dimensional data, especially when handling large datasets. ### Experiments and Applications - **Dataset**: Custom dataset collected from the Youth Spirit Artworks (YSA) Tiny House Empowerment Village website. - **Model**: Fine-tuned LLM model publicly available on Hugging Face. - **Application**: A Hugging Face-based chatbot application for practical testing and deployment. Through these innovations, the paper not only provides new technical directions for the AI community but also offers better access to information and resources for low-income and underserved populations.