Saved in:
Bibliographic Details
Main Authors: Vasilevski, Kirill, Lin, Dayi, Hassan, Ahmed E.
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2411.09837
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916770807283712
author Vasilevski, Kirill
Lin, Dayi
Hassan, Ahmed E.
author_facet Vasilevski, Kirill
Lin, Dayi
Hassan, Ahmed E.
contents To balance the quality and inference cost of a Foundation Model (FM, such as large language models (LLMs)) powered software, people often opt to train a routing model that routes requests to FMs with different sizes and capabilities. Existing routing models rely on learning the optimal routing decision from carefully curated data, require complex computations to be updated, and do not consider the potential evolution of weaker FMs. In this paper, we propose Real-time Adaptive Routing (RAR), an approach to continuously adapt FM routing decisions while using guided in-context learning to enhance the capabilities of weaker FM. The goal is to reduce reliance on stronger, more expensive FMs. We evaluate our approach on different subsets of the popular MMLU benchmark. Over time, our approach routes 50.2% fewer requests to computationally expensive models while maintaining around 90.5% of the general response quality. In addition, the guides generated from stronger models have shown intra-domain generalization and led to a better quality of responses compared to an equivalent approach with a standalone weaker FM.
format Preprint
id arxiv_https___arxiv_org_abs_2411_09837
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Real-time Adapting Routing (RAR): Improving Efficiency Through Continuous Learning in Software Powered by Layered Foundation Models
Vasilevski, Kirill
Lin, Dayi
Hassan, Ahmed E.
Machine Learning
Artificial Intelligence
Multiagent Systems
To balance the quality and inference cost of a Foundation Model (FM, such as large language models (LLMs)) powered software, people often opt to train a routing model that routes requests to FMs with different sizes and capabilities. Existing routing models rely on learning the optimal routing decision from carefully curated data, require complex computations to be updated, and do not consider the potential evolution of weaker FMs. In this paper, we propose Real-time Adaptive Routing (RAR), an approach to continuously adapt FM routing decisions while using guided in-context learning to enhance the capabilities of weaker FM. The goal is to reduce reliance on stronger, more expensive FMs. We evaluate our approach on different subsets of the popular MMLU benchmark. Over time, our approach routes 50.2% fewer requests to computationally expensive models while maintaining around 90.5% of the general response quality. In addition, the guides generated from stronger models have shown intra-domain generalization and led to a better quality of responses compared to an equivalent approach with a standalone weaker FM.
title Real-time Adapting Routing (RAR): Improving Efficiency Through Continuous Learning in Software Powered by Layered Foundation Models
topic Machine Learning
Artificial Intelligence
Multiagent Systems
url https://arxiv.org/abs/2411.09837