Data Science with LLMs and Interpretable Models

Sebastian Bordt, Ben Lengerich, Harsha Nori, Rich Caruana
2/22/2024
cs.LGcs.CL

Official code: interpretml/TalkToEBM (72 stars, MIT, last commit 2024-07-05).

Abstract

Recent years have seen important advances in the building of interpretable models, machine learning models that are designed to be easily understood by humans. In this work, we show that large language models (LLMs) are remarkably good at working with interpretable models, too. In particular, we show that LLMs can describe, interpret, and debug Generalized Additive Models (GAMs). Combining the flexibility of LLMs with the breadth of statistical patterns accurately described by GAMs enables dataset summarization, question answering, and model critique. LLMs can also improve the interaction between domain experts and interpretable models, and generate hypotheses about the underlying phenomenon. We release \url{https://github.com/interpretml/TalkToEBM} as an open-source LLM-GAM interface.

View on arXivView PDF

Code Implementations(1)

MIT

A Natural Language Interface to Explainable Boosting Machines

7212Python, Jupyter NotebookJul 21, 20232 years agoMIT

Similar Papers

Cite this paper

@article{bordt2024data,
  title  = {Data Science with LLMs and Interpretable Models},
  author = {Sebastian Bordt and Ben Lengerich and Harsha Nori and Rich Caruana},
  year   = {2024},
  eprint = {2402.14474},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2402.14474v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2402.14474.svg)](https://codeofpaper.com/paper/2402.14474)

Discussion