IntSeqBERT: Learning Arithmetic Structure in OEIS via Modulo-Spectrum Embeddings

TL;DR

IntSeqBERT is a novel Transformer-based model that encodes integer sequences using magnitude and modulo embeddings, significantly improving prediction accuracy and revealing arithmetic structures in OEIS sequences.

Contribution

The paper introduces IntSeqBERT, a dual-stream Transformer that effectively models OEIS sequences by combining magnitude and modulo embeddings, outperforming standard tokenized models.

Findings

Achieves 95.85% magnitude accuracy on OEIS sequences.

Improves next-term prediction by 7.4 times over baseline.

Reveals strong correlation between information gain and Euler's totient ratio.

Abstract

Integer sequences in the OEIS span values from single-digit constants to astronomical factorials and exponentials, making prediction challenging for standard tokenised models that cannot handle out-of-vocabulary values or exploit periodic arithmetic structure. We present IntSeqBERT, a dual-stream Transformer encoder for masked integer-sequence modelling on OEIS. Each sequence element is encoded along two complementary axes: a continuous log-scale magnitude embedding and sin/cos modulo embeddings for 100 residues (moduli $2$--$101$), fused via FiLM. Three prediction heads (magnitude regression, sign classification, and modulo prediction for 100 moduli) are trained jointly on 274,705 OEIS sequences. At the Large scale (91.5M parameters), IntSeqBERT achieves 95.85% magnitude accuracy and 50.38% Mean Modulo Accuracy (MMA) on the test set, outperforming a standard tokenised Transformer baseline by $+8.9$ pt and $+4.5$ pt, respectively. An ablation removing the modulo stream confirms it accounts for $+15.2$ pt of the MMA gain and contributes an additional $+6.2$ pt to magnitude accuracy. A probabilistic Chinese Remainder Theorem (CRT)-based Solver converts the model's predictions into concrete integers, yielding a 7.4-fold improvement in next-term prediction over the tokenised-Transformer baseline (Top-1: 19.09% vs. 2.59%). Modulo spectrum analysis reveals a strong negative correlation between Normalised Information Gain (NIG) and Euler's totient ratio $\varphi(m)/m$ ($r = -0.851$, $p < 10^{-28}$), providing empirical evidence that composite moduli capture OEIS arithmetic structure more efficiently via CRT aggregation.