LLM4Causal: Democratized Causal Tools for Everyone via Large Language Model

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Jiang, Haitao, Ge, Lin, Gao, Yuhe, Wang, Jianian, Song, Rui
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912088837849088
author Jiang, Haitao
Ge, Lin
Gao, Yuhe
Wang, Jianian
Song, Rui
author_facet Jiang, Haitao
Ge, Lin
Gao, Yuhe
Wang, Jianian
Song, Rui
contents Large Language Models (LLMs) have shown their success in language understanding and reasoning on general topics. However, their capability to perform inference based on user-specified structured data and knowledge in corpus-rare concepts, such as causal decision-making is still limited. In this work, we explore the possibility of fine-tuning an open-sourced LLM into LLM4Causal, which can identify the causal task, execute a corresponding function, and interpret its numerical results based on users' queries and the provided dataset. Meanwhile, we propose a data generation process for more controllable GPT prompting and present two instruction-tuning datasets: (1) Causal-Retrieval-Bench for causal problem identification and input parameter extraction for causal function calling and (2) Causal-Interpret-Bench for in-context causal interpretation. By conducting end-to-end evaluations and two ablation studies, we showed that LLM4Causal can deliver end-to-end solutions for causal problems and provide easy-to-understand answers, which significantly outperforms the baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2312_17122
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle LLM4Causal: Democratized Causal Tools for Everyone via Large Language Model
Jiang, Haitao
Ge, Lin
Gao, Yuhe
Wang, Jianian
Song, Rui
Computation and Language
Artificial Intelligence
Machine Learning
Large Language Models (LLMs) have shown their success in language understanding and reasoning on general topics. However, their capability to perform inference based on user-specified structured data and knowledge in corpus-rare concepts, such as causal decision-making is still limited. In this work, we explore the possibility of fine-tuning an open-sourced LLM into LLM4Causal, which can identify the causal task, execute a corresponding function, and interpret its numerical results based on users' queries and the provided dataset. Meanwhile, we propose a data generation process for more controllable GPT prompting and present two instruction-tuning datasets: (1) Causal-Retrieval-Bench for causal problem identification and input parameter extraction for causal function calling and (2) Causal-Interpret-Bench for in-context causal interpretation. By conducting end-to-end evaluations and two ablation studies, we showed that LLM4Causal can deliver end-to-end solutions for causal problems and provide easy-to-understand answers, which significantly outperforms the baselines.
title LLM4Causal: Democratized Causal Tools for Everyone via Large Language Model
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2312.17122