Jump to content

AlphaTensor

From Wikipedia, the free encyclopedia

AlphaTensor
DeveloperDeepMind
Initial releaseOctober 5, 2022 (2022-10-05)
TypeArtificial intelligence
reinforcement learning
algorithm discovery
LicenseApache License 2.0
WebsiteAlphaTensor website
Repositorygithub.com/google-deepmind/alphatensor

AlphaTensor is an artificial intelligence system developed by DeepMind for discovering efficient matrix multiplication algorithms using reinforcement learning. Introduced in 2022, the system was based on AlphaZero and formulated the search for matrix multiplication algorithms as a single-player game called TensorGame.[1][2]

AlphaTensor was designed to search for new ways to multiply matrices with fewer scalar multiplication operations. Matrix multiplication is a fundamental operation in linear algebra, numerical analysis, scientific computing, computer graphics, and machine learning.[1] The system discovered thousands of matrix multiplication algorithms, including algorithms that rediscovered known human-designed methods and others that improved on previously known results for particular matrix sizes and mathematical settings.[1][3]

Background

[edit]

Matrix multiplication is one of the basic operations in numerical computing. The standard algorithm for multiplying two square matrices has cubic time complexity, while faster algorithms such as the Strassen algorithm reduce the number of multiplication operations by using more complex algebraic decompositions.[1] Finding optimal matrix multiplication algorithms can be difficult because it involves searching through a large space of possible tensor decompositions.[1]

AlphaTensor approached this problem by representing algorithm discovery as TensorGame, in which each move corresponds to an operation that reduces a tensor representing matrix multiplication. The goal of the game is to find a low-rank decomposition of the matrix multiplication tensor, corresponding to an efficient multiplication algorithm.[1]

Development

[edit]

AlphaTensor was developed by DeepMind and described in a paper published in Nature in October 2022.[1] The system built on the reinforcement-learning approach used in AlphaZero, which had previously been applied to games such as Go, chess, and shogi.[2] Unlike those games, TensorGame involved a very large search space, requiring changes to the AlphaZero-style search method and neural network architecture.[1]

DeepMind released source code and discovered algorithms associated with the publication through a public GitHub repository.[4]

Results

[edit]

AlphaTensor discovered matrix multiplication algorithms over both standard arithmetic and finite fields.[1] One widely reported result was a method for multiplying 4 × 4 matrices over the field with two elements using 47 multiplication operations, improving on the 49 operations required by applying Strassen's algorithm recursively in that setting.[1][5]

The system also found algorithms optimized for particular computer hardware, including algorithms designed for graphics processing units and Tensor Processing Units.[2] DeepMind stated that some of the hardware-specific algorithms improved practical execution time compared with commonly used algorithms on the tested hardware.[2]

Significance

[edit]

AlphaTensor was described as an example of using machine learning not only to apply existing algorithms, but to assist in discovering new ones.[5] The work was connected to broader research in algorithm discovery, automated machine learning, program synthesis, and computational complexity theory, especially the open problem of determining the optimal complexity of matrix multiplication.[1][3]

AlphaTensor later became part of a broader group of Google DeepMind systems for algorithm and mathematical discovery, alongside systems such as AlphaDev and AlphaEvolve.[2]

See also

[edit]

References

[edit]
  1. ^ a b c d e f g h i j k Fawzi, Alhussein; Balog, Matej; Huang, Aja; Hubert, Thomas; Romera-Paredes, Bernardino; et al. (October 2022). "Discovering faster matrix multiplication algorithms with reinforcement learning". Nature. 610 (7930): 47–53. doi:10.1038/s41586-022-05172-4. PMC 9534758. PMID 36198780.
  2. ^ a b c d e "Discovering novel algorithms with AlphaTensor". DeepMind. October 5, 2022. Retrieved April 29, 2026.
  3. ^ a b Brubaker, Ben (November 23, 2022). "AI Reveals New Possibilities in Matrix Multiplication". Quanta Magazine. Retrieved April 29, 2026.
  4. ^ "google-deepmind/alphatensor". GitHub. Google DeepMind. Retrieved April 29, 2026.
  5. ^ a b Hutson, Matthew (October 5, 2022). "DeepMind AI invents faster algorithms to solve tough maths puzzles". Nature. doi:10.1038/d41586-022-03166-w. Retrieved April 29, 2026.
[edit]