Automatic Configuration of LLM Post-Training Pipelines
Not on the map. Why?
Abstract (arXiv)
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-stage 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 consistently outperforms offline-only baselines and achieves comparable performance with the strongest online HPO baselines while using less than 10% of their computational cost.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 24 of its 33 references that are in the map. First posted on arXiv 19 Mar 2026.
Builds on
24 of 33 references.
References · 24
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,455
- 2023A Survey on Large Language Model based Autonomous Agentscited by 2,203
- 2022Large Language Models Encode Clinical Knowledgecited by 1,807
- 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,758
- 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,324
- 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
- 2016Hyperband: A Novel Bandit-Based Approach to Hyperparameter Optimizationcited by 688
12 more
- 2018BOHB: Robust and Efficient Hyperparameter Optimization at Scalecited by 466
- 2024Benchmarking Retrieval-Augmented Generation for Medicinecited by 326
- 2024HuatuoGPT-o1, Towards Medical Complex Reasoning with LLMscited by 257
- 2023MedCPT: Contrastive Pre-trained Transformers with Large-scale PubMed Search Logs for Zero-shot Biomedical Information Retrievalcited by 182
- 2024Benchmarking Large Language Models on Answering and Explaining Challenging Medical Questionscited by 106
- 2023PFNs4BO: In-Context Learning for Bayesian Optimizationcited by 70
- 2025Quagmires in SFT-RL Post-Training: When High SFT Scores Mislead and What to Use Insteadcited by 24
- 2025MedS3: Towards Medical Slow Thinking with Self-Evolved Soft Dual-sided Process Supervisioncited by 23
- 2023Quick-Tune: Quickly Learning Which Pretrained Model to Finetune and Howcited by 17
- 2023Hyperparameter Optimization for Large Language Model Instruction-Tuningcited by 17
- 2025Faster, Cheaper, Better: Multi-Objective Hyperparameter Optimization for LLM and RAG Systemscited by 11
- 2025LaMDAgent: An Autonomous Framework for Post-Training Pipeline Optimization via LLM Agentscited by 6
Built on by
Citing papers per year: 2026: 2.
2 papers.