The Geometry of Sequential Learning: Lie-Bracket Prediction of Transfer Order
Abstract
Lay Summary
AI models are often improved by training them on several datasets in sequence, such as general text, instruction data, and human preference data. The order of these datasets can change the final model, but trying every possible order quickly becomes impossible: with many datasets, the number of possible schedules becomes enormous. This paper introduces a way to predict good training orders without actually training the model in every order. The method takes a small number of measurements near the starting model and uses them to estimate which dataset should come before another, then combines those pairwise comparisons into a full schedule. Across language-model adaptation, instruction tuning, preference tuning, and many-dataset scheduling, the method often chooses orders that perform close to the best schedules found by much more expensive searches. This can make multi-stage model training cheaper and more systematic, while also helping practitioners choose orders that preserve important capabilities rather than relying on trial and error.