Graph neural networks automate conjecture discovery in number theory research

Graph neural networks, the deep learning architectures that have transformed everything from molecular design to social network analysis, are now being pointed at one of the oldest and most stubbornly abstract corners of mathematics: number theory. In a study published in the Journal of Big Data, a research team from Jilin Agricultural Science and Technology University and the Jilin Institute of Chemical Technology in China describes a structured computational pipeline that uses graph neural networks to generate new mathematical conjectures automatically, moving beyond intuition-driven guesswork toward a reproducible, statistically grounded method of pattern discovery in pure mathematics.

Conjectures are the lifeblood of number theory. Statements such as the Riemann hypothesis or the prime number theorem began as observed regularities in numerical data before mathematicians set out to prove or disprove them. Historically, formulating such conjectures has depended on the pattern-recognition abilities of exceptional human minds, supported by increasing amounts of computer-assisted experimentation. The new work aims to formalize that experimentation stage. Rather than treating conjecture generation as a black-box or a purely heuristic exercise, the researchers define an explicit sequential pipeline that runs from data acquisition through representation learning to hypothesis construction, with statistical validation woven into every stage.

At the heart of the approach is the encoding of number-theoretic objects as graphs. Prime numbers, for example, are organized into what the authors call a prime number graph, in which individual primes become nodes and the relationships between them, defined by modular and structural properties, become edges. Similarly, the coefficients of the Ramanujan tau function, a central object in the theory of modular forms, are represented in a modular form graph. Once the arithmetic objects are cast in graphical form, graph neural networks can be applied in earnest. These networks operate by passing messages between nodes, allowing each node to build up a representation, or embedding, that captures not only its own properties but also the structure of its neighborhood. Crucially, the embedding stage is designed to preserve arithmetic and structural invariants, ensuring that the learned representations do not discard the very mathematical properties that matter.

The team trained and evaluated their models on curated datasets drawn from established mathematical repositories, including the L-functions and modular forms database (LMFDB) and the Online Encyclopedia of Integer Sequences (OEIS). In practice, the experiments covered 15,000 prime numbers and 2,000 Ramanujan tau coefficients. The neural architectures were implemented using PyTorch Geometric, the widely used graph deep learning library, with graph autoencoders employed to learn compact representations in an unsupervised manner and ReLU activations and Adam optimization standardizing the training procedure. The results were striking: the models achieved area-under-the-curve scores of 0.91 on the prime data and 0.88 on the tau coefficient data, indicating strong discriminative power, while clustering quality in the learned embedding spaces remained robust, with silhouette scores of at least 0.52.

Once stable embeddings are obtained, the pipeline turns to the extraction of latent statistical patterns. The researchers apply clustering algorithms to group similar objects in the high-dimensional representation space, and then deploy symbolic regression to search for closed-form mathematical relationships that describe the regularities the network has uncovered. Symbolic regression is particularly significant here because it produces human-readable formulas rather than opaque numerical predictions, allowing mathematicians to inspect, test, and potentially prove the relationships the machine has identified. In this sense, the system acts less like an oracle and more like a highly disciplined research assistant, surfacing candidate patterns that would otherwise require months of manual computation to notice.

The candidate conjectures that emerge are then subjected to rigorous evaluation. The experimental design partitions datasets into training, validation, and testing subsets, and multiple random seeds are used across runs to ensure that results are not artifacts of a particular initialization or data split. Statistical significance is assessed through controlled hypothesis testing, with false discovery rate correction applied to account for the multiple comparisons problem, a well-known pitfall in which testing many hypotheses simultaneously inflates the chance of spurious findings. Quantitative assessment relies on goodness-of-fit measures such as the coefficient of determination and out-of-sample validation, while qualitative assessment checks the extracted patterns for consistency with known theoretical results, including classical results such as the prime number theorem.

The outcomes are notable both for their quality and their diversity. According to the study, the pipeline’s GNN-based models were able to predict trends in prime gaps, the irregular intervals between consecutive primes, and properties of tau coefficients, and 75 percent of the generated conjectures were independently rated as either plausible or novel by the evaluation framework. Visual interpretability played a supporting role throughout: t-distributed stochastic neighbor embedding (t-SNE) plots, scatter plots, and silhouette diagrams were used to visualize how number-theoretic objects organize themselves in the learned embedding space, giving researchers an intuitive window into why the model groups certain primes or tau coefficients together. The authors are candid, however, that scalability remains a challenge; extending the framework to vastly larger datasets and more complex arithmetic structures will require further engineering and theoretical work.

What distinguishes this study from earlier experiments at the intersection of machine learning and mathematics, such as the well-known work on using neural networks to study knots and representation theory, is its emphasis on the full conjecture-generation pipeline rather than a single predictive task. The framework formally defines each transformation, from raw numerical data to embedding, from embedding to extracted pattern, and from pattern to formulated hypothesis, and it builds bias-reduction into the process by replacing intuition with statistically validated regularities. The authors argue that this reduces the subjective bias associated with purely human-driven conjecture formation while establishing a bridge between deep learning-based pattern discovery and formal mathematical reasoning.

The implications reach beyond number theory. A reproducible, interpretable, and scalable pipeline for conjecture generation could in principle be adapted to other fields of pure mathematics where large structured datasets exist, from combinatorics to representation theory. The work resonates with a broader movement in the mathematical community toward machine-assisted discovery, exemplified in recent years by collaborative projects that pair mathematicians with large language models and automated reasoning tools. By providing an open, documented pipeline built on established libraries and public databases, the Jilin team has contributed a template that other groups can replicate, extend, and stress-test. The study’s openness is reinforced by its publication under a Creative Commons license, allowing unrestricted non-commercial reuse with attribution.

For mathematicians, the message is not that machines will soon replace the profound creative leaps of human conjecture-making, but that the earliest, most laborious stage of discovery, combing through oceans of numerical data for hints of structure, can now be systematically automated. For computer scientists, the study demonstrates that graph neural networks, already proven on molecules and social graphs, can carry genuine mathematical weight when paired with careful statistical hygiene. And for the growing community of researchers exploring AI for mathematics, the result offers a concrete answer to a pressing question: how do you know a machine-generated conjecture is worth a mathematician’s time? The answer emerging from Jilin is that you test it the way you would test any scientific hypothesis, with controlled experiments, corrected statistics, and honest out-of-sample validation. If 75 percent of the output passes that bar today, the framework’s authors and their successors will be working to raise that figure as the architecture scales to the vast, unexplored territories of the mathematical universe.