hraness

saved

44% on ARC-AGI-1 in 67 cents

by Mithil VakdeMithil Vakdepublished

gist

Mithil Vakde trains a small transformer from scratch at test time on ARC-AGI and scores 44% on ARC-1, plus 7% on ARC-2, for 67 cents of 5090 compute, matching TRM and HRM without recursion. The lift comes from 3D RoPE and per-task embeddings, a modern SwiGLU and RMSNorm stack, NorMuon, fewer augmentations, and supervised output-only loss. Ablations fall to about 24% without those representations. He argues ARC should measure sample efficiency, ban synthetic pretraining, and stop mixing LLM and from-scratch models on one cost-per-task chart.

ideas

  • Sample efficiency is the research target. ARC is useful because it has few puzzles, is unsaturated without synthetic data, and is cheap enough for GPU-poor iteration.
  • Representations, not recursion, carry the score. 3D RoPE plus per-task embeddings do the work; drop either and the model saturates near 24%. Recursion's confirmed benefit is extra compute without extra memory movement.
  • Val-loss chasing can fail. Switching to supervised output-only loss made test loss worse yet raised the score from 40% to 44% with less variance.
  • Do not mix model classes on one cost chart. Lifetime compute (init plus inference on all tasks) is 67 cents; amortizing across tasks and omitting LLM pretraining makes comparisons unfair.
  • 65% looks reachable inside transformers. Open-source mdlARC already unions to 55% across runs. He asks others to improve score or cost, not add training data.

quotes

I think sample efficiency is the most important problem in AI today and I want to solve it.

Mithil Vakde, stating why he works on ARC.

The biggest contribution to performance seems to be good representations (3D RoPE + per-task embedding).

Mithil Vakde, naming the ablation that actually moves the score.

What’s weird is that the test loss is now worse, yet it scores better!

Mithil Vakde, noting the failure mode of chasing validation loss.

A single leaderboard graph comparing multiple types of models doesn’t make sense.

Mithil Vakde, arguing for separate charts of LLM and from-scratch systems.