Skip to Content
ProcessesProcesses
  • Article
  • Open Access

15 February 2026

Virtual-Document-Augmented Retrieval-Augmented Generation for Power-Domain Knowledge-Base Question Answering with Noise-Enhanced Robustness

,
,
,
,
,
and
1
Beijing Key Laboratory of Demand Side Multi-Energy Carriers Optimization and Interaction Technique, China Electric Power Research Institute, Beijing 100192, China
2
School of Computer and Communication Engineering, University of Science and Technology Beijing, Beijing 100083, China
3
Shunde Innovation School, University of Science and Technology Beijing, Foshan 528399, China
*
Author to whom correspondence should be addressed.
This article belongs to the Section AI-Enabled Process Engineering

Abstract

The Retrieval-Augmented Generation (RAG) technology enables large language models (LLM) to access external knowledge bases by introducing external documents, enhancing their capability for knowledge question answering in professional domains and generating more reliable responses. It effectively addresses issues such as LLM hallucinations and knowledge obsolescence. In the electric power domain, RAG technology can be leveraged to fully utilize accumulated corporate data and resources. However, in the retrieval phase of RAG, there are significant differences in the semantic space representation between short sentences and long text documents. Additionally, when generating answers based on retrieved relevant documents, the generator prioritizes highly relevant document fragments, a strategy that may overlook sub-relevant documents containing useful information. This paper uses an LLM to generate hypothetical documents. These documents are combined with the original question to perform similarity retrieval in the corpus, followed by the first round of answer generation. Subsequently, the original question is combined with the answer generated in the first round, and this combined content is used to retrieve relevant documents. Finally, irrelevant documents are added to the context of the retrieved relevant documents to enhance the LLM’s attention to the relevant documents. Based on the above strategies, experiments are conducted on the electricity dataset. The results show that, compared with the naive RAG method, the proposed model achieves a relative improvement of 4.63% in the ROUGE-L metric and 11.32% in the BLEU-4 metric on the electricity dataset. Meanwhile, experiments are also carried out on the public CMRC dataset, and the effectiveness of the proposed method is verified.

1. Introduction

With the in-depth integration of machine learning technologies and complex system theories, Large Language Models (LLM) have demonstrated potential application value in scenarios involving intelligent analysis and decision-making for power systems. Large language models such as the GPT3 model [1] and the Llama2 model [2] have demonstrated impressive language and knowledge mastery capabilities. However, their ability to adapt to highly specialized complex systems remains significantly limited. In complex power systems characterized by knowledge intensity and dynamic changes, LLM suffer from two key drawbacks: hallucinations [3] and insufficient professional knowledge. The former stems from the models’ lack of real information support for unseen complex scenarios, which easily leads to conclusions that contradict the actual operational laws of the system. The latter, due to the scarcity and professionalism of training data in the power system domain [4], results in the models’ inability to accurately parse technical terminology, operational logic, and regulatory frameworks within the field. When faced with system-related questions beyond the scope of their training data, LLM often fabricate answers or provide ineffective responses.
The complexity of power systems is reflected in the collaborative operation of subsystems throughout the entire chain of power generation, transmission, transformation, distribution, and consumption. Meanwhile, high-value knowledge is scattered across various types of document carriers, such as power standards, operating procedures, and safety regulations. This creates a contradiction between fragmented knowledge and the full-process system [5]. Such a contradiction makes it difficult for general-purpose LLM to achieve in-depth understanding of professional contexts and accurate parsing of semantics in intelligent processing scenarios for complex power systems. This directly leads to insufficient professionalism and accuracy in the content they generate for key tasks such as equipment maintenance recommendations and business development, failing to meet the stringent requirements for knowledge reliability in complex system optimization and decision-making.
To address the adaptation challenge of LLM in the intelligent processing of complex systems, the Retrieval-Augmented Generation (RAG) [6] paradigm integrated with machine learning technologies has been proposed, emerging as a key technology bridging general-purpose models and specialized complex systems. In tasks involving intelligent question-answering and text generation for complex power systems, RAG enables the effective utilization of professional knowledge through a dual-module collaborative mechanism. The retrieval module leverages retrieval models to efficiently locate contextual information relevant to the current system-related question within the power system domain corpus. The generation module, through prompt engineering techniques, integrates the professionally retrieved knowledge with the generative capabilities of LLM, and utilizes the capabilities of LLM to achieve the rational application of knowledge. By explicitly introducing external professional knowledge sources, this paradigm fundamentally enhances the output reliability of LLM in intelligent processing tasks for complex power systems, providing technical support for machine learning-aided optimization and decision-making in power systems.
When retrieving relevant documents, RAG always retrieves the most relevant documents. However, studies have shown that semantically similar distracting information can adversely affect LLM [7]. Interestingly, in certain scenarios, introducing unrelated “noisy” documents into the context of a RAG system can improve model performance [8], as such irrelevant documents help the model better focus on useful information.
In view of the above issues, this paper proposes a fusion method. First, a large language model generates a hypothetical document to the given question, and then the question is concatenated with this document. After that, the retriever performs similarity retrieval in the corpus based on the concatenated document and submits the results to the large language model for answer generation. Next, the question is concatenated with the answer generated by the first round of RAG; this concatenated content is used to retrieve relevant documents from the corpus. Before submitting to the large language model for generation, different numbers of completely irrelevant noisy documents are added to different positions in the context of the retrieved relevant documents, so as to help the large language model focus better on the relevant documents. Figure 1 presents an example of the method.
Figure 1. An example of generating virtual documents and adding noise for iterative RAG.
In our experiments, we demonstrate the performance of the method proposed in this paper using Llama3-Chinese-8B-Instruct as the generative model on the power dataset and the CMRC2018 dataset. Overall, our main contributions can be summarized as follows:
(1) A two-stage RAG optimization framework via hypothetical answer-guided retrieval and dynamic noise augmentation: This framework breaks through the inherent paradigm of traditional RAG, which is limited to question-only retrieval and noise-free context. Leveraging hypothetical answers generated by LLM, it expands the semantic dimensions of retrieval, addressing the critical pain point in professional domains where “vague question formulation leads to retrieval bias”. Meanwhile, it innovatively designs a dynamic noise injection mechanism, clarifying the quantity gradient of noisy documents and position selection strategies, thus providing a reproducible technical solution for revealing the intrinsic mechanism of “irrelevant information enhancing model focus”.
(2) Verification of the effectiveness and universality of the noise augmentation strategy in professional complex domains: Contrast experiments are conducted for the first time on a power domain dataset and the general CMRC2018 dataset. The study systematically analyzes the influence patterns of noise quantity and injection position on model performance, verifying the effectiveness of the proposed method in improving the accuracy of professional question answering.
(3) Provision of technical references for AI-assisted intelligent question answering in power complex systems: Targeting the characteristics of fragmented knowledge and dense professional terminology in the power domain, the proposed method—through the dual guarantees of retrieval optimization and noise augmentation—reduces the hallucination rate of general LLM in power professional question answering. It thereby offers a new implementation path for the reliable utilization of knowledge in the intelligent processing of complex systems.

