AQA: Adaptive Question Answering in a Society of LLMs via Contextual Multi-Armed Bandit

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Hoveyda, Mohanna, de Vries, Arjen P., de Rijke, Maarten, Oosterhuis, Harrie, Hasibi, Faegheh
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912040945188864
author Hoveyda, Mohanna
de Vries, Arjen P.
de Rijke, Maarten
Oosterhuis, Harrie
Hasibi, Faegheh
author_facet Hoveyda, Mohanna
de Vries, Arjen P.
de Rijke, Maarten
Oosterhuis, Harrie
Hasibi, Faegheh
contents In question answering (QA), different questions can be effectively addressed with different answering strategies. Some require a simple lookup, while others need complex, multi-step reasoning to be answered adequately. This observation motivates the development of a dynamic method that adaptively selects the most suitable QA strategy for each question, enabling more efficient and effective systems capable of addressing a broader range of question types. To this aim, we build on recent advances in the orchestration of multiple large language models (LLMs) and formulate adaptive QA as a dynamic orchestration challenge. We define this as a contextual multi-armed bandit problem, where the context is defined by the characteristics of the incoming question and the action space consists of potential communication graph configurations among the LLM agents. We then train a linear upper confidence bound model to learn an optimal mapping between different question types and their corresponding optimal multi-LLM communication graph representation. Our experiments show that the proposed solution is viable for adaptive orchestration of a QA system with multiple modules, as it combines the superior performance of more complex strategies while avoiding their costs when simpler strategies suffice.
format Preprint
id arxiv_https___arxiv_org_abs_2409_13447
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle AQA: Adaptive Question Answering in a Society of LLMs via Contextual Multi-Armed Bandit
Hoveyda, Mohanna
de Vries, Arjen P.
de Rijke, Maarten
Oosterhuis, Harrie
Hasibi, Faegheh
Computation and Language
In question answering (QA), different questions can be effectively addressed with different answering strategies. Some require a simple lookup, while others need complex, multi-step reasoning to be answered adequately. This observation motivates the development of a dynamic method that adaptively selects the most suitable QA strategy for each question, enabling more efficient and effective systems capable of addressing a broader range of question types. To this aim, we build on recent advances in the orchestration of multiple large language models (LLMs) and formulate adaptive QA as a dynamic orchestration challenge. We define this as a contextual multi-armed bandit problem, where the context is defined by the characteristics of the incoming question and the action space consists of potential communication graph configurations among the LLM agents. We then train a linear upper confidence bound model to learn an optimal mapping between different question types and their corresponding optimal multi-LLM communication graph representation. Our experiments show that the proposed solution is viable for adaptive orchestration of a QA system with multiple modules, as it combines the superior performance of more complex strategies while avoiding their costs when simpler strategies suffice.
title AQA: Adaptive Question Answering in a Society of LLMs via Contextual Multi-Armed Bandit
topic Computation and Language
url https://arxiv.org/abs/2409.13447