arXiv AI

A Risk Decomposition Framework for Pre-Hoc Fine-Tuning Prediction

arXiv:2606. 17649v1 Announce Type: cross Abstract: The high cost of fine-tuning LLMs poses a significant economic barrier; pre-hoc performance prediction offers a critical solution to substantially reduce this expense.

arXiv Machine Learning
Jun 11

Calibrating Decision Robustness via Inverse Conformal Risk Control

arXiv:2510. 07750v3 Announce Type: replace-cross Abstract: Robust optimization safeguards decisions against uncertainty by optimizing against worst-case scenarios, yet their effectiveness hinges on a prespecified robustness level that is often chosen ad hoc, leading to either insufficient protection or overly conservative and costly solutions.

By Wenbin Zhou, Shixiang Zhu
arXiv Machine Learning
Jul 27

Smart predict-then-robustly-optimize

arXiv:2607. 21773v1 Announce Type: new Abstract: In this paper, we propose and study a robust variant of the smart predict-then-optimize approach that accounts for prediction shifts due to disturbance in the covariate feature space.

By Aakil Caunhye, Xuefei Lu, Belen Martin-Barragan
arXiv Machine Learning
Aug 27

Forecasting Multiple Observables with SCROLL: Score-Trained Uncertainty for Stochastic Dynamics

The paper introduces SCROLL, a method for forecasting multiple observables in stochastic dynamical systems by composing each observable’s likelihood into per‑task free‑routed last‑layer beliefs on a shared backbone. This approach learns unit‑dependent loss scaling directly from data, enabling accurate predictive variance estimation without separate tuning. Experiments on the Ornstein–Uhlenbeck process, stochastic Lorenz‑63, and real air‑quality data show that SCROLL recovers analytic kernels, achieves superior negative log‑likelihood on state and regime tasks, and maintains calibration while reducing hyper‑parameter search costs.

By Pavel Prochazka
arXiv AI
Jun 17

TuneAhead: Predicting Fine-tuning Performance Before Full Training Begins

arXiv:2606. 17660v1 Announce Type: cross Abstract: Fine-tuning large language models (LLMs) is compute-intensive and error-prone: model performance depends sensitively on data quality and hyperparameter choices, and na\"ive runs can even degrade model performance.

By Yuxiang Luo, Haonan Long, Chen Wang, Qiqi Duan, Xiaotian Lin, Yanwei Xu, Yuyu Luo, Weikai Yang, Nan Tang
arXiv Machine Learning
Sep 23

Statistical Gains from Looped Estimation under Parameter Budgets

The paper investigates whether a looped estimator—one that repeatedly applies a single fitted operator with shared parameters—can enhance statistical accuracy while staying within a fixed parameter budget. It establishes upper and lower bounds on squared Hellinger risk for looped sieve maximum likelihood and compares them to the untied counterpart, revealing a tradeoff between parameter sharing, iteration count, and accuracy. For models with known H"older smoothness, looped residual feedforward networks and a post‑layer‑normalized Transformer achieve minimax polynomial rates with a fixed number of bounded real parameters, and under certain conditions the looped estimator’s worst‑case risk vanishes as sample size grows, outperforming the untied approach.

By Xinyu Tian, Xiaotong Shen