2. Related Works

In 2017, Google introduced the Self-Attention mechanism in its research and constructed the Transformer model [9]. This architecture broke through the processing limitations of traditional sequence models, opening up a new path for the field of natural language processing. Subsequently, based on the technical accumulation of Transformer, Google further developed the pre-trained language model BERT [10]. In subsequent research, researchers found that expanding the scale of language model parameters or data can improve the performance of the model. Therefore, some studies have begun to attempt to train pre-trained language models with larger parameter and data scales. LLM are more powerful than any models previously seen in Natural Language Processing (NLP).
The basic RAG architecture consists of a retrieval system and a generator. The retriever converts user input queries into vectors through an encoding model, then retrieves information from an external corpus based on similarity, and integrates the retrieved relevant documents with user queries into a prompt template, which is then fed to a large language model for generation. This approach allows developers to provide incremental information input to the model through external knowledge bases without retraining the entire large model for specific tasks, thereby achieving more accurate responses while keeping model parameters unchanged. This hierarchical architecture of “retrieval-prompt-generation” not only avoids the high computational costs of large-scale model fine-tuning but also provides a flexible interface for the dynamic update of professional domain knowledge, making RAG a key technology for enhancing the practicality of large language models in current knowledge-intensive tasks. With the development of RAG technology, it has gradually evolved into a modular structure. Modular RAG [11] breaks through the framework of traditional RAG by integrating various methods at different stages to expand functional modules, making RAG more flexible and diverse.
Information retrieval plays a crucial role in the RAG process. Sparse retrieval methods such as Term Frequency-Inverse Document Frequency (TF-IDF) [12] rely on precise keyword matching and demonstrate high efficiency in large-scale document retrieval. However, they often struggle to understand the semantic relationships between words. In contrast, dense retrieval transforms documents and queries into dense vectors through deep learning models, leveraging vector similarity (e.g., cosine distance) to retrieve semantically matched documents. Dense retrieval methods like DPR [13] have proven to be highly effective in handling semantic matching tasks.
A method called Generation-Augmented Retrieval (GAR) [14] optimizes the retrieval process by generating heuristic relevant context to enhance queries, without external resources for supervision. Hypothetical Document Embeddings (HyDE) [15] is used for dense retrieval in zero-shot scenarios. With the help of the power of the LLM, it can achieve efficient retrieval without the need for relevant annotation data. Instead of directly converting the query into an embedding vector, it generates a fictional document that can answer the query through the LLM and then converts the generated virtual answer into an embedding vector. The Rewrite-Retrieve-Read (RRR) [16] framework adds a query rewriting step before retrieval to bridge the gap between the input text and retrieval needs, which helps generate more precise queries. This method first uses a large language model to generate queries, then employs a web search engine to retrieve relevant documents, and then uses a small trainable language model as a rewriter, which is trained via reinforcement learning with the performance of the large language model as a reward. In this way, the rewriter can adjust queries based on feedback from the large language model.
Noise injection in LLM refers to the process of adding noise to input data during the training or inference phase. A Robust Contrastive Pre-training (RCP) [17] approach is proposed, combining multilingual noisy data augmentation with contrastive loss in the pre-training phase. Existing studies have also employed noisy data for data augmentation [18]. In recent years, researchers have focused their attention on noise injection techniques for RAG systems [19]. Some studies have classified noise into beneficial noise and detrimental noise [20,21]. Their research found that in the presence of beneficial noise, the model enhances its attention to key context and achieves better discrimination between correct and incorrect information. Furthermore, beneficial noise contributes to the formation of more standardized answer formats. When combined with beneficial noise, LLM typically exhibit lower uncertainty and greater confidence in their outputs. The underlying principle behind these phenomena is that the semantic similarity between the noise and the query is extremely low. After Softmax normalization, these noisy documents obtain attention weights close to 0. Since the sum of attention weights equals 1, the relative attention weight of the Golden Context is passively amplified.

3. Methodology

To address the semantic gap between queries and corpora in RAG systems and enhance the robustness of LLM for improved overall performance, we propose a novel framework illustrated in Figure 2. More specifically, our approach optimizes the traditional RAG pipeline across three key dimensions:
Figure 2. The Flowchart of the Methodology.
(1)
Virtual Document Substitution: Using a pre-trained LLM, we generate context-rich pseudo-documents to semantically expand the original query. This mechanism closes the representational mismatch between concise queries and lengthy corpus documents by simulating the semantic distribution of complete documents, which in turn improves retrieval accuracy.
(2)
Iterative Query-Answer Retrieval: We first integrate the retrieved documents into a prompt template to generate an initial answer. We then concatenate this intermediate answer with the original query and embed them as a refined retrieval vector; this allows for a second-stage retrieval that focuses on more relevant documents.
(3)
Noise-Enhanced Robustness: Before inputting documents into the LLM, we introduce irrelevant “noise” documents alongside the retrieved content. This training strategy forces the model to discern signal from noise, which enhances its ability to focus on informative content and improves output robustness.
We treat the process of virtual document substitution and second-stage retrieval generation as one module, and adding irrelevant documents as another module. We conducted experiments on the electricity power dataset and the CMRC dataset [22], and validated through ablation studies that both modules contribute to performance improvements in generated results. The technical details and experimental design of this approach will be elaborated in the subsequent sections.

