arXiv cs.AI· Channe Chwa, Xinle Wu, Yao Lu·· 14 小时前AI 评分31
AutoPipe：面向 LLM 后训练流水线的预算感知自动配置框架
Automatic Configuration of LLM Post-Training Pipelines
AI 导读
AutoPipe 是一个预算感知的两阶段框架，用于 LLM 后训练流水线的配置选择。它离线学习以数据集为条件的学习排序代理模型，在线用该引导加速贝叶斯优化并以高斯过程残差建模数据集差异，同时通过早停和学习预测器降低评估成本。在生物医学推理任务上，AutoPipe 以远低于全保真 HPO 基线的在线成本达到相当性能，并优于纯离线方法和早停 HPO 变体。
正文 · 原文
Abstract:LLM post-training pipelines that combine supervised fine-tuning and reinforcement learning are difficult to configure under realistic compute budgets: the configuration space is high-dimensional and heterogeneous, stages are strongly coupled, and each end-to-end evaluation is expensive. We propose AutoPipe, a budget-aware two-phase framework for configuration selection in LLM post-training. Offline, AutoPipe learns a dataset-conditioned learning-to-rank surrogate from historical runs, capturing within-dataset preferences and providing transferable guidance toward promising regions of the configuration space. Online, for a new dataset, AutoPipe uses the offline guidance to steer Bayesian optimization and models dataset-specific deviations with a Gaussian-process residual surrogate. To reduce evaluation cost, each trial is early-stopped and scored by a learned predictor that maps early training signals to a low-cost proxy for final post-training performance. Experiments on biomedical reasoning tasks show that AutoPipe achieves performance comparable to full-fidelity HPO baselines at a fraction of their online cost, while outperforming offline-only approaches and early-stopped HPO variants.
| Subjects: | Machine Learning (cs.LG); Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2603.18773 [cs.LG] |
| (or arXiv:2603.18773v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2603.18773 arXiv-issued DOI via DataCite |
Submission history
From: Channe Chwa [view email]
[v1]
Thu, 19 Mar 2026 11:26:56 UTC (107 KB)
[v2]
Sun, 25 Oct 2026 06:30:57 UTC (130 KB)
来源：arXiv cs.AI · arxiv.org