← Search

Adityanarayanan Radhakrishnan

4 accepted papers

2026

xRFM: Accurate, scalable, and interpretable feature learning models for tabular data

ICLR 2026poster

Inference from tabular data, collections of continuous and categorical variables organized into matrices, is a foundation for modern technology and science. Yet, in contrast to the explosive changes in the rest of AI, the best practice for these predictive tasks has been relatively unchanged and is…

Cited by 0SourcecodeScholar
2025

Emergence in non-neural models: grokking modular arithmetic via average gradient outer product

ICML 2025oral

Neural networks trained to solve modular arithmetic tasks exhibit grokking, a phenomenon where the test accuracy starts improving long after the model achieves 100% training accuracy in the training process. It is often taken as an example of "emergence", where model ability manifests sharply throug…

Cited by 6SourcePDFScholar
2024

Catapults in SGD: spikes in the training loss and their impact on generalization through feature learning

ICML 2024poster

In this paper, we first present an explanation regarding the common occurrence of spikes in the training loss when neural networks are trained with stochastic gradient descent (SGD). We provide evidence that the spikes in the training loss of SGD are "catapults", an optimization phenomenon originall…

2024

Quadratic models for understanding catapult dynamics of neural networks

ICLR 2024poster

While neural networks can be approximated by linear models as their width increases, certain properties of wide neural networks cannot be captured by linear models. In this work we show that recently proposed Neural Quadratic Models can exhibit the "catapult phase" Lewkowycz et al. (2020) that arise…