MNIST is solvable using two pixels. It shouldn’t be one of two benchmarks in a paper, again just in my opinion. It’s useful for debugging only.
really? do you have any details?
agree it has no business being in a modern paper
Training a GPT-2 sized model costs ~$20 nowadays in respect to compute: https://github.com/karpathy/llm.c/discussions/481
Even figuring out how to induce grokking behavior on a 100M model or OpenWebText would be a big leap in the understanding of grokking. It's perfectly reasonable for a paper like this to show results on the standard tasks for which grokking has already been characterized.
Academic research is often times about making incremental steps and limiting uncertainty.
Making something work for MNIST is already so much work, researchers don’t have the time, money, and energy to run experiments for 10 datasets.
Complex datasets are much harder to get a proper model trained on due to increased complexity - larger images, tasks, classes, etc.
Also, as soon as you run your experiments on more datasets, you create an opportunity for reviewers to take you down - “why didn’t you test it on this other dataset?”
https://arxiv.org/abs/2405.19454
It's a bit surprising that apparently this hasn't been done before - to see how universal of a phenomenon this is.