← Search

Darshil Doshi

4 accepted papers

2025

(How) Can Transformers Predict Pseudo-Random Numbers?

ICML 2025poster

Transformers excel at discovering patterns in sequential data, yet their fundamental limitations and learning mechanisms remain crucial topics of investigation. In this paper, we study the ability of Transformers to learn pseudo-random number sequences from linear congruential generators (LCGs), def…

Cited by 0SourcePDFScholar
2024

Learning to grok: Emergence of in-context learning and skill composition in modular arithmetic tasks

NeurIPS 2024oral

Large language models can solve tasks that were not present in the training set. This capability is believed to be due to in-context learning and skill composition. In this work, we study the emergence of in-context learning and skill composition in a collection of modular arithmetic tasks. Specific…

2024

To Grok or not to Grok: Disentangling Generalization and Memorization on Corrupted Algorithmic Datasets

ICLR 2024poster

Robust generalization is a major challenge in deep learning, particularly when the number of trainable parameters is very large. In general, it is very difficult to know if the network has memorized a particular set of examples or understood the underlying rule (or both). Motivated by this challenge…

2023

Critical Initialization of Wide and Deep Neural Networks using Partial Jacobians: General Theory and Applications

NeurIPS 2023spotlight

Deep neural networks are notorious for defying theoretical treatment. However, when the number of parameters in each layer tends to infinity, the network function is a Gaussian process (GP) and quantitatively predictive description is possible. Gaussian approximation allows one to formulate criteria…