Towards Logically Consistent Language Models via Probabilistic Reasoning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Calanzone, Diego, Teso, Stefano, Vergari, Antonio
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909175783620608
author Calanzone, Diego
Teso, Stefano
Vergari, Antonio
author_facet Calanzone, Diego
Teso, Stefano
Vergari, Antonio
contents Large language models (LLMs) are a promising venue for natural language understanding and generation tasks. However, current LLMs are far from reliable: they are prone to generate non-factual information and, more crucially, to contradict themselves when prompted to reason about beliefs of the world. These problems are currently addressed with large scale fine-tuning or by delegating consistent reasoning to external tools. In this work, we strive for a middle ground and introduce a training objective based on principled probabilistic reasoning that teaches a LLM to be consistent with external knowledge in the form of a set of facts and rules. Fine-tuning with our loss on a limited set of facts enables our LLMs to be more logically consistent than previous baselines and allows them to extrapolate to unseen but semantically similar factual knowledge more systematically.
format Preprint
id arxiv_https___arxiv_org_abs_2404_12843
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Towards Logically Consistent Language Models via Probabilistic Reasoning
Calanzone, Diego
Teso, Stefano
Vergari, Antonio
Machine Learning
Computation and Language
Large language models (LLMs) are a promising venue for natural language understanding and generation tasks. However, current LLMs are far from reliable: they are prone to generate non-factual information and, more crucially, to contradict themselves when prompted to reason about beliefs of the world. These problems are currently addressed with large scale fine-tuning or by delegating consistent reasoning to external tools. In this work, we strive for a middle ground and introduce a training objective based on principled probabilistic reasoning that teaches a LLM to be consistent with external knowledge in the form of a set of facts and rules. Fine-tuning with our loss on a limited set of facts enables our LLMs to be more logically consistent than previous baselines and allows them to extrapolate to unseen but semantically similar factual knowledge more systematically.
title Towards Logically Consistent Language Models via Probabilistic Reasoning
topic Machine Learning
Computation and Language
url https://arxiv.org/abs/2404.12843