acceptodds
Under review as a conference paper at ICLR 2027

Input Relational Structure Shapes Training Dynamics in Neural Networks

Abstract

Input representations are known to influence machine learning models, yet their effects are typically evaluated only after training has converged. We ask whether the training trajectory of learned representations carries a signature of the relational structure encoded in the inputs, and whether this signature predicts robustness. Using molecular property prediction, we compare models operating on graph-, sequence-, and vector-based encodings, and track the linear extractability of chemical properties from their representations across training checkpoints. All three encodings support substantial extraction already at initialization, but differ in how they reorganize it: sequence models largely preserve their initial extractability profile (Spearman between initial and peak extractability), whereas graph and vector models reorganize substantially ( and ), each on different property subsets. To determine whether these representational dynamics are related to robustness, we introduce controlled corruption across all input relational structures and repeat training. Although corruption has little effect on overall task performance, it alters representations in an encoding-specific way: disruption is diffuse and strongest for vectors, intermediate for sequences, and selective for graphs, concentrated on properties that depend on the removed components. Finally, the evolution of extractability during uncorrupted training correlates with corruption sensitivity (), whereas geometric summaries of the same trajectories do not, suggesting representation dynamics as a complementary lens on robustness.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.