中文

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}
}