Uncovering a Universal Abstract Algorithm for Modular Addition in Neural Networks

Gavin McCracken (McGill University) · Gabriela Moisescu-Pareja (McGill University, Mila) · Vincent Létourneau (Université de Montréal) · Doina Precup (McGill University / Mila / DeepMind Montreal) · Jonathan Love (Leiden University, Dept. of Mathematics, Leiden University)
abstract algorithmapproximate chinese remainder theoremapproximate cosetsdeep neural networksgeneralizable interpretabilitygroup multiplicationhidden layermodular additionmulti-level analysesmultilayer perceptronsneural network solutionsneuron-level representationstrainable embeddingstransformersuniversality hypothesis

We propose a testable universality hypothesis, asserting that seemingly disparate neural network solutions observed in the simple task of modular addition actually reflect a common abstract algorithm. While prior work interpreted variations in neuron-level representations as evidence for distinct algorithms, we demonstrate