arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.02499cs.CLcs.AIcs.CVcs.LGstat.ML
收藏

AutoML-GPT: 基于GPT的自动机器学习

AutoML-GPT: Automatic Machine Learning with GPT

The University of Texas at Austin(德克萨斯大学奥斯汀分校)

Shujian Zhang, Chengyue Gong, Lemeng Wu, Xingchao Liu, Mingyuan Zhou

AI总结:

针对传统AI模型需人工调试架构与超参数的问题,提出AutoML-GPT框架,以GPT为桥梁连接各类AI模型,自动完成从数据处理到训练的全流程,在多领域任务中展现出通用性与有效性。

摘要:

AI任务涵盖广泛的领域与方向。尽管已有大量针对特定任务和应用设计的AI模型,但它们往往需要大量人力来寻找合适的模型架构、优化算法和超参数。近期,以ChatGPT为代表的大语言模型(LLM)在推理、理解和交互的多方面展现出卓越能力。为此,我们提出开发面向任务的提示词,自动利用LLM实现训练流程自动化。为落实这一构想,我们提出AutoML-GPT,它以GPT作为连接各类AI模型的桥梁,使用优化后的超参数动态训练模型。AutoML-GPT从模型卡和数据卡中动态获取用户需求,并生成对应的提示段落。最终,借助该提示段落,AutoML-GPT将自动开展从数据处理到模型架构设计、超参数调优,再到生成预测训练日志的全部实验。凭借GPT强大的语言能力和现有AI模型,AutoML-GPT能够处理跨多种任务和数据集的复杂AI任务,在计算机视觉、自然语言处理及其他高难度领域均取得了出色成果。大量实验与消融研究表明,我们的方法具有通用性、有效性,可为众多AI任务提供助力。

英文摘要:

AI tasks encompass a wide range of domains and fields. While numerous AI models have been designed for specific tasks and applications, they often require considerable human efforts in finding the right model architecture, optimization algorithm, and hyperparameters. Recent advances in large language models (LLMs) like ChatGPT show remarkable capabilities in various aspects of reasoning, comprehension, and interaction. Consequently, we propose developing task-oriented prompts and automatically utilizing LLMs to automate the training pipeline. To implement this concept, we present the AutoML-GPT, which employs GPT as the bridge to diverse AI models and dynamically trains models with optimized hyperparameters. AutoML-GPT dynamically takes user requests from the model and data cards and composes the corresponding prompt paragraph. Ultimately, with this prompt paragraph, AutoML-GPT will automatically conduct the experiments from data processing to model architecture, hyperparameter tuning, and predicted training log. By leveraging {\ours}'s robust language capabilities and the available AI models, AutoML-GPT can tackle numerous intricate AI tasks across various tasks and datasets. This approach achieves remarkable results in computer vision, natural language processing, and other challenging areas. Extensive experiments and ablation studies demonstrate that our method can be general, effective, and beneficial for many AI tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