I'm Arman, a senior studying Mathematics at Williams College. I'm interested in mathematical foundations and intuitions behind machine learning. Last summer, I was an undergraduate research intern in Cengiz Pehlevan's Group at Harvard University, working on ML theory and high-dimensional statistics. The summer before that, I was doing research in the Number Theory and Probability Group at SMALL REU. I am applying for PhD programs in theoretical ML this year.
Arman Rysmakhanov, Mary Letey, Yue M. Lu, Jacob Zavatone-Veth, Cengiz Pehlevan
We propose an exact high-dimensional theory of in-context binary classification with linear attention through deriving an Gaussian equivalent for the prediction logit and applying the leave-one-out (cavity) approach. We analyze test performance and calibration, tracing how context length, pretraining data, regularization, and temperature affect them.
Selected findings: (a) asympototic loss; below a critical amount of training data the model breaks down; above it, theory and simulation agree; (b) calibration moves from overconfidence to underconfidence with more data; (c) sharper teachers and longer contexts need more data before things start working.
We study in-context binary classification in a simple attention model trained to minimise a logistic loss. Each prompt has a latent logistic teacher, and the learner must infer its direction from Bernoulli-labeled context examples. We derive an asymptotic prediction for the test logit in a high-dimensional regime with context length $\ell \propto d$ and total training prompts $n \propto d^2$. Our formula cleanly separates calibration, test-time context noise, and non-isotropic parameter-estimation noise as interpretable effects on in-context performance. Our analysis also identifies the ridgeless separability regime and addresses the effects of context length, pretraining-set size, regularisation, and sampling temperature on in-context classification performance. The theory's predictions closely agree with finite-dimensional simulations and fully trained linear-attention models, while experiments with standard softmax attention show qualitatively related behaviour beyond the regime assumed by our analysis.
Mary Letey, Arman Rysmakhanov, Yue M. Lu, Cengiz Pehlevan, Jacob Zavatone-Veth
This is a small extension of the existing theory of in-context linear regression using linear attention. What if the labels were nonlinear $y=f(x^\top w)+\varepsilon$ instead of the single-index $y=x^\top w+\varepsilon$? We prove a nonlinearity-noise equivalence: with new labels, the model learns only the linear Hermite component of $f$, and the rest acts as extra noise.
Thanks to the noise-nonlinearity equivalence, for every nonlinear target there exists a linear twin with just the right amount of extra noise, with test loss exactly the same.
Linear attention is a tractable model for understanding the mechanisms governing in-context learning in transformers. For linear regression tasks, recent asymptotic analyses have characterised its learning and generalisation behaviour. We extend this theory to nonlinear single-index targets, $y = f(x^\top w) + \varepsilon$. Our main result establishes a nonlinearity-noise equivalence: linear attention extracts only the linear Hermite component of $f$, while the remaining nonlinear structure contributes to the generalisation error as effective noise. This reduction allows results from the corresponding linear theory to be transferred to nonlinear tasks. We illustrate its implications for finite pretraining data and for the transition from task memorisation to task generalisation as task diversity increases. These results identify a limitation of the reduced linear-attention model and provide a tractable starting point for studying nonlinear in-context learning.
Steven J. Miller, Arman Rysmakhanov (alphabetical)
We characterize the limiting spectral distribution and the outlier eigenvalue of the adjacency matrix for a dense variant of Barabási-Albert graphs using random matrix theory tools, specifically graphons and quadratic vector equations.
Spectrum of a random dense Barabási-Albert multigraph.
Preferential attachment graphs model networks whose growth produces highly uneven degree distributions, describing many real-world systems. Their adjacency spectra are important because they allow graph-theoretic questions to be studied through the eigenvalues of matrices. We analyze the adjacency matrix of a dense Barabási-Albert (B-A) multigraph, where the number of edges added at each step is proportional to the final number of vertices. First, we compute the large-$n$ limit of the expected adjacency matrix and show that it is described by a rank-one limiting kernel, viewed as a continuous analogue of the adjacency matrix. After centering and scaling, the fluctuations form a random matrix with a computable variance profile. Using the quadratic vector equation approach, we derive the limiting bulk spectral distribution. We also determine the asymptotic location of the leading eigenvalue generated by the rank-one mean component.
Summer-Fall 2025, to appear in the Fibonacci Quarterly
Ivan Bortnovskyi, June Duvivier, Pedro Espinosa, Michael Lucas, Steven J. Miller, Tiancheng Pan, Arman Rysmakhanov, Iana Vranesko, Ren Watson, Steven Zanetti (alphabetical)
The $k$-bonacci numbers are a generalization of the Fibonacci numbers, and all positive integers can be uniquely expressed as a sum of them with no $k$ consecutive terms used. Building on the work that extends the representation to vectors, we develop two fast algorithms for finding unique $k$-bonacci representation of integer vectors, and also prove that the vector representation's summands have the same count, gap behavior and minimality as scalar $k$-bonacci decompositions.
The first few two-dimensional Fibonacci (well, tribonacci) $2d$ vectors.
Zeckendorf's Theorem says that for all $k \ge 3$, every nonnegative integer has a unique $k$-Zeckendorf representation as a sum of distinct $k$-bonacci numbers, where no $k$ consecutive $k$-bonacci numbers are present in the representation. Anderson and Bicknell-Johnson extend this result to the multidimensional context: letting the $k$-bonacci vectors $\vec{\mathbf{X}}_i \in \mathbb{Z}^{k-1}$ be given by $\vec{\mathbf{X}}_0 = \vec{\mathbf{0}}$, $\vec{\mathbf{X}}_{-i} = \vec{\mathbf{e}}_i$ for $1 \le i \le k-1$, and $\vec{\mathbf{X}}_n = \sum_{i=1}^{k} \vec{\mathbf{X}}_{n-i}$ for all $n \in \mathbb{Z}$, they show that for all $k \ge 3$, every $\vec{\mathbf{v}} \in \mathbb{Z}^{k-1}$ has a unique $k$-bonacci vector Zeckendorf representation, a sum of distinct $k$-bonacci vectors where no $k$ consecutive $k$-bonacci vectors are present in the representation. Their proof provides an inductive algorithm for finding such representations. We present two improved algorithms for finding the $k$-bonacci vector Zeckendorf representation of $\vec{\mathbf{v}}$ and analyze their relative efficiency. We utilize a projection map $S_n : \mathbb{Z}^{k-1} \to \mathbb{Z}_{\ge 0}$, introduced in Anderson and Bicknell-Johnson work, that reduces the study of $k$-bonacci vector representations to the setting of $k$-bonacci number representations, provided a lower bound is established for the most negatively indexed $k$-bonacci vector present in the $k$-bonacci vector Zeckendorf representation of $\vec{\mathbf{v}}$. Using this map and a bijection between $\mathbb{Z}^{k-1}$ and $\mathbb{Z}_{\ge 0}$, we further show that the number of and gaps between summands in $k$-bonacci vector Zeckendorf representations exhibit the same properties as those in $k$-Zeckendorf representations and that $k$-bonacci vector Zeckendorf representations exhibit summand minimality.
Julian Duvivier, Xiaoyao Huang, Ava Kennon, Say-Yeon Kwon, Steven J. Miller, Arman Rysmakhanov, Pramana Saldin, Ren Watson (alphabetical)
We extend the so-called problem of More-Sums-Than-Differences from integers to groups in general. Specifially, in a non-Abelian group, combining elements of a finite subset as $ab^{-1}$ or as $a^{-1}b$ can result in different numbers of distinct elements. We prove that the difference between these two counts can take any integer value for the infinite dihedral group but only even integer value for the free group on two generators, finding the smallest subset for which a gap can occur.
This is what we called a difference graph: each dot is one product $a_i a_j^{-1}$, while lines join the dots whose products are equal. So the problem of counting distinct products becomes a problem counting connected components.
For a finite subset $A$ of a group $G$, we define the right quotient set and the left quotient set of $A$ as $$\begin{aligned} AA^{-1} &:= \{a_1a_2^{-1} : a_1, a_2 \in A\}, \\ A^{-1}A &:= \{a_1^{-1}a_2 : a_1, a_2 \in A\} \end{aligned}$$ respectively. While the right and left quotient sets are equal if $G$ is abelian, subtleties arise when $G$ is a non-abelian group, where the cardinality difference $|AA^{-1}| - |A^{-1}A|$ may take on arbitrarily large values. Using the results of Martin and O'Bryant on the cardinality differences of sum sets and difference sets in $\mathbb{Z}$, we prove that in the infinite dihedral group, $D_\infty \cong \mathbb{Z} \rtimes \mathbb{Z}/2\mathbb{Z}$, every integer difference is achievable. Further, we prove that in $F_2$, the free group on 2 generators, an integer difference is achievable if and only if that integer is even, and we explicitly construct subsets of $F_2$ that achieve every even integer. We further determine the minimum cardinality of $A \subset G$ so that the difference between the cardinalities of the left and right quotient sets is nonzero, depending on the existence of order 2 elements in $G$. To prove these results, we construct difference graphs $D_A$ and $D_{A^{-1}}$ which encode equality in the right and left quotient sets respectively. We observe a bijection from edges in $D_A$ to edges in $D_{A^{-1}}$ and count connected components in order to obtain our results on cardinality differences $|AA^{-1}| - |A^{-1}A|$.
Steven J. Miller, Arman Rysmakhanov, Sheafification of G (alphabetical)
How fast can one compute the millionth Fibonacci number? We walk the reader through various methods, from naive recursion to fast matrix exponentiation and the fast Fourier transform.
This article serves as a companion to the video essay by Sheafification of G on computing large Fibonacci numbers quickly. We follow the same progression of methods—from naïve recursion to sophisticated Fourier transforms—and elaborate the concepts that were impractical to discuss in video format.