Multi-Task Inference: Can Large Language Models Follow Multiple Instructions at Once?

ICLR 2024 Workshop ME-FoMo Submission39 Authors

Published: 04 Mar 2024, Last Modified: 30 Apr 2024ME-FoMo 2024 PosterEveryoneRevisionsBibTeXCC BY 4.0
Keywords: multi-task inference, multi-processing, large language models, evaluation
TL;DR: We introduce a benchmark for testing the multiprocessing capabilities of large language models. Alongside, we show that the larger or proprietary models are capable of solving multiple instructions with only one inference call.
Abstract: Large language models (LLMs) are typically prompted to follow a single instruction per inference call. In this work, we analyze whether LLMs also hold the capability to handle multiple instructions simultaneously, denoted as Multi-Task Inference. For this purpose, we introduce the MTI Bench (Multi-Task Inference Benchmark), a comprehensive evaluation benchmark encompassing 5,000 instances across 25 tasks. Each task in the MTI Bench involves 2 to 3 sub-tasks. As expected, we first demonstrate that Multi-Task Inference reduces the total inference time by ×1.46 times in average since it does not require multiple inference calls. Interestingly, contrary to the expectation that LLMs would perform better when tasks are divided, we find that state-of-the-art LLMs, such as Llama-2-Chat-70B and GPT-4, show up to 7.3% and 12.4% improved performance with Multi-Task Inference compared to Single-Task Inference on the MTI BENCH. We release the MTI Bench dataset and our code at https://github.com/guijinSON/MTI-Bench.
Submission Number: 39
Loading