3.1. Text Chunk Indexing

We optimize text chunking in three ways. First, we standardize text by removing special characters and fixing encoding errors, while preserving content accuracy. This improves RAG performance. Second, we use semantic chunking. Semantic chunking is employed to maximize the relevance of text chunk contents, which not only facilitates retrieval but also collects as much context as possible while reducing noise. Third, we embed metadata directly into chunks, which makes retrieval more targeted and efficient.

3.2. Virtual Document Retrieval

Dense retrieval evaluates the similarity between a query and documents by computing their inner product. For a question q and a document d i from the corpus D, two encoders E q and E d are employed to embed them respectively. Specifically, E d maps M passages into d-dimensional dense vectors. For a question q, E q maps it into a d-dimensional vector and calculates the inner product with document d i . The inner product calculation formula is as follows:
sim ( q , d i ) = E q ( q ) · E d ( d i ) .
This score quantifies the relevance of each document to the query by measuring their similarity in the embedded vector space. Documents are then ranked based on these scores, and the top-k documents with the highest rankings are selected as relevant documents:
top- k docs = arg max d i D { sim ( q , d i ) } .
We introduce a LLM to generate virtual documents:
d v i r t u a l = LLM ( q ) ,
which replaces the query for retrieval. The retrieved relevant documents are added to a prompt template for generation, yielding an intermediate answer denoted as q v . Combining q v with q, we perform a second retrieval to obtain relevant documents, which are then added to the prompt template together with irrelevant documents for generation, resulting in the final answer.

3.3. The Introduction of Irrelevant Documents

In natural language processing tasks, the introduction of irrelevant documents serves as a key means to evaluate the robustness of retrieval systems and verify their semantic filtering capabilities. By introducing a certain amount of irrelevant information interference, the model’s ability to distinguish between semantic relevance and superficial feature matching is tested. The irrelevant documents used in this paper are composed of 3000 randomly selected common Chinese characters and short sentences completely unrelated to the electric power industry. When embedding relevant documents into the prompt template, different quantities of irrelevant documents are inserted at various positions—such as before or after the relevant documents—to test the impact of relevant documents’ positions within the prompt template on the generation results.

4. Experiments

In this section, we introduce the datasets, metrics, comparative methods, and experimental results. Based on the experimental findings, we also discuss the impact of placing relevant documents in different positions of the prompt template on the results. Finally, we validate the above experiments on the public dataset CMRC.

4.1. Setup

4.1.1. Implementation

