Green Runner:一种从模型仓库高效选择模型的工具
软件工程
2023-05-29 v1 机器学习
摘要
深度学习模型已成为软件工程中不可或缺的组成部分,赋能图像字幕和文档生成等智能功能。然而,其流行引发了关于环境影响与模型选择效率低下的担忧。本文介绍 GreenRunnerGPT,一种基于特定用例高效选择深度学习模型的新工具。它采用大语言模型(LLM)为质量指标建议权重,以优化资源利用。该工具利用多臂老虎机框架针对目标数据集评估模型,考量权衡取舍。我们证明 GreenRunnerGPT 能够识别适配目标用例的模型,而避免暴力模型选择方法会产生的浪费性计算。
引用
@article{arxiv.2305.16849,
title = {Green Runner: A tool for efficient model selection from model repositories},
author = {Jai Kannan and Scott Barnett and Anj Simmons and Taylan Selvi and Luis Cruz},
journal= {arXiv preprint arXiv:2305.16849},
year = {2023}
}