English

"Oh LLM, I'm Asking Thee, Please Give Me a Decision Tree": Zero-Shot Decision Tree Induction and Embedding with Large Language Models

Artificial Intelligence 2025-05-28 v2 Computation and Language Machine Learning

Abstract

Large language models (LLMs) provide powerful means to leverage prior knowledge for predictive modeling when data is limited. In this work, we demonstrate how LLMs can use their compressed world knowledge to generate intrinsically interpretable machine learning models, i.e., decision trees, without any training data. We find that these zero-shot decision trees can even surpass data-driven trees on some small-sized tabular datasets and that embeddings derived from these trees perform better than data-driven tree-based embeddings on average. Our decision tree induction and embedding approaches can therefore serve as new knowledge-driven baselines for data-driven machine learning methods in the low-data regime. Furthermore, they offer ways to harness the rich world knowledge within LLMs for tabular machine learning tasks. Our code and results are available at https://github.com/ml-lab-htw/llm-trees.

Keywords

Cite

@article{arxiv.2409.18594,
  title  = {"Oh LLM, I'm Asking Thee, Please Give Me a Decision Tree": Zero-Shot Decision Tree Induction and Embedding with Large Language Models},
  author = {Ricardo Knauer and Mario Koddenbrock and Raphael Wallsberger and Nicholas M. Brisson and Georg N. Duda and Deborah Falla and David W. Evans and Erik Rodner},
  journal= {arXiv preprint arXiv:2409.18594},
  year   = {2025}
}

Comments

KDD 2025 Research Track