CLAINov 2, 2023

Revisiting the Knowledge Injection Frameworks

arXiv:2311.01150v1132 citationsh-index: 17
Originality Incremental advance
AI Analysis

This work addresses a critical flaw in knowledge injection methods for adapting LLMs to vertical domains, though it appears incremental as it builds on prior frameworks.

The paper identifies that injecting random knowledge into large language models (LLMs) often performs comparably or better than aligned knowledge in domain-specific tasks, undermining existing knowledge injection frameworks. It proposes a technique to prune and purify external knowledge bases, which overcomes this issue and improves performance for domain-adaptive LLMs.

In recent years, large language models (LLMs), such as GPTs, have attained great impact worldwide. However, how to adapt these LLMs to better suit the vertical domain-specific tasks by utilizing external knowledge remains not completely solved. Indeed, there have emerged a few works on this line where most of them rely on an alignment heuristic that is built to inject the corresponding knowledge tuple into the associated text sample. However, despite the promise, we identify a pivotal problem in this work ubiquitously. Simply put, we find that injecting unaligned (i.e., random) knowledge tuple into the LLMs achieves comparable (and sometimes better) results than the aligned knowledge being injected. We therefore take a thorough investigation of this frustrating finding on a variety of related prior work and further provide a chain of potential interpretations for the phenomenon. Based on all that, we offer a simple remediated technique. Briefly, the core of this technique is rooted in an ideological emphasis on the pruning and purification of the external knowledge base to be injected into LLMs. At last, we show that by integrating this technique into most (if not all) knowledge injection frameworks and recent LLMs, it manages to overcome the aforementioned sanity problem and further pushes the boundary of the performance of the domain-adaptive LLMs.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes