James Town
I'm a current student in the MathSys II CDT under the supervision of Professor Ranko Lazic (DCS), Dr Matthias Englert (DCS), and Professor Nikolaos Zygouras (WMI). My research focuses on understanding the implicit bias of gradient based learning algorithms. By implicit bias, we mean the tendency of a class of algorithms designed to optimise the parameters of some function to find solutions with certain properties. Examples might include the minimisation of the norm of parameters or the rank of a feature matrix. See Vardi (2023)[1] for a review of the area.
I'm also interested in the phenomenon of model collapse, a process in which the performance of generative models declines (losing both diversity and accuracy) over iterations of retraining the model on its own data.
[1] Gal Vardi. 2023. On the Implicit Bias in Deep-Learning Algorithms. Commun. ACM 66, 6 (June 2023), 86–93. https://doi.org/10.1145/3571070
Publications and Preprints
Abstract
The successful training of neural networks hinges on the use of first order optimization methods, yet the theoretical characterization of these methods remains incomplete. This is especially true in settings with mild overparameterization. In this work, we study the gradient flow dynamics of two-layer ReLU networks from small initialization with orthogonal training data. We prove the limiting flow converges to a saddle-to-saddle jump process as the initialization scale tends to zero, revealing an incremental learning phenomenon in which a new neuron activates at each saddle. This analysis recovers the known result of Dana et al. (2025) that the network interpolates the training data with high probability as soon as , where
is the network width and
is the number of training samples. This incremental process characterization also allows us to derive a novel implicit bias result: the learned interpolator has a squared
-norm scaling as
, which is within a constant factor of the minimal
-norm interpolator. More broadly, our work provides the first rigorous proof of an incremental learning process for ReLU networks, whilst suggesting mildly overparameterized networks can converge to interpolating solutions whose complexity is of the same order as that of the optimal interpolator.
Workshops Attended
- Machine Learning and Theory, University of East Anglia (May 2024)
- ML@DCS, University of Warwick (Feb 2025)
Teaching Experience
- Senior Graduate Teaching Assistant for CS130 Sets and Proofs in the academic year 25/26; module led by Professor Ranko Lazic
- Senior Graduate Teaching Assistant for CS146-10 Discrete Mathematics and its Applications 1 in the academic year 24/25; module led by Dr Charilaos Efthymiou
Education
- MSc Mathematics of Systems, University of Warwick (Sep 2023 - Sep 2024)
- BSc Mathematics, University of Durham (Oct 2020 - Jun 2023)