English

Learning to Order: Task Sequencing as In-Context Optimization

Machine Learning 2026-03-17 v1

Abstract

Task sequencing (TS) is one of the core open problems in Deep Learning, arising in a plethora of real-world domains, from robotic assembly lines to autonomous driving. Unfortunately, prior work has not convincingly demonstrated the generalization ability of meta-learned TS methods to solve new TS problems, given few initial demonstrations. In this paper, we demonstrate that deep neural networks can meta-learn over an infinite prior of synthetically generated TS problems and achieve a few-shot generalization. We meta-learn a transformer-based architecture over datasets of sequencing trajectories generated from a prior distribution that samples sequencing problems as paths in directed graphs. In a large-scale experiment, we provide ample empirical evidence that our meta-learned models discover optimal task sequences significantly quicker than non-meta-learned baselines.

Keywords

Cite

@article{arxiv.2603.14550,
  title  = {Learning to Order: Task Sequencing as In-Context Optimization},
  author = {Jan Kobiolka and Christian Frey and Arlind Kadra and Gresa Shala and Josif Grabocka},
  journal= {arXiv preprint arXiv:2603.14550},
  year   = {2026}
}

Comments

Under Review

R2 v1 2026-07-01T11:20:59.352Z