N-gram-like Language Models Predict Naturalistic Reading Time Best
Authors: James A. Michaelov, Roger P. Levy
Organizations: Department of Brain and Cognitive Sciences, Massachusetts Institute of Technology · MIT Libraries CREOS, Massachusetts Institute of Technology
Recent work has found that contemporary language models such as transformers can become so good at next-word prediction that the probabilities they calculate become worse for predicting naturalistic reading time. In this paper, we propose that this can be explained by reading time being shaped by simple n-gram statistics rather than the more complex statistics learned by state-of-the-art transformer language models. We demonstrate that the neural language models whose predictions are most correlated with n-gram probability are also those that calculate probabilities that are the most correlated with eye-tracking-based metrics of reading time on naturalistic text.
Figures & tables
Figure 1: The correlation between n -gram surprisal and reading time in the Provo Corpus.
Figure 2: (A) The relationship between language model surprisal over the course of training and both n -gram surprisal and reading time in the Provo Corpus. (B) The relationship between the two sets of correlations.
Figure 3: The relationship between the correlation between language model surprisal and n -gram surprisal and the correlation between language model surprisal and each measure of reading time in the (A) Provo and (B) GECO datasets.
GECO
Provo
Family
N-gram
FFD
FPD
SGPD
GPD
TD
FFD
FPD
GPD
TD
Gemstone
Unigram Surprisal
0.964
0.981
0.984
0.990
0.980
0.884
0.980
0.955
0.949
Gemstone
Bigram Surprisal
0.942
0.986
0.984
0.970
0.966
0.896
0.986
0.972
0.966
Gemstone
Trigram Surprisal
0.976
0.970
0.974
0.982
0.982
0.886
0.959
0.969
0.962
Open GPT-2
Unigram Surprisal
0.564
0.643
0.662
0.771
0.646
0.511
0.772
0.723
0.677
Open GPT-2
Bigram Surprisal
0.918
0.974
0.976
0.938
0.964
0.955
0.989
0.973
0.988
Table 1: The Pearson correlation coefficient r between the correlation between language model surprisal and n -gram surprisal and the correlation between language model surprisal and First Fixation Duration (FFD), First Pass Duration (FPD), Go-Past Duration (GPD), and Total Duration (TD); with the highest bolded.
Probing has shown that language model representations encode rich linguistic information, but it remains unclear whether they also capture cognitive signals about human processing. In this work, we probe language model representations for human reading times. Using regularized linear regression on two eye-tracking corpora spanning five languages (English, Greek, Hebrew, Russian, and Turkish), we compare the representations from every model layer against scalar predictors -- surprisal, information value, and logit-lens surprisal. We find that the representations from early layers outperform surprisal in predicting early-pass measures such as first fixation and gaze duration. The concentration of predictive power in the early layers suggests that human-like processing signatures are captured by low-level structural or lexical representations, pointing to a functional alignment between model depth and the temporal stages of human reading. In contrast, for late-pass measures such as total reading time, scalar surprisal remains superior, despite its being a much more compressed representation. We also observe performance gains when using both surprisal and early-layer representations. Overall, we find that the best-performing predictor varies strongly depending on the language and eye-tracking measure.
Eleftheria Tsipidi, Samuel Kiegeland, Francesco Ignazio Re +4
ETH Zürich · Toyota Technological Institute at Chicago · University College London
Tokenization-free, pixel-based language models (PLMs) have recently been proposed as an alternative to conventional token-based language models (TLMs), typically motivated by practical considerations, e.g., the flexibility to support new languages or vocabularies. Nevertheless, PLMs also have the potential to serve as infrastructure for computational (psycho)linguistics, where tokenization has been a source of bias in estimating information-theoretic values, such as surprisal, and has pushed the bottom-up visual orthographic features outside the scope. In this study, we first integrate PLMs into the modeling of naturalistic human reading and, in particular, examine their use to estimate next-word predictability, which can also integrate orthographic and parafoveal information. Our results demonstrate that next-word prediction difficulty, computed in pixel space, has strong predictive power for gaze duration and word skipping behavior across various languages. In particular, their advantage is complementary to existing token-level surprisal and entropy estimates. Cross-lingual analyses with script variations also clarify their distinctive visual processing biases, facilitating research into how bottom-up visual information is incrementally integrated into expectation-based reading.
Junlin Li, Sara Alhajeri, Tatsuki Kuribayashi
Mohamed bin Zayed University of Artificial Intelligence · Tohoku University
Transformer-based language models are widely used as models of human language processing, yet their attention mechanisms allow lossless access to the full preceding context, unlike the limited memory systems of humans. We hypothesize that installing memory constraints into transformers' attention mechanisms can improve their fit to human behavioral data. While previous work has explored individual constraints in isolation, we conduct a systematic comparison of multiple attention-based memory mechanisms across different model sizes and training corpora, evaluating both psychometric predictive power for human reading times and grammatical competence. We additionally compare static constraints, in which the constraint strength is fixed throughout training, to dynamic memory curricula. We find that constraints that are sensitive to the content of intervening tokens consistently achieve the highest alignment with human reading times, outperforming distance-based constraints. We observe a dissociation between psychometric fit and grammatical competence under dynamic memory curricula, suggesting that Transformers cannot serve as a one-size-fits-all cognitive model.
Lanni Bu, Xiulin Yang, Christian Clark +2
Georgetown University · The Ohio State University · UC San Diego