TaskLAMA: Probing the Complex Task Understanding of Language Models

Abstract

Structured Complex Task Decomposition (SCTD) is the problem of breaking down a complex real-world task (such as planning a wedding) into a directed acyclic graph over individual steps that contribute to achieving the task, with edges specifying temporal dependencies between steps. SCTD is an important component of assistive planning tools, and a challenge for commonsense reasoning systems. We probe how accurately SCTD can be done with the knowledge extracted from pre-trained Large Language Models (LLMs). We introduce a new high-quality human-annotated dataset for this problem and novel metrics to fairly assess performance of LLMs against several baselines. Our experiments reveal that LLMs are able to decompose complex tasks into individual steps effectively, with a relative improvement of 15% to 280% over the best baseline. We also propose a number of approaches to further improve their performance, with a relative improvement of 7% to 37%. However, we find that LLMs still struggle to predict pairwise temporal dependencies, which reveals a gap in their understanding of complex tasks.

Cite

Text

Yuan et al. "TaskLAMA: Probing the Complex Task Understanding of Language Models." AAAI Conference on Artificial Intelligence, 2024. doi:10.1609/AAAI.V38I17.29918

Markdown

[Yuan et al. "TaskLAMA: Probing the Complex Task Understanding of Language Models." AAAI Conference on Artificial Intelligence, 2024.](https://mlanthology.org/aaai/2024/yuan2024aaai-tasklama/) doi:10.1609/AAAI.V38I17.29918

BibTeX

@inproceedings{yuan2024aaai-tasklama,
  title     = {{TaskLAMA: Probing the Complex Task Understanding of Language Models}},
  author    = {Yuan, Quan and Kazemi, Mehran and Xu, Xin and Noble, Isaac and Imbrasaite, Vaiva and Ramachandran, Deepak},
  booktitle = {AAAI Conference on Artificial Intelligence},
  year      = {2024},
  pages     = {19468-19476},
  doi       = {10.1609/AAAI.V38I17.29918},
  url       = {https://mlanthology.org/aaai/2024/yuan2024aaai-tasklama/}
}