In our experiments, the bge-large-zh-v1.5 [23] (https://huggingface.co/BAAI/bge-large-zh-v1.5, accessed on 11 February 2026) model is employed for retrieval, and metadata (such as dates) is added during index construction to enhance retrieval performance. For the generative model, we use Llama3-Chinese-8B-Instruct (https://huggingface.co/hfl/llama-3-chinese-8b-instruct, accessed on 11 February 2026), an 8-billion-parameter model that incorporates Grouped Query Attention (GQA), significantly improving inference speed and efficiency. The experiments were conducted by deploying LLM on NVIDIA A800 80 G GPUs. We conducted experiments using the sequential RAG method pipeline provided by FlashRAG [24], which offers a highly modular RAG framework allowing free combination of different components.

4.1.2. Datasets

The electricity dataset used in this paper is generated by domain experts in the electricity field based on professional knowledge documents within the power sector. These questions cover aspects such as data governance, power analysis and prediction, and business mechanisms. The dataset consists of a total of 500 entries, each containing: (1) a question; (2) a reference answer strictly grounded in the corpus; (3) the source paragraph of the reference answer; and (4) the source document identifier.
To ensure the consistency and reliability of annotation quality, we strictly adhere to a standardized annotation workflow, with the specific guidelines as follows: (1) Each answer to the annotated questions must be derived from exactly one paragraph-level source, ensuring the uniqueness of answer provenance. (2) Reference answers must be strictly grounded in the knowledge provided in the corpus, with no external knowledge or subjective inferences permitted. (3) Ambiguous or vague questions shall be rewritten to ensure their standardization.
In the annotation execution phase, each sample is independently drafted by a dedicated annotator. Upon completion, the sample must be submitted to at least one senior domain expert for review and validation, to ensure the accuracy and authority of the final annotation results. Among the question types, definition-based questions account for 12.4%, procedural questions for 37.2%, and analytical questions for 50.4%.
Additionally, we conducted experiments using the public dataset CMRC to validate the effectiveness of the proposed method. CMRC is a dataset designed for Chinese machine reading comprehension tasks. It comprises a series of carefully designed questions and their corresponding texts, annotated by human experts from Wikipedia paragraphs, aiming to evaluate machines’ ability to understand and parse Chinese language corpora. We randomly selected 500 examples from this dataset for experimentation.

4.1.3. Metrics

For evaluating the experimental results, we employed ROUGE family metrics [25] and BLEU score [26] to measure the similarity between the reference answers and the generated answers. ROUGE is an indicator widely used in text summarization evaluation and machine translation. ROUGE mainly focuses on the recall rate, that is, it emphasizes the coverage and integrity of text information. Let x be the candidate text, { s ( 1 ) , , s ( K ) } be the set of reference texts, and W be the set of n-grams extracted from the set of reference texts. The calculation formula for ROUGE-N is as follows:
ROUGE - N ( x ) = k = 1 K w W min ( c w ( x ) , c w ( s ( k ) ) ) k = 1 K w W c w ( s ( k ) ) ,
where c w ( x ) is the number of occurrences of the n-gram w appears in the candidate text x, and c w ( s ( k ) ) is the number of occurrences of the n-gram w in the reference text s ( k ) . There is a variant of the ROUGE series, namely ROUGE-L. The “L” in ROUGE-L stands for the longest common subsequence, which measures the similarity between two sequences. ROUGE-L is calculated by the F1 score, which combines the information of precision and recall. BLEU is an evaluation metric widely adopted in the field of machine translation. It evaluates by calculating the lexical similarity between the candidate text and the reference text. BLEU mainly calculates the co-occurrence frequency of n-grams between the candidate text and the reference text. The specific calculation formula is as follows:
BLEU = BP × exp n = 1 N w n × log p n
Among them, w n is the weight of the n-gram, which is used to adjust the influence of n-gram of different lengths on the final score. In this study, n is usually set to 4, and w n follows an equal weight distribution. BP is the penalty coefficient, which is used to penalize the differential impact caused by short sentences.
In knowledge-based question answering systems, RAGAS [27] is commonly employed to evaluate the factual consistency and faithfulness of the QA system. This paper utilizes four metrics from RAGAS: Context Relevance, Answer Relevance, Faithfulness, and Response Groundedness. Context Relevance assesses whether all content in the retrieved documents is relevant to the question. It decomposes all retrieved contexts into individual sentences using an LLM, where the total number of sentences is denoted as | S c | . It then counts the number of sentences that are helpful to the question, denoted as | V c | . The ratio of these two values yields the Context Relevance score:
CR = | V c | | S c | .
Answer Relevance measures the alignment between the answer and the question. It uses an LLM to reverse-engineer the question from the answer, and employs an embedding model to extract the semantic vectors of the answer and the generated question. The similarity between these vectors is calculated, with the formula expressed as:
AR = 1 n i = 1 n s i m ( q , q i ) ,
where n is the number of generated questions, q i represents the i-th generated question, and q denotes the actual question.
Faithfulness evaluates the extent to which the answer is consistent with the retrieved contexts. Since LLMs have the ability to fabricate responses, in an ideal RAG system, the answer should be entirely inferred from the provided contexts. The method involves extracting key points from the answer using an LLM, with the total number of key points denoted as | S a | . An LLM is then used to verify whether these key points can be inferred from the contexts, where the number of inferable key points is denoted as | V a | . The ratio of these two values is computed as:
F = | V a | | S a | .
Response Groundedness is a core metric that evaluates whether the generated content is grounded in the retrieved context. It measures the factual consistency between the model’s response and the retrieved documents, serving as a critical tool for detecting “hallucinations” in RAG systems.

4.1.4. Failure Cases Analysis

We analyze the failure cases to better understand the failure causes of the RAG pipeline in this study. Detailed cases and corresponding reasons are provided in Appendix A.

4.2. Main Result

Table 1 and Table 2 compare our method with other baselines on the electricity power and CMRC datasets.
Table 1. Experimental results of comparing different methods on electricity power dataset.
Table 2. Experimental results of comparing different methods on CMRC dataset.
Our method achieves the best results on both datasets, improving retrieval accuracy and generation quality. Analyzing these results reveals why our framework performs better:
First, virtual document generation overcomes the query-document mismatch in standard RAG. By expanding raw queries into full pseudo-documents, we capture wider semantic context and improve retrieval coverage. Second, we concatenate the initial answer with the original query for a second retrieval. This refines the search using partial results from the first stage, improving retrieval accuracy. Finally, adding noise documents during training improves robustness. The model learns to ignore distractors, maintaining output quality when irrelevant content appears.
Table 3 presents the RAGAS evaluation results on the CMRC dataset. Our method beats the naive baseline on most metrics. ITER-RETGEN performance best overall, confirming that iterative refinement works well. IRCoT match ITER-RETGEN on Answer Relevancy and Response Groundedness. On Faithfulness, ur method surpasses the naive approach and trails ITER-RETGEN, while exceeding all other comparative methods. This indicates that pseudo-document generation and secondary retrieval do not substantially introduce hallucination risks, and the factual quality of retrieved documents is preserved. But the gap with ITER-RETGEN suggests that simple query expansion is less effective than iterative self-correction in ensuring factual consistency.
Table 3. Multi-Dimensional evaluation results of different RAG methods on the CMRC dataset.

4.3. Ablation Study

To explore the roles of different modules, we conducted an ablation study, and the results are shown in Table 4. We use w/o to denote the version without a specific module. It can be observed that: the performance of the method declines when any component is removed, indicating that all components are highly effective. The results indicate that on electricity power dataset, removing the HyDE module leads to a relative decline of 1.91% in Rouge-L and 6.01% in BLEU-4. This suggests that the generated semantically expanded documents can effectively bridge the representational gap between the query and the corpus. When irrelevant documents are excluded, Rouge-L decreases by 2.19% relatively and BLEU-4 by 4.65% relatively, verifying the necessity of the noise enhancement strategy for improving the robustness of LLM.
Table 4. The ablation results under different module settings.

4.4. Impact of Irrelevant Documents

Table 5 shows how adding irrelevant documents affects answer quality. We let N denote the number of irrelevant documents added.
Table 5. Relevant documents are behind different numbers of irrelevant documents.
It can be observed that as the number of irrelevant documents increases, the generation quality of the large language model first rises and then declines. Likely because without noise, the model fixates on highly relevant documents and misses weaker but valid ones. Irrelevant documents provide contrast, redirecting attention to these weaker relevant documents. As more irrelevant documents are added, performance degrades and eventually falls below the baseline. This indicates that noise receives low attention scores in LLM. Moderate noise helps the LLM focus attention, while excessive noise increases its weight and diverts attention, degrading generation quality.

4.5. Impact of Relevant Documents Positioning

Figure 3 and Figure 4 show how the position of relevant documents in the prompt affects performance.
Figure 3. The ROUGE-L scores when relevant documents are located at different positions in the prompt template.
Figure 4. The BLEU-4 scores when relevant documents are located at different positions in the prompt template.
We test three arrangements: relevant documents placed after, among, or before irrelevant ones. These correspond to placing relevant information far from, near, or closest to the query. When relevant documents are placed after irrelevant ones, the model achieves better overall and best performance than the other two groups. This arrangement also outperforms the baseline across all noise levels, while the other two positions fall below baseline when noise increases.
When processing long prompts, generative LLM tend to focus more on end-sequence content. Placing relevant documents after irrelevant ones puts key information in this high-attention region, helping the model better extract relevant content from noise and reduce interference.
Table 6 demonstrates the optimal performance of relevant documents in different positions. R stands for relevant documents.
Table 6. The influence of the position of relevant documents on the results.
Results in Table 6 further demonstrate that the optimal performance achieved by varying the position of relevant documents consistently outperforms the naive RAG method.This indicates that, without modifying the base retrieval and generation framework, simply optimizing the position of relevant documents within the prompt can stably improve the model robustness in complex noisy environments.Compared with naive RAG without positional optimization, properly arranging document order in prompts enables the model to focus more effectively on critical information.

4.6. Efficiency Analysis

Figure 5 illustrates the resource consumption characteristics of various methods when processing power domain datasets. The horizontal axis represents the total running time, which reflects the inference speed of each method. The vertical axis represents the peak memory footprint, indicating the computational resource overhead during execution. The bubble size is positively correlated with the answer generation quality, where the ROUGE-L score is adopted as the quantitative metric. Meanwhile, the color intensity indicates the value of the comprehensive efficiency score: darker colors represent better overall performance, while lighter colors represent poorer performance. To quantitatively evaluate the trade-off among efficiency, resource cost, and generation quality, we define the comprehensive efficiency score E s c o r e as a weighted sum of three normalized indicators, formulated as follows:
E s c o r e = w t E t i m e + w m E m e m + w q E q u a l .
Figure 5. Three-Dimensional perspective on RAG method efficiency comparison: a collaborative evaluation of time, memory footprint, and answer quality.
Here, E t i m e , E m e m , and E q u a l denote the min-max normalized scores of running time, memory footprint, and ROUGE-L score, respectively. All original metrics are linearly normalized to the range [0, 1] to ensure comparability. To ensure visual interpretability, the ROUGE-L scores were first normalized to a range of [0, 1], where the minimum normalized value is 0. Despite this, the bubble size in Figure 5 never reaches 0, as we set a fixed minimum bubble size to avoid imperceptible bubbles for samples with the lowest ROUGE-L scores. In this study, the weights are empirically set to w t = 0.4 , w m = 0.3 , and w q = 0.3 , placing a slightly higher emphasis on inference efficiency while balancing memory usage and answer quality.
Naive, Flare, and selective-context in the lower-left quadrant show fast inference and low GPU memory footprint. Among them, selective-context is fastest but its small bubbles show it sacrifices quality. Flare uses the least VRAM and maintains fast inference, with its dark color indicating high overall efficiency. Naive RAG balances quality and speed. It runs as fast as Flare but has much larger bubbles, making it a stable baseline.
In the lower-right quadrant, the Sure method represents the extreme case of inference latency while maintaining negligible GPU memory overhead. Its intermediate color intensity suggests moderate overall efficiency, rendering it suitable for memory-constrained environments that can tolerate prolonged latency. ITER-RETGEN, conversely, offers high quality at moderate speed—bubbles match Naive RAG but take longer.
The center of the scatter plot shows a balanced trade-off. The proposed method (Ours) is positioned centrally, with its deep red hue denoting superior composite efficiency. Compared to ITER-RETGEN, our approach achieves a 20% reduction in runtime while maintaining comparable GPU memory utilization; relative to Naive RAG, our method incurs an approximately 2× increase in computational time but offers advantages in accuracy and inference stability. Ircot occupies the same vertical segment as ITER-RETGEN but exhibits marginally smaller bubble dimensions, indicating slightly inferior generation quality.
Finally, Repluge manifests as an outlier with anomalous GPU memory consumption, utilizing approximately 1.7× the VRAM of other methods. Its pale yellow coloration signals low composite efficiency, likely attributable to full-context loading mechanisms that lack memory optimization.

5. Conclusions

This study addresses the core challenges faced by RAG technology in knowledge-base question answering scenarios within the power domain, including semantic space mismatch, the tendency to overlook sub-relevant information, and insufficient model robustness. We propose a framework that combines virtual document generation, iterative retrieval, and noise injection to address these problems. Using LLMs to generate virtual documents broadens the query semantics, and the iterative retrieval with “query + preliminary answer” helps locate information more accurately. Additionally, motivated by the long-tail effect of the attention mechanism, adding an appropriate amount of irrelevant documents can enhance the LLM attention toward relevant documents, and appropriately adjusting the position of relevant documents within the context also effectively improves their attention weights. Experiments on power-domain and general-domain (CMRC) datasets show consistent improvements.
However, this research still has several limitations that require critical reflection and subsequent improvement:
(1) Semantic chunking sometimes split critical power-domain concepts across chunks, hurting retrieval completeness.
(2) The retrieval performance degrades when facing abstract or implicit queries that lack explicit terminological matches.
(3) The model incurs increased latency and cannot well meet high real-time requirements.
Future work can address these limitations in the following directions: First, we can improve retrieval by integrating power-specific terminology and knowledge graphs to better handle abstract queries. Second, explore lightweight iterative retrieval strategies and model compression technologies to improve inference efficiency while ensuring performance, thereby promoting the practical application of the method in real-time scenarios. Finally, we will investigate how to improve the generalization ability of the method, enabling effective deployment in entirely new professional domains.

Author Contributions

Conceptualization, X.L. and Y.C.; methodology, Y.C., Q.P. and Y.Y.; software, Y.C. with assistance from Y.Y.; validation, Y.Z. and Q.P.; formal analysis, Y.C. and Q.P.; investigation, Y.C., Y.Z. and K.C.; resources, Y.Z., Y.L. and K.C.; data curation, Y.C. and Y.L.; writing—original draft preparation, Y.C.; writing—review and editing, Y.Y., X.L. and Y.C.; supervision, X.L. and Q.P.; project administration, Y.L., K.C. and X.L.; funding acquisition, X.L. All authors have read and agreed to the published version of the manuscript.

Funding

This work is financially supported by 2025 Open Fund of Beijing Key Laboratory of Demand Side Multi-Energy Carriers Optimization and Interaction Technique (Project title: Research on multi-source cognition-driven intelligent modeling and generative reasoning techniques for user load forecasting, Project No. SGDK0000YDJS2504840-B1).

Data Availability Statement

The original contributions presented in the study are included in the article; further inquiries can be directed to the corresponding author.

Acknowledgments

All authors who contributed to this study are gratefully acknowledged.

Conflicts of Interest

The authors declare no conflicts of interest.

Appendix A. Failure Cases Analysis

This appendix presents representative failure cases of the method proposed in this paper for analytical purposes. All examples have been anonymized and simplified. To avoid disclosing sensitive content, the retrieved relevant documents are summarized to retain only their core content and key terms.
  • Case1
  • Question: What are the future work plans of Hebei Electric Power for 2023?
  • Gold answer: (1) Refine hierarchical responsibilities for power supply assurance; (2) Promote the construction of a new-type load management system; (3) Establish a market analysis and forecasting framework.
  • Top-5 documents:
  • Challenges faced by Hebei Electric Power regarding high-voltage transmission.
  • Electricity consumption in the Beijing-Tianjin-Hebei region during Winter 2023 and subsequent planning: (1) Refine hierarchical responsibilities for power supply assurance; (2) Promote the construction of a new-type load management system.
  • Power supply and demand conditions in Hebei during Summer 2023.
  • Electricity consumption patterns of Hebei’s manufacturing sector in 2023.
  • Analysis of total social installed capacity in Hebei for 2023.
  • LLM output: Based on the provided documents, the future work plans of Hebei Electric Power for 2023 cannot be determined.
  • Diagnosis: The complete answer is distributed across two separate text chunks; however, the retriever only recalled one of them. Furthermore, the relevant answer segment constitutes only a minor fraction of the retrieved chunk, which the generator consequently failed to exploit effectively. The root cause of this failure lies in semantic fragmentation during the document preprocessing stage: critical information required for a comprehensive response was inadvertently segmented into multiple disparate chunks. This incomplete retrieval prevented the LLM from accessing coherent and comprehensive context, resulting in semantic discontinuity and comprehension bias. Ultimately, the system generated an “unanswerable” response.
  • Case2
  • Question: What were the objectives of State Grid Corporation of China regarding market forecasting for the second half of 2022?
  • Gold answer: Forecast peak load, anticipate power supply gaps, and complete electricity market analysis and forecasting reports.
  • Top-5 documents:
  • Promoting the establishment of a market-oriented system for market forecasting and analysis.
  • Work schedule for market analysis and forecasting, emphasizing inter-departmental coordination.
  • Standardizing and optimizing business processes, with specific work requirements and procedures aimed at compliant operations and excellent service.
  • New requirements for the market work system, refining job standards and work standards.
  • Focusing on establishing and improving business systems for market business analysis, and coordinating business platforms.
  • LLM output: The market professional management system has been fully established, refining load management organization and implementation, as well as market analysis and forecasting business processes.
  • Diagnosis: This represents a case of semantic retrieval failure. The embedding model was unable to effectively distinguish between “objectives of market forecasting by the grid corporation” and “market analysis and forecasting” as routine activities. Due to the high degree of semantic overlap between these keywords, the retriever recalled generic documents concerning routine market forecasting tasks, which subsequently led the generator to produce an incorrect response.

References

  1. Brown, T.B.; Mann, B.; Ryder, N.; Subbiah, M.; Kaplan, J.; Dhariwal, P.; Neelakantan, A.; Shyam, P.; Sastry, G.; Askell, A.; et al. Language models are few-shot learners. In Proceedings of the 33th International Conference on Neural Information Processing Systems; Curran Associates Inc.: Red Hook, NY, USA, 2020; pp. 1877–1901. [Google Scholar] [CrossRef] [Scilit]
  2. Touvron, H.; Martin, L.; Stone, K.; Albert, P.; Almahairi, A.; Babaei, Y.; Bashlykov, N.; Batra, S.; Bhargava, P.; Bhosale, S.; et al. Llama 2: Open foundation and fine-tuned chat models. arXiv 2023, arXiv:2307.09288. [Google Scholar] [CrossRef] [Scilit]
  3. Zhang, Y.; Li, Y.; Cui, L.; Cai, D.; Liu, L.; Fu, T.; Huang, X.; Zhao, E.; Zhang, Y.; Chen, Y.; et al. Siren’s Song in the AI Ocean: A Survey on Hallucination in Large Language Models. arXiv 2023, arXiv:2309.01219. [Google Scholar] [CrossRef] [Scilit]
  4. Kandpal, N.; Deng, H.; Roberts, A.; Wallace, E.; Raffel, C. Large Language Models Struggle to Learn Long-Tail Knowledge. In Proceedings of the 40th International Conference on Machine Learning, Honolulu, HI, USA, 23–29 July; PMLR: Cambridge, MA, USA, 2023; pp. 15696–15707. Available online: https://dl.acm.org/doi/abs/10.5555/3618408.3619049 (accessed on 11 February 2026).
  5. Majumder, S.; Dong, L.; Doudi, F.; Cai, Y.; Tian, C.; Kalathil, D.; Ding, K.; Thatte, A.A.; Li, N.; Xie, L. Exploring the capabilities and limitations of large language models in the electric energy sector. Joule 2024, 8, 1544–1549. [Google Scholar] [CrossRef] [Scilit]
  6. Lewis, P.; Perez, E.; Piktus, A.; Petroni, F.; Karpukhin, V.; Goyal, N.; Küttler, H.; Lewis, M.; Yih, W.t.; Rocktäschel, T.; et al. Retrieval-Augmented Generation for Knowledge-Intensive NLP Tasks. In Proceedings of the 33th International Conference on Neural Information Processing Systems; Curran Associates, Inc.: Red Hook, NY, USA, 2020; pp. 9459–9474. [Google Scholar] [CrossRef] [Scilit]
  7. Pandia, L.; Ettinger, A. Sorting through the noise: Testing robustness of information processing in pre-trained language models. In Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing, Online and Punta Cana, Dominican Republic, 7–11 November 2021; Association for Computational Linguistics: Stroudsburg, PA, USA, 2021; pp. 1583–1596. [Google Scholar] [CrossRef] [Scilit]
  8. Cuconasu, F.; Trappolini, G.; Siciliano, F.; Filice, S.; Campagnano, C.; Maarek, Y.; Tonellotto, N.; Silvestri, F. The Power of Noise: Redefining Retrieval for RAG Systems. In Proceedings of the 47th International ACM SIGIR Conference on Research and Development in Information Retrieval; Association for Computing Machinery: New York, NY, USA, 2024; pp. 719–729. [Google Scholar] [CrossRef] [Scilit]
  9. Vaswani, A.; Shazeer, N.; Parmar, N.; Uszkoreit, J.; Jones, L.; Gomez, A.N.; Kaiser, L.u.; Polosukhin, I. Attention is All you Need. In Proceedings of the 30th International Conference on Neural Information Processing Systems, Long Beach, CA, USA, 4–9 December 2017; Curran Associates, Inc.: Red Hook, NY, USA, 2017. [Google Scholar] [CrossRef] [Scilit]
  10. Devlin, J.; Chang, M.W.; Lee, K.; Toutanova, K. BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding. In Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers), Minneapolis, MN, USA, 2–7 June 2019; Association for Computational Linguistics: Stroudsburg, PA, USA, 2019; pp. 4171–4186. [Google Scholar] [CrossRef] [Scilit]
  11. Gao, Y.; Xiong, Y.; Gao, X.; Jia, K.; Pan, J.; Bi, Y.; Dai, Y.; Sun, J.; Guo, Q.; Wang, M.; et al. Retrieval-Augmented Generation for Large Language Models: A Survey. arXiv 2023, arXiv:2312.10997. [Google Scholar] [CrossRef] [Scilit]
  12. Sparck Jones, K. A statistical interpretation of term specificity and its application in retrieval. J. Doc. 1972, 28, 11–21. [Google Scholar] [CrossRef] [Scilit]
  13. Karpukhin, V.; Oğuz, B.; Min, S.; Lewis, P.; Wu, L.; Edunov, S.; Chen, D.; Yih, W. Dense passage retrieval for open-domain question answering. In Proceedings of the EMNLP 2020—2020 Conference on Empirical Methods in Natural Language Processing, Proceedings of the Conference, Online, 16–20 November 2020; Association for Computational Linguistics: Stroudsburg, PA, USA, 2020; pp. 6769–6781. [Google Scholar] [CrossRef] [Scilit]
  14. Mao, Y.; He, P.; Liu, X.; Shen, Y.; Gao, J.; Han, J.; Chen, W. Generation-Augmented Retrieval for Open-Domain Question Answering. In Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (Volume 1: Long Papers); Association for Computational Linguistics: Stroudsburg, PA, USA, 2021; pp. 4089–4100. [Google Scholar] [CrossRef] [Scilit]
  15. Gao, L.; Ma, X.; Lin, J.; Callan, J. Precise Zero-Shot Dense Retrieval without Relevance Labels. In Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics, Toronto, ON, Canada, 9–14 July 2023; Association for Computational Linguistics: Stroudsburg, PA, USA, 2023; pp. 1762–1777. [Google Scholar] [CrossRef] [Scilit]
  16. Ma, X.; Gong, Y.; He, P.; Zhao, H.; Duan, N. Query Rewriting in Retrieval-Augmented Large Language Models. In Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing, Singapore, 6–10 December 2023; Association for Computational Linguistics: Stroudsburg, PA, USA, 2023; pp. 5303–5315. [Google Scholar] [CrossRef] [Scilit]
  17. Cooper Stickland, A.; Sengupta, S.; Krone, J.; Mansour, S.; He, H. Robustification of Multilingual Language Models to Real-world Noise in Crosslingual Zero-shot Settings with Robust Contrastive Pretraining. In Proceedings of the 17th Conference of the European Chapter of the Association for Computational Linguistics, Dubrovnik, Croatia, 2–6 May 2023; Association for Computational Linguistics: Stroudsburg, PA, USA, 2023; pp. 1375–1391. [Google Scholar] [CrossRef] [Scilit]
  18. Ye, J.; Xu, N.; Wang, Y.; Zhou, J.; Zhang, Q.; Gui, T.; Huang, X. LLM-DA: Data Augmentation via Large Language Models for Few-Shot Named Entity Recognition. arXiv 2024, arXiv:2402.14568. [Google Scholar] [CrossRef] [Scilit]
  19. Chen, J.; Lin, H.; Han, X.; Sun, L. Benchmarking large language models in retrieval-augmented generation. In Proceedings of the AAAI Conference on Artificial Intelligence; AAAI Press: Cambridge, MA, USA, 2024; Volume 38, pp. 17754–17762. [Google Scholar] [CrossRef] [Scilit]
  20. Al Sharou, K.; Li, Z.; Specia, L. Towards a Better Understanding of Noise in Natural Language Processing. In Proceedings of the International Conference on Recent Advances in Natural Language Processing, Online, 1–3 September 2021; INCOMA Ltd.: Shoumen, Bulgaria, 2021; pp. 53–62. Available online: https://aclanthology.org/2021.ranlp-1.7 (accessed on 11 February 2026).
  21. Wu, J.; Zhang, S.; Che, F.; Feng, M.; Shao, P.; Tao, J. Pandora’s box or aladdin’s lamp: A comprehensive analysis revealing the role of rag noise in large language models. In Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), Vienna, Austria, 27 July–1 August 2025; Association for Computational Linguistics: Stroudsburg, PA, USA, 2025; pp. 5019–5039. [Google Scholar] [CrossRef] [Scilit]
  22. Cui, Y.; Liu, T.; Che, W.; Xiao, L.; Chen, Z.; Ma, W.; Wang, S.; Hu, G. A Span-Extraction Dataset for Chinese Machine Reading Comprehension. In Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP), Hong Kong, China, 3–7 November 2019; Association for Computational Linguistics: Stroudsburg, PA, USA, 2019; pp. 5883–5889. [Google Scholar] [CrossRef] [Scilit]
  23. Xiao, S.; Liu, Z.; Zhang, P.; Muennighoff, N.; Lian, D.; Nie, J.Y. C-Pack: Packed Resources For General Chinese Embeddings. In Proceedings of the 47th International ACM SIGIR Conference on Research and Development in Information Retrieval, Washington, DC, USA, 14–18 July 2024; Association for Computing Machinery: New York, NY, USA, 2024; pp. 641–649. [Google Scholar] [CrossRef] [Scilit]
  24. Jin, J.; Zhu, Y.; Dou, Z.; Dong, G.; Yang, X.; Zhang, C.; Zhao, T.; Yang, Z.; Wen, J.R. FlashRAG: A Modular Toolkit for Efficient Retrieval-Augmented Generation Research. In Proceedings of the Companion Proceedings of the ACM on Web Conference 2025; Association for Computing Machinery: New York, NY, USA, 2025; pp. 737–740. [Google Scholar] [CrossRef] [Scilit]
  25. Lin, C.Y. Rouge: A package for automatic evaluation of summaries. In Proceedings of the ACL Workshop: Text Summarization Branches Out 2004, Barcelona, Spain, 25–26 July 2004; Association for Computational Linguistics: Stroudsburg, PA, USA, 2004; pp. 74–81. [Google Scholar]
  26. Papineni, K.; Roukos, S.; Ward, T.; Zhu, W.J. BLEU: A method for automatic evaluation of machine translation. In Proceedings of the 40th Annual Meeting on Association for Computational Linguistics, Philadelphia, PA, USA, 6–12 July 2002; Association for Computational Linguistics: Stroudsburg, PA, USA, 2002; pp. 311–318. [Google Scholar] [CrossRef] [Scilit]
  27. Es, S.; James, J.; Anke, L.E.; Schockaert, S. Ragas: Automated evaluation of retrieval augmented generation. In Proceedings of the 18th Conference of the European Chapter of the Association for Computational Linguistics: System Demonstrations, St. Julians, Malta, 17–22 March 2024; Association for Computational Linguistics: Stroudsburg, PA, USA, 2024; pp. 150–158. [Google Scholar] [CrossRef] [Scilit]
  28. Shao, Z.; Gong, Y.; Shen, Y.; Huang, M.; Duan, N.; Chen, W. Enhancing Retrieval-Augmented Large Language Models with Iterative Retrieval-Generation Synergy. In Proceedings of the Findings of the Association for Computational Linguistics: EMNLP 2023, Singapore, 6–10 December 2023; Association for Computational Linguistics: Stroudsburg, PA, USA, 2023; pp. 9248–9274. [Google Scholar] [CrossRef] [Scilit]
  29. Kim, J.; Nam, J.; Mo, S.; Park, J.; Lee, S.W.; Seo, M.; Ha, J.W.; Shin, J. SuRe: Summarizing Retrievals using Answer Candidates for Open-domain QA of LLMs. In Proceedings of the International Conference on Learning Representations; International Conference on Learning Representations: Appleton, WI, USA, 2024; pp. 15213–15245. [Google Scholar] [CrossRef] [Scilit]
  30. Jiang, Z.; Xu, F.; Gao, L.; Sun, Z.; Liu, Q.; Dwivedi-Yu, J.; Yang, Y.; Callan, J.; Neubig, G. Active Retrieval Augmented Generation. In Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing, Singapore, 6–10 December 2023; Association for Computational Linguistics: Stroudsburg, PA, USA, 2023; pp. 7969–7992. [Google Scholar] [CrossRef] [Scilit]
  31. Trivedi, H.; Balasubramanian, N.; Khot, T.; Sabharwal, A. Interleaving Retrieval with Chain-of-Thought Reasoning for Knowledge-Intensive Multi-Step Questions. In Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics, Toronto, ON, Canada, 9–14 July 2023; Association for Computational Linguistics: Stroudsburg, PA, USA, 2023; pp. 10014–10037. [Google Scholar] [CrossRef] [Scilit]
  32. Shi, W.; Min, S.; Yasunaga, M.; Seo, M.; James, R.; Lewis, M.; Zettlemoyer, L.; Yih, W.t. REPLUG: Retrieval-Augmented Black-Box Language Models. In Proceedings of the 2024 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers), Mexico City, Mexico, 16–21 June 2024; Association for Computational Linguistics: Stroudsburg, PA, USA, 2024; pp. 8371–8384. [Google Scholar] [CrossRef] [Scilit]
  33. Li, Y. Unlocking Context Constraints of LLMs: Enhancing Context Efficiency of LLMs with Self-Information-Based Content Filtering. arXiv 2023, arXiv:2304.12102. [Google Scholar] [CrossRef] [Scilit]
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Article Metrics

Citations

Article Access Statistics

Multiple requests from the same IP address are counted as one view.