Breaking Data Symmetry is Needed For Generalization in Feature Learning Kernels

14d ago · Global · primary source: export.arxiv.org

A new study posted to arXiv finds that feature-learning kernel models only achieve delayed generalization, a phenomenon known as grokking, when a specific symmetry in the training data is broken. The work examines how the Recursive Feature Machine algorithm learns algebraic tasks. The paper, submitted on 31 Mar 2026 and revised on 24 Jun 2026, focuses on grokking, where a model reaches high training accuracy long before it begins to generalize to unseen test data [1][2]. This behavior was first documented on algebraic problems like modular arithmetic by Power et al. in 2022 [1][2]. The authors, led by Marcel Tomas Bernal, investigate this effect using the Recursive Feature Machine (RFM) algorithm, a method introduced by Radhakrishnan et al. in 2024 [1][2]. RFM iteratively updates feature matrices through the Average Gradient Outer Product (AGOP) of an estimator to learn task-relevant features [1][2]. The central experimental finding is that generalization occurs only when a certain symmetry in the training set is broken [1][2]. The study empirically demonstrates that RFM generalizes by recovering the underlying invariance group action inherent in the data, with the learned feature matrices encoding specific elements of that invariance group [1][2]. This explains the dependence of generalization on symmetry breaking [1][2]. The research was posted on arXiv, an open-access repository for electronic preprints that has been operating since August 14, 1991, and as of November 2024 receives about 24,000 submissions per month [7]. The platform hosts papers across mathematics, physics, computer science, and related fields, and its content is moderated but not peer-reviewed [7]. The paper's abstract page also links to several community-developed tools under the arXivLabs framework, a formalized collaboration space launched in 2020 that allows third-party developers to build features directly on the site [5][6]. These tools include the Bibliographic Explorer for navigating citation trees and the CORE Recommender for surfacing related open-access papers [5][6]. arXivLabs partners must adhere to the repository's values of openness, community, excellence, and user data privacy, and are granted only minimal, anonymized data access [5].

research-paper

Background sources we checked (8)
  • arxiv.org ↗ Grokking occurs when a model achieves high training accuracy but generalization to unseen test points happens long after that. This phenomenon was initially observed on a class of algebraic problems, such as learning modular arithmetic (Power et al., 2022). We study grokking on a…
  • en.wikipedia.org ↗ In linear algebra, the singular value decomposition (SVD) is a factorization of a real or complex matrix into a rotation, followed by a scaling, followed by another rotation. It generalizes the eigendecomposition of a square normal matrix with an orthonormal eigenbasis to any ⁠ …
  • info.arxiv.org ↗ arXiv Labs - arXiv info | arXiv e-print repository Skip to content # arXiv Labs Attention arXiv Users: arXiv Labs is pausing new proposals ## What are arXiv Labs? arXiv Labs are a way for the community to contribute new, useful features to arXiv. These integrations are avail…
  • blog.arxiv.org ↗ arXivLabs: a space for community innovation – arXiv blog arXiv has launched a new, formalized framework enabling innovative collaborations with individuals and organizations. “Members of our community want to contribute tools that enhance the arXiv experience, and we val…
  • info.arxiv.org ↗ arXivLabs: Showcase - arXiv info | arXiv e-print repository ... # arXivLabs: Showcase ... arXiv is surrounded by a community of researchers and developers working at the cutting edge of information science and technology. ... While the arXiv team is focused on our core mission—pr…
  • en.wikipedia.org ↗ arXiv (pronounced as "archive"—the X represents the Greek letter chi ⟨χ⟩) is an open-access repository of electronic preprints and postprints (known as e-prints) approved for posting after moderation, but not peer reviewed. It consists of scientific papers in the fields of mathem…
  • en.wikipedia.org ↗ 14 (fourteen) is the natural number following 13 and preceding 15.…
  • en.wikipedia.org ↗ A large language model (LLM) is a type of machine learning model designed for natural language processing tasks such as language generation. LLMs are language models with many parameters, and are trained with self-supervised learning on a vast amount of text.…

Sources

Spot something wrong? Report an issue