Downloads · 30 days
0
arc-agi2-solver
aryaman27/arc-agi2-solver
arc-agi2-solver is a machine learning model from aryaman27. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Multi-strategy ensemble solver for the ARC Prize 2026 Kaggle competition.
Access
Public
Updated May 7, 2026
Repo size
—
Likes
0
Public
Repo makeup
Click a slice to open those files.
.py54.1 KB · 94%
At a glance
- Access
- Public
- Created
- May 7, 2026
- Updated
- May 7, 2026
- SHA
- 79e89fbc
Model card
From the Hugging Face model README
ARC Prize 2026 - ARC-AGI-2 Solver
Multi-strategy ensemble solver for the ARC Prize 2026 Kaggle competition.
Architecture
Three-pronged ensemble:
-
DSL Solver — 32 primitive transforms (rotations, flips, cropping, border extraction, hole filling, object detection, color operations) + depth-2 composition. Solves geometric/structural tasks in milliseconds.
-
Object Solver — Connected components detection: extracts largest objects, converts object sets to color bars.
-
TTT Solver — 236K-parameter encoder-decoder Transformer trained from scratch per task using test-time training. 16 augmentations per example (D8 symmetries + color permutations).
Ensemble logic: DSL → Object → TTT → Identity fallback
Performance
- DSL accuracy on ARC-AGI-2 training: ~1.8% (18/1000 tasks)
- TTT model: 236K parameters, ~5-10s per task on L4 GPU
- Optimized for: Kaggle 4×L4 GPU, 12-hour time limit
Usage
- Download
kaggle_notebook.py - Upload to Kaggle as a notebook
- Run — it reads from
/kaggle/input/arc-prize-2026/and writes/kaggle/working/submission.json
Files
| File | Description |
|---|---|
kaggle_notebook.py | Complete Kaggle submission notebook |
kaggle_solver.py | Standalone Python module (more features) |
train_sft_barc.py | Optional: SFT pre-training on BARC dataset |
References
Built on research from:
- NVARC (2025 winner, 24% on ARC-AGI-2): Test-time training + heavy augmentation
- Product-of-Experts (Franzen et al., 2025): DFS + probability threshold + PoE scoring
- SOAR (2025, 2nd place paper): Self-improving evolutionary program synthesis
- MARC (NeurIPS 2025): The Surprising Effectiveness of TTT
- CompressARC (2025, 3rd place paper): MDL-based code golf, 76K parameters
License
MIT — free to use, modify, and distribute.
- Created
- May 7, 2026
- Updated
- May 7, 2026