Which Task Differences Matter? The Geometry of Optimal Shared Iteration
Abstract
A solver that reuses one learned coefficient schedule across many tasks can underperform because of insufficient training, a restrictive update rule, or missing task information, but its loss cannot identify the cause. For positive-definite quadratics, we separate these sources of excess risk and characterize which task differences change the optimal program, the residual polynomial the schedule implements. Task laws with different spectra and oracle risks can require the same program. Given the population risk of every depth- program (equivalently, moments through degree ), moment vectors sharing an optimal depth- program form affine fibers of dimension through each interior point. Across fibers, the leading oracle-relative cost of sharing a program is the squared radius of the smallest ball enclosing the linearized optimal programs in the local risk metric. Under this criterion, the minimax program is optimal for a least-favorable mixture of laws; we prove a sharp worst-case bound, attained by smooth positive densities, on how many laws this mixture needs. A momentum recurrence implements this program while preserving the optimal first steps shared by all laws in the family.
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.