MiniCheck: Efficient Fact-Checking of LLMs on Grounding Documents

TL;DR

MiniCheck achieves GPT-4 level fact-checking using synthetic data, reducing costs by 400x.

cs.CL 🔴 Advanced 2024-04-17 31 views
Liyan Tang Philippe Laban Greg Durrett
fact-checking LLMs synthetic data NLP GPT-4

Key Findings

Methodology

The study introduces a novel synthetic data generation method using GPT-4 to create challenging factual error instances. Training on this data enables MiniCheck to perform sentence-level fact-checking and recognize information synthesis across sentences. The method significantly improves model accuracy by fine-tuning on synthetic and standard entailment data.

Key Results

  • MiniCheck-FT5 excels in the LLM-AggreFact benchmark, outperforming all systems of comparable size and achieving GPT-4 accuracy.
  • Experimental results show MiniCheck surpasses AlignScore on 6 out of 10 datasets, with an average improvement of 4.3%.
  • MiniCheck achieves high performance without additional claim decomposition steps by decomposing sentences into atomic facts.

Significance

This study significantly reduces the computational cost of fact-checking, providing a more efficient solution for various NLP tasks. MiniCheck's introduction is not only significant in academia but also offers a cost-effective fact-checking tool for the industry.

Technical Contribution

MiniCheck achieves GPT-4 level performance on a small model through synthetic data generation and fine-tuning methods. This technical breakthrough offers new possibilities for applying small models to complex tasks and provides new insights for future model optimization.

Novelty

This study is the first to achieve GPT-4 level fact-checking performance on a small model through synthetic data generation. The method innovates in both data generation and model training, significantly enhancing model efficiency and accuracy.

Limitations

  • MiniCheck may underperform in handling very complex cross-document reasoning as it primarily relies on single-document fact-checking.
  • The quality of synthetic data generation directly impacts model performance and may require further optimization.

Future Work

Future research could explore applying MiniCheck to more complex cross-document reasoning tasks and further optimize synthetic data generation methods to enhance model adaptability and accuracy.

AI Executive Summary

In the field of natural language processing, recognizing whether the output of large language models is grounded in evidence is a crucial task. Existing methods often require significant computational resources to verify each generated piece, making the cost very high.

MiniCheck successfully achieves GPT-4 level performance on a small model by constructing synthetic training data, reducing costs by 400 times. This method trains models to perform sentence-level fact-checking and recognize information synthesis across sentences by generating challenging factual error instances.

Experimental results show that MiniCheck excels in the LLM-AggreFact benchmark, outperforming all systems of comparable size and achieving GPT-4 accuracy. This research is significant not only in academia but also offers a cost-effective fact-checking tool for the industry. Future research could explore applying MiniCheck to more complex cross-document reasoning tasks and further optimize synthetic data generation methods.

Deep Analysis

Background

In natural language processing, fact-checking is a key task involving retrieval-augmented generation, summarization, and document-grounded dialogue. Existing methods often rely on large language models for piece-by-piece verification, leading to high computational costs.

Core Problem

Existing fact-checking methods require multiple model calls to verify a single response, which is not only time-consuming but also costly. Reducing costs while maintaining accuracy is a pressing issue.

Innovation

MiniCheck achieves GPT-4 level fact-checking performance on a small model through synthetic data generation. This method generates challenging factual error instances to train models for sentence-level verification and cross-sentence information synthesis.

Methodology

  • �� Use GPT-4 to generate synthetic data simulating real factual error scenarios.
  • �� Fine-tune the Flan-T5 model on synthetic and standard entailment data.
  • �� Evaluate model performance on the LLM-AggreFact benchmark.

Experiments

Experiments used the LLM-AggreFact benchmark, which integrates 10 existing datasets covering both closed-book and grounded generation settings. MiniCheck was compared with other models on these datasets.

Results

MiniCheck excels in the LLM-AggreFact benchmark, outperforming all systems of comparable size and achieving GPT-4 accuracy. Experimental results show MiniCheck surpasses AlignScore on 6 out of 10 datasets, with an average improvement of 4.3%.

Applications

MiniCheck can be directly applied to tasks like retrieval-augmented generation, summarization, and document-grounded dialogue, significantly reducing the computational cost of fact-checking.

Limitations & Outlook

MiniCheck may underperform in handling very complex cross-document reasoning as it primarily relies on single-document fact-checking. The quality of synthetic data generation directly impacts model performance and may require further optimization.

Plain Language Accessible to non-experts

Imagine you're in a library trying to verify if a fact in a book is correct. You can check other books to confirm this fact. MiniCheck acts like a smart assistant that quickly finds relevant books and tells you if the fact is supported by evidence. By using synthetic data, MiniCheck can perform this task quickly and accurately without needing a lot of resources.

ELI14 Explained like you're 14

Imagine you're playing a game where you need to check if each task has evidence to support it. MiniCheck is like a super helper that quickly finds all the relevant evidence and tells you if the task is correct. It uses a smart way to generate data, so you don't need to spend a lot of time and effort checking each task.

Glossary

Large Language Model (LLM)

A large machine learning model capable of generating and understanding natural language, often used in NLP tasks.

Used in the study to generate and verify text accuracy.

Synthetic Data

Data created through artificial generation or simulation, used for training or testing machine learning models.

Used to train the MiniCheck model to enhance its fact-checking capabilities.

Fact-Checking

The process of verifying whether information is supported by evidence, often used to assess the accuracy of text generation.

The core task of MiniCheck is to perform efficient fact-checking.

GPT-4

The fourth generation of the Generative Pre-trained Transformer model developed by OpenAI, known for its powerful NLP capabilities.

Used to generate synthetic data for training the MiniCheck model.

LLM-AggreFact

A benchmark integrating multiple datasets for evaluating model fact-checking performance.

Used to assess the accuracy and efficiency of the MiniCheck model.

Open Questions Unanswered questions from this research

  • 1 How to improve MiniCheck's performance in cross-document reasoning needs further research.
  • 2 The impact of synthetic data generation quality on model performance requires more exploration.

Applications

Immediate Applications

Retrieval-Augmented Generation

MiniCheck can be used to improve the accuracy of retrieval-augmented generation tasks, reducing computational costs.

Summarization

Applying MiniCheck in summarization can ensure the accuracy and consistency of generated content.

Long-term Vision

Cross-Document Reasoning

In the future, MiniCheck could be extended to more complex cross-document reasoning tasks, further expanding its application scope.

Abstract

Recognizing if LLM output can be grounded in evidence is central to many tasks in NLP: retrieval-augmented generation, summarization, document-grounded dialogue, and more. Current approaches to this kind of fact-checking are based on verifying each piece of a model generation against potential evidence using an LLM. However, this process can be very computationally expensive, requiring many calls to a model to check a single response. In this work, we show how to build small fact-checking models that have GPT-4-level performance but for 400x lower cost. We do this by constructing synthetic training data with GPT-4, which involves creating realistic yet challenging instances of factual errors via a structured generation procedure. Training on this data teaches models to check each fact in the claim and recognize synthesis of information across sentences. For evaluation, we unify datasets from recent work on fact-checking and grounding LLM generations into a new benchmark, LLM-AggreFact. Our best system MiniCheck-FT5 (770M parameters) outperforms all systems of comparable size and reaches GPT-4 accuracy. We release LLM-AggreFact, code for data synthesis, and models.

cs.CL cs.AI