Command Palette
Search for a command to run...
Frontis-MA1: 기계 학습 엔지니어링에서 재귀적 자기 개선을 위한 AI4AI 모델 훈련
Frontis-MA1: 기계 학습 엔지니어링에서 재귀적 자기 개선을 위한 AI4AI 모델 훈련
초록
재귀적 자기 개선(RSI)은 AI를 구축하는 과정을 개선하는 AI 시스템, 즉 AI4AI를 필요로 하며, 기계 학습 엔지니어링(MLE)은 이 능력을 연구하기 위한 구체적이고 실행 가능한 테스트베드를 제공한다. 본 논문은 MLE에서 RSI 연구를 위한 개방형 풀스택 시스템인 OpenMLE을 소개한다. 이 시스템은 실행 피드백이 포함된 검증 가능한 작업 환경(OpenMLE-Gym), 연산자 학습(OpenMLE-ERL), 그리고 장기 탐색(OpenMLE-Evo)을 포괄한다. 이 스택을 기반으로 우리는 Frontis-MA1-35B를 MLE용 메타 진화 에이전트로 사후 훈련하여, 네 가지 원자적 프로그램 진화 연산자(초안 작성, 개선, 디버깅, 교차)를 중심으로 사후 훈련과 추론을 정렬한다. 동일한 연산자들을 모든 평가 벤치마크에 대해 중복 제거된 데이터로 실행 기반 지도 미세 조정(SFT) 및 강화 학습(RL)을 통해 훈련한 후, 장기 탐색으로 구성하여 학습과 진화를 단일 루프로 결합한다. RTX 4090 1대에서 12GB VRAM 제한 하에 작업당 12시간 예산으로 MLE-Bench Lite에서 평가한 결과, Frontis-MA1-35B는 OpenMLE-Evo를 통해 기본 모델 대비 메달 평균을 39.39%에서 60.61%로 향상시켰으며, OpenMLE-Evo-Max(벤치마크 독립적 경험 사전 정보 및 비동기 탐색)를 사용하여 71.21%에 도달하여 GPT-5.5 + Codex를 능가하고 GPT-5.6 Sol 및 2.8T Kimi K3에 근접했다. 별도로 확보된 NatureBench Lite에서도 두 구성 요소 모두 전이되었다. 프레임워크를 고정하고 훈련된 모델로 교체했을 때 Match-SOTA가 50%에서 70%로 상승했으며, 모델을 고정하고 OpenMLE-Evo로 교체했을 때는 20%에서 50%로 상승했다. 우리는 실행 가능한 AI4AI 및 RSI 연구의 재현 가능성을 위해 모델 가중치와 전체 OpenMLE 스택을 공개한다.
One-sentence Summary
Researchers from Horizon Research, Frontis.AI, and Tsinghua University introduce Frontis-MA1-35B, a meta-evolution agent that posttrains four atomic program-evolution operators via execution-grounded SFT and RL, composes them into long-horizon search, and boosts Medal Average on MLE-Bench Lite from 39.39% to 71.21% with OpenMLE-Evo-Max, surpassing GPT-5.5+Codex and approaching GPT-5.6 Sol and Kimi K3, while also transferring to NatureBench Lite.
Key Contributions
- OpenMLE is an open full-stack system for meta-evolution research in machine learning engineering. It provides verifiable task environments with execution feedback, execution-grounded operator learning via supervised fine-tuning and reinforcement learning, and long-horizon evolutionary search guided by structured experience.
- Frontis-MA1-35B is post-trained as a meta-evolution agent by aligning training and inference around four atomic program-evolution operators (Draft, Improve, Debug, Crossover) trained with execution-grounded SFT and RL on data deduplicated against evaluation benchmarks, then composed into long-horizon search, coupling learning and evolution.
- Combining the trained model with OpenMLE-Evo improves Medal Average from 39.39% to 60.61% on MLE-Bench Lite, and with OpenMLE-Evo-Max reaches 71.21%. Both the model and the search framework transfer to NatureBench Lite, where the model raises Match-SOTA from 50% to 70% and the search framework raises it from 20% to 50%.
Introduction
The authors address the challenge of building AI systems that can autonomously improve other AI systems, a direction known as AI for AI (AI4AI) with the long-term goal of recursive self-improvement. Machine learning engineering (MLE) serves as a concrete testbed, where an agent must iteratively develop ML solutions under delayed, noisy execution feedback. Prior work advances MLE agents through evolutionary search, task environments, or execution-grounded post-training, but no existing public system jointly spans scalable task construction, agent training from execution feedback, and a long-horizon evolutionary harness in a reproducible workflow. The authors introduce OpenMLE, an open full-stack solution that unifies these components: OpenMLE-Gym provides quality-gated tasks and structured feedback, OpenMLE-ERL trains reusable program-transformation operators via supervised and reinforcement learning, and OpenMLE-Evo composes those operators into experience-guided search. This integrated stack enables training a meta-evolution agent that serves as the variation engine of its own evolutionary harness, demonstrating significant gains on MLE benchmarks and releasing all artifacts for broader study.
Dataset
The authors construct OpenMLE-Gym, a suite of 5,758 executable machine learning engineering tasks designed for post-training, search, and evaluation of LLM agents. Each task is a verifiable environment that couples a problem description, training data, hidden test labels, and a metric script into a standardized package. The dataset is built from three complementary sources:
- Curated Anchors (156 tasks): Manually selected from existing papers and benchmarks, offering the highest design confidence but limited scale. Original assets are processed directly into executable packages.
- Kaggle Datasets (3,362 tasks): Generated at scale using the MLE-Smith dataset-to-task pipeline, then filtered with package-level quality control. Objectives are automatically induced, so quality can vary.
- Kaggle Competitions (2,240 tasks): Built through a custom crawling and construction pipeline that downloads competition files, extracts schemas and metadata from Meta Kaggle, and generates task descriptions, data splits, and evaluation scripts. Competitions overlapping with MLE-Bench are excluded to preserve evaluation integrity.
All tasks share a unified layout: raw assets stay in raw/, agent-visible training data, test inputs, and a sample submission reside in data/public/, hidden answers are isolated in data/private/, and a metric.py script validates predictions and returns a scalar score. For competition-derived tasks, the automated framework uses local evidence (schemas, data types, dimensions, sampled rows) together with Meta Kaggle records to produce prepare.py, which deterministically splits labeled data, separates public inputs from private answers, and creates a schema-compatible sample submission. The package is then validated by executing the preparation code and scoring the sample submission; unbuildable packages or those that fail to produce a valid scalar metric are discarded.
A subsequent LLM-based quality filter inspects each task package (description, raw files, processing script, outputs, data samples) and returns structured judgments on five dimensions: task validity, data sufficiency, raw-data usage, task complexity, and data quality. Only metric-valid tasks that receive the strictest recommendation are retained, removing degenerate targets, data leakage, annotation errors, and malformed processing. For the competition branch, this semantic gate is applied after leaderboard-length screening, MLE-Bench overlap removal, and licensing checks.
The final dataset spans tabular, text, time-series, image, and other modalities (11% are multimodal), with classification and regression accounting for 87% of tasks. The authors use these environments to provide agents with verifiable feedback: an agent reads the public training data, produces predictions on the test inputs, and receives a scalar reward from metric.py, enabling post-training, test-time search, and recursive self-improvement loops.
Method
The authors construct OpenMLE-Gym, a unified executable task suite, by mapping three source-specific paths into a shared executable task package. As shown in the figure below, the curation hierarchy balances quality and scale across Curated Anchors, Kaggle Competitions, and Kaggle Datasets. For the competition-derived branch, an automated framework converts a Kaggle competition slug into a standardized executable package, applying a strict semantic quality gate after executability validation. The resulting executable task format separates agent-visible public inputs from hidden private answers and includes executable utilities to ensure reproducible scoring.
To train the model for evolutionary inference, the authors separate local skills from the search algorithm by training a compact set of reusable program-transformation operators: Draft, Improve, Debug, and Crossover. The training process consists of an execution-grounded supervised warm start followed by reinforcement learning.
For the supervised warm start, the authors employ two sampling paths. The parallel path independently samples and executes complete Draft solutions, while the evolutionary path applies Improve, Debug, and Crossover over executed programs to retain useful steps from high-quality local trajectory segments. Collection stops based on a budget-adaptive rule, forming a large-scale SFT corpus.
Building on this warm start, the reinforcement learning phase focuses on making heterogeneous outcomes comparable and concentrating the learning signal. The authors first define a bounded base reward and then derive tighter adaptive bounds from the task historical on-policy score frontier to preserve resolution. To further amplify the reward gaps near the top of each rollout group, an entropic advantage is applied, directing substantially more learning signal to the best candidates. As illustrated in the figure below, combining entropic weighting with adaptive bounds yields a stronger smoothed Group Best Reward trajectory compared to previous reward constructions.
For test-time scaling, the authors introduce OpenMLE-Evo, an experience-driven search harness that converts execution outcomes into reusable evidence. The search process maintains a persistent representation of prior attempts through structured experience accumulation. Each evaluated node generates a node-level experience card capturing provenance and performance, which are aggregated into a task-global experience board.
Parent selection is guided by a multi-factor utility combining normalized validation score, positive improvement over the strongest parent, and method-family novelty. Once a parent is selected, the system defers rich natural-language memory synthesis until an operator call is made, retrieving relevant ancestors and siblings to construct a bounded, operator-conditioned context. The overall experience-guided search framework is depicted in the figure below, showing how experience-card metadata updates the global search state and informs memory injection for subsequent operations.
Experiment
The evaluation uses the 22-task MLE-Bench Lite split under a fixed 12-hour single-GPU budget, reporting Valid Rate, Medal Average, and Human Rank across three runs. Execution-grounded post-training substantially improves performance over base models, and combining it with a domain-specific evolutionary search harness that distills cross-task priors yields complementary gains that surpass strong external systems. Long-horizon search with structured memory and multi-factor selection converts additional compute into sustained improvement beyond the first executable solution, raising solution quality toward Gold medals rather than merely crossing the Bronze threshold. These benefits transfer across input modalities and to a scientific benchmark, where the post-trained model and adapted search framework together improve paper-relative scientific progress.
On MLE-Bench Lite, the Frontis-MA1-35B system with the OPENMLE-EVO framework substantially outperforms its base Qwen model, delivering a higher valid submission rate, medal average, and human rank. The OPENMLE-EVO-MAX variant further improves all metrics, achieving perfect validity and the highest medal average. The smaller Frontis-MA1-30B model trails the 35B configuration across the same measures. Frontis-MA1-35B with OPENMLE-EVO raises the medal average and valid rate markedly above the Qwen3.6-35B-A3B base model. Using OPENMLE-EVO-MAX, the system reaches perfect validity (22/22) and a medal average of 71.21%, the highest in the comparison. Human rank improves from 0.58 for the base model to 0.76 with EVO and to 0.81 with the MAX variant. The smaller Frontis-MA1-30B model scores lower on valid rate, medal average, and human rank than the 35B system.
The comparison evaluates Frontis-MA1-30B and GLM-5.2 on an MLE benchmark across different harnesses. OPENMLE-EVO-MAX yields perfect validity and the highest medal average for both models, while GLM-5.2 with this framework attains the top human rank. The OPENMLE-EVO framework shows lower validity but can still improve medal average over the Claude Code baseline for GLM-5.2. OPENMLE-EVO-MAX achieves a flawless valid rate of 22/22 and a medal average of 66.67% for both Frontis-MA1-30B and GLM-5.2. GLM-5.2 with OPENMLE-EVO-MAX records the highest human rank (0.8164), edging out Frontis-MA1-30B with the same framework (0.8053).
On the 10-task NatureBench Lite, the best reference agents achieve 70% Surpass-SOTA and perfect Match-SOTA. The post-trained Frontis-MA1-35B improves over its base model by 10 percentage points in Surpass-SOTA and 20 points in Match-SOTA, while the adapted search harness alone contributes gains of 10 and 30 points, respectively. Together, the model and harness match several strong reference systems, indicating that execution-grounded post-training transfers beyond competition-style machine learning engineering. Claude Opus 4.7 and GLM-5.2 lead the reference agents with 70% Surpass-SOTA and 100% Match-SOTA on the 10-task subset. Frontis-MA1-35B raises Match-SOTA by 20 points over its base model and, combined with the OpenMLE-Evo adapter, reaches performance on par with GPT-5.4 and GLM-5.1.
Experiments on MLE-Bench Lite show that the OPENMLE-EVO framework, especially its MAX variant, dramatically improves the Frontis-MA1-35B model's validity, medal average, and human rank over the base model, with perfect validity and the highest medal average; the smaller 30B model lags behind. A head-to-head comparison confirms that OPENMLE-EVO-MAX yields perfect validity and top medal averages for both Frontis-MA1-30B and GLM-5.2, with GLM-5.2 leading in human rank. On NatureBench Lite, post-trained Frontis-MA1-35B combined with the harness matches top reference agents, validating the transfer of execution-grounded post-training beyond competition tasks.