Evolutionary Multi-Task Optimization for LLM-Guided Program Discovery
summary
The gist
The gist: EMO-STA, a shared-then-adapt framework, improves over matched-compute single-task evolution by evolving a reusable program archive across related tasks and then adapting selected candidates
In short
EMO-STA is a two-stage framework for discovering reusable programs across related tasks using LLMs. It first evolves a shared archive of executable programs based on an aggregated family objective, then adapts these shared candidates to individual target tasks. This method outperforms single-task evolution by leveraging shared structure, leading to more generalizable and robust algorithms.
Key concepts
- Evolutionary Multi-Task Optimization (EMO)
- This is the overall approach where an evolutionary search process is applied not just to one task, but across a whole family of related tasks simultaneously. The goal is to find programs that are good at many related problems by optimizing them against a combined objective that rewards generalizability.
- Shared Evolution
- The first phase of the framework where the system optimizes a pool of executable programs against an aggregated objective function. This process favors discovering core, reusable program structures common across all tasks in the family, creating a 'shared archive' of potential candidates.
- Adaptation Evolution
- The second phase where selected programs from the shared archive are specialized for each individual target task. The system adapts these general candidates to perform well on specific problems, using strategies like warm-starting or selecting the best performing candidate for a given task.
Terminology used across episodes
This episode discusses
- Evolutionary Multi-Task Optimization for LLM-Guided Program Discovery · Paper Radio
- GIFT-Eval: A Benchmark For General Time Series Forecasting Model Evaluation
- The Surprising Effectiveness of Test-Time Training for Few-Shot Learning
- Chronos-2: From Univariate to Universal Forecasting
- Promptbreeder: Self-Referential Self-Improvement Via Prompt Evolution
- Practical Transfer Learning for Bayesian Optimization
- Seeding the Initial Population of Multi-Objective Evolutionary Algorithms: A Computational Study
- Test-Time Training Provably Improves Transformers as In-context Learners
- DeltaEvolve: Accelerating Scientific Discovery through Momentum-Driven Evolution
- NAACL2025 Tutorial: Adaptation of Large Language Models
- Can Language Models Discover Scaling Laws?
- EvoX: Meta-Evolution for Automated Discovery
- AlphaEvolve: A coding agent for scientific and algorithmic discovery
- Warm Starting Bayesian Optimization
- PEFT A2Z: Parameter-Efficient Fine-Tuning Survey for Large Language and Vision Models
- An Overview of Multi-Task Learning in Deep Neural Networks
- ThetaEvolve: Test-time Learning on Open Problems
- Beyond Model Adaptation at Test Time: A Survey
- LLMs and the Abstraction and Reasoning Corpus: Successes, Failures, and the Importance of Object-based Representations
The paper
Evolutionary Multi-Task Optimization for LLM-Guided Program Discovery · Read on arXiv
University of Michigan - Ann Arbor · University of California San Diego
Recent LLM-guided evolutionary search methods have shown that iterative program mutation can discover strong algorithms, but they typically optimize each task independently, even when related tasks share reusable structure. We study evolutionary multitask optimization (EMO) for LLM-guided program discovery and propose EMO-STA (Shared-Then-Adapt), a two-stage framework that first evolves a shared archive of executable programs across a task family and then adapts selected shared candidates to each target task. Within EMO-STA, we explore multiple adaptation strategies, including warm-starting from the shared archive, adapting the best average shared program, and adapting the shared program that performs best on each target task. Across eight task families spanning continuous optimization, geometric construction, modeling, and algorithmic optimization, EMO-STA improves over single-task evolution under matched evolutionary-iteration budgets in most settings, with STA Best-Local providing the strongest in-distribution adaptation and STA Best-Shared yielding robust transfer to unseen tasks. Iteration-allocation experiments show that devoting a substantial fraction of the family-level iteration budget to shared evolution is effective, with roughly balanced shared and aggregate adaptation budgets often yielding the best performance. ARC and time-series feature-engineering case studies further suggest that shared evolution can mitigate overfitting in low-evidence settings by favoring programs that perform well across related tasks rather than exploiting brittle task-specific artifacts.
Transcript
Introduction to the show: ident: AI Radio. Generated commentary on the latest Artificial Intelligence papers.
Tom: I'm Tom.
Jane: And I'm Jane, here with Lu, Meng and our in-house large language model, Lalam.
Tom: Today's paper: "Evolutionary Multi-Task Optimization for LLM-Guided Program Discovery".
Jane: The gist: EMO-STA, a shared-then-adapt framework, improves over matched-compute single-task evolution by evolving a reusable program archive across related tasks and then adapting selected candidates to individual tasks.
Tom: First, who's behind it and why it matters.
Paper summary: Tom: So we're looking at this paper called Evolutionary Multi-Task Optimization for LLM-Guided Program Discovery. Basically, it tackles the problem where you try to find algorithms using evolutionary search, but you usually optimize each task completely separately.
Jane: Right? Even when those tasks are related and share the same underlying structure, they don't benefit from that shared knowledge in the optimization process itself.
Tom: Exactly. The core idea here is introducing EMO-STA, which stands for Shared-Then-Adapt, a two-stage framework. It first evolves a shared archive of executable programs across an entire family of tasks and then adapts selected candidates to each specific target task individually.
Lu: What makes this interesting is that it works whenever related tasks can be expressed through a unified program interface, which they enforce by defining one entry point function and prompting the LLM to implement that format.
Meng: So, instead of running eight separate evolutions for eight different tasks, you evolve one shared base program and then tweak it for each one? That sounds like it could save a ton of compute time.
Jane: It's about finding programs that are inherently generalizable across the whole family first, and then using those good candidates as starting points for the specific tasks. The paper claims this approach improves over single-task evolution when you have a matched evolutionary-iteration budget in most settings.
Tom: And they found some strong results across eight task families—they cover continuous optimization, geometric construction, modeling, and algorithmic optimization. They specifically point out that STA Best-Local is the strongest strategy for in-distribution adaptation across twenty-three out of forty cells.
Lalam: I see what they mean; it seems like this shared evolution introduces a structural bias to the search space itself, which should lead to better programs overall.
Meng: But what about when we try to run this at scale? The paper mentions that allocating a substantial fraction of your family-level iteration budget to shared evolution is effective, and roughly balanced budgets between shared and aggregate adaptation often yield the best performance.
Jane: And they even look at low-evidence settings, like ARC tasks and time-series feature engineering, where you can see that this shared evolution helps mitigate overfitting by favoring programs that work across related tasks instead of exploiting task-specific artifacts.
Tom: It’s suggesting that the structural bias from the shared evolution is actually a net gain in performance for finding good algorithms. This paper is called Evolutionary Multi-Task Optimization for LLM-Guided Program Discovery and it proposes EMO-STA as the solution to independent task optimization.
Conclusion: Tom: So, looking at this paper, Evolutionary Multi-Task Optimization for LLM-Guided Program Discovery, the main thing is that they’ve successfully converted shared structure across related problems into stronger and more generalizable LLM-discovered programs.
Jane: It moves away from the old way of optimizing every task in isolation and instead uses a two-stage process: first, evolve something common for the family, then specialize it for each target.
Lu: The implication here is that this framework is particularly useful when tasks share a meaningful program structure, and they suggest that we should look into automatically grouping related tasks whenever possible.
Meng: If we're thinking about deployment, the idea of using a shared archive to generate candidates means we have a more robust starting population for our local adaptation phase.
Lalam: I think what this paper really does is organize related discovery tasks around a shared program interface, which is how it achieves that transferability across different problem types.
Tom: It's about using the search budget to find better programs by exploiting the relationships between tasks rather than just brute-forcing each one individually. That’s the essence of EMO-STA.
More episodes
- 2507.03411-Tourist arrival forecasting with game-theoretic social influence weighting and deep learning
- 2508.16531-Quality control in sublinear time: a case study via random graphs
- 2509.26226-Thinking-Free Policy Initialization Makes Distilled Reasoning Models More Effective and Efficient Reasoners
- 2511.03068-Graph Homomorphism Distortion: A structural (pseudo)-metric for attributed graphs
- 2512.02004-AlignSAE: Concept-Aligned Sparse Autoencoders
- 2601.10471-DeFlow: Decoupling Behavior-Prior Modeling and Value Maximization for Offline Policy Extraction
- 2602.07640-TASTE: Task-Aware Out-of-Distribution Detection via Stein Operators
- 2603.11611-Fractional Rotation, Full Potential? Investigating Performance and Convergence of Partial RoPE
- 2602.06603-The impact of temporal resampling on the safety of clinical reinforcement learning: A simulated trial
- 2602.08755-Align and Adapt: Multimodal Multiview Human Activity Recognition under Arbitrary View Combinations