arXiv:2603.17566cs.CL2026-03

让大模型自己找出知识盲区,针对性提问提升训练效率。

KA2L: A Knowledge-Aware Active Learning Framework for LLMs

  • 通过分析模型隐空间识别已知与未知知识分布
  • 减少50%标注和计算成本,提升垂直领域性能
  • 适合想高效微调大模型、优化知识掌握的开发者

用高质量知识微调大语言模型(LLMs)可显著提升其性能,但现有研究缺乏对模型领域知识理解深度的评估及针对性主动学习的应用。为此,本文提出知识感知主动学习框架KA2L。该框架通过潜空间分析评估模型对特定知识点的掌握程度,构建无法回答或未知的问题。此策略聚焦模型未掌握的知识,减少对已有知识的冗余学习,提高训练效率。创新性地采用知识分布探测技术,分析Transformer特定层的隐藏状态,识别模型中已知与未知知识的分布;并提出隐状态解码方法,从潜在知识空间生成大量自然语言的未知问题。在九个开源大模型上验证表明,KA2L在两个开放域和一个垂直域数据集上均实现50%的标注与计算成本降低,同时获得更优性能,为大模型主动学习提供新思路。代码已开源:https://github.com/greenjerry/KA2L。

原文摘要 · Abstract (English)

Fine-tuning large language models (LLMs) with high-quality knowledge has been shown to enhance their performance effectively. However, there is a paucity of research on the depth of domain-specific knowledge comprehension by LLMs and the application of targeted active learning to improve their expertise. To address this gap, we introduce the Knowledge-Aware Active Learning (KA2L) framework. This framework assesses LLMs' mastery of specific knowledge points to aid in constructing unanswerable or unknowable questions through latent space analysis. This active learning strategy enhances training efficiency by focusing on knowledge the model has yet to master, thereby minimizing redundancy in learning already acquired information. This study innovatively employs a knowledge distribution probing technique to examine the hidden states of specific Transformer layers and identify the distribution of known and unknown knowledge within the LLM. Additionally, a hidden-state decoding method is proposed to generate numerous unknown questions in natural language from the latent knowledge space. In our experiments, we selected nine open-source LLMs to validate the effectiveness of the proposed framework. Results indicate that KA2L not only significantly reduces 50% annotation and computation costs across two open-domain and one vertical-domain dataset but also achieves better performance, offering valuable insights into active learning strategies for LLMs. The code is available at https://github.com/greenjerry/KA2L.

大模型主动学习知识探查高效微调

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。