本文目录导读:

AutoML(自动机器学习)领域在近两年(2023-2024年)有了不少新进展,主要围绕大语言模型(LLM)的集成、更细粒度的自动特征工程、以及生产化部署的简化,以下是值得关注的一些新工具和更新:
深度融合LLM的AutoML工具
这是目前最热门的趋势,利用GPT-4、Claude等大模型作为智能体来驱动AutoML流程。
- Auto-GPT / BabyAGI 相关AutoML分支:虽然它们不是纯AutoML工具,但社区已开发出利用它们自主编写、调试和执行机器学习代码的插件,用户只需描述业务问题,AI自动负责数据清洗、模型选择、调参和报告生成。
- DataRobot 的 AI Assistant:DataRobot作为老牌商业化AutoML平台,现在融入了生成式AI,用户可以用自然语言询问模型性能、请求特征建议,甚至让AI解释模型决策,这在企业级应用中比较领先。
更高效、更强大的开源框架
开源领域不再仅限于传统的TPOT或Auto-sklearn,出现了针对特定场景(如时间序列、大规模深度学习)的专用工具。
-
AutoGluon (AWS):近期版本(v1.0+)更新显著,它主打多模态和集成学习的极端优化,新特性包括:
- Zero-Shot AutoML:无需大量自动搜索,即使用预设的高质量模型集成,也能在表格数据上碾压传统方法。
- 时间序列预测:原生支持多变量、多时间序列的自动预测,引入了深度学习模型(如Temporal Fusion Transformer)。
- 快速部署:一行代码即可将训练好的管道导出为Docker镜像或API端点。
-
FLAML (Microsoft):专注于极其高效的超参数搜索,它的新版本(v2.0+)亮点在于:
- AutoML for LLM微调:能自动找到最适合你数据的大语言模型(LLM)微调方法(如LoRA的rank值、学习率等),而不仅仅是传统ML模型。
- 预算感知:允许用户在训练时明确设定时间预算或金钱预算,算法会动态调整搜索策略,在预算内找到最优解。
-
PyCaret v3:它更像是一个低代码ML库而非纯粹的AutoML,最近更新强调了:
- 管道即代码:所有预处理、建模、部署步骤都封装在一个统一的Pipeline对象中,便于与MLOps工具(MLflow、DVC)集成。
- 支持GPU加速:原生支持RAPIDS cuML,极大加速了在GPU上的训练。
面向生产化和MLOps的AutoML
这类工具不再只追求“模型精度最高”,而是追求“从数据到上线”全流程的自动化和稳定性。
- H2O Driverless AI:虽然收费,但其功能在持续进化,新版本更侧重于可解释性和公平性自动检测,它可以自动生成合规报告,显示模型是否存在对特定群体的偏见。
- MLflow (核心更新):虽然MLflow不是AutoML本身,但它是管理AutoML流程的标准工具,新版本引入了MLflow Pipelines(后更名为MLflow Projects的增强版),允许用户以声明式YAML文件定义完整的AutoML工作流,包括数据验证、模型训练、注册和部署。
针对特定领域的新兴工具
- 时序预测:
- NeuralProphet:由Facebook Prophet的团队开发,结合了Prophet的易用性和深度学习的灵活性(如AR-Net),它内置了自动超参数搜索(基于Optuna)。
- 图机器学习(Graph ML):
- AutoGL:清华大学研发的自动图神经网络(Auto-GNN)工具,它能自动设计图结构、自动选择消息传递函数,极大降低了图神经网络的入门门槛。
如何选择?
| 你的需求 | 推荐尝试的新工具 |
|---|---|
| 小白用户,用自然语言指挥AI | Auto-GPT / DataRobot AI Assistant |
| 追求表格数据的极限精度 | AutoGluon (新版) |
| 资源有限,想快速找到好模型 | FLAML (新版, 预算感知能力出色) |
| 做时间序列预测 | AutoGluon (时序模块) 或 NeuralProphet |
| 想微调大模型(LLM) | FLAML (LLM自动微调) |
| 需要完整的上线解决方案 | PyCaret v3 (转成管道) + MLflow |
| 重视模型公平性和可解释性 | H2O Driverless AI |
建议:如果你是初学者或希望快速验证想法,可以优先试试 AutoGluon(精度高、接口简洁,一行代码出结果)或 FLAML(速度极快,支持新场景),如果希望探索大模型结合机器学习的新玩法,可以关注 Auto-GPT 相关的ML实验。