Neural Learning of Fast Matrix Multiplication Algorithms: A StrassenNet Approach

Paolo Andreini; Alessandra Bernardi; Monica Bianchini; Barbara Toniella Corradini; Sara Marziali; Giacomo Nunziati; Franco Scarselli

Neural Learning of Fast Matrix Multiplication Algorithms: A StrassenNet Approach

Paolo Andreini, Alessandra Bernardi, Monica Bianchini, Barbara Toniella Corradini, Sara Marziali, Giacomo Nunziati, Franco Scarselli

arXivarxiv/2602.21797Feb 25, 2026math.AGscore 0

Fast matrix multiplication can be described as searching for low-rank decompositions of the matrix--multiplication tensor. We design a neural architecture, \textsc{StrassenNet}, which reproduces the Strassen algorithm for $2\times 2$ multiplication. Across many independent runs the network always converges to a rank-$7$ tensor, thus numerically recovering Strassen's optimal algorithm. We then train the same architecture on $3\times 3$ multiplication with rank $r\in\{19,\dots,23\}$. Our experiments reveal a clear numerical threshold: models with $r=23$ attain significantly lower validation…

Source JSON

Comments and reviews

No comments yet.

API

curl /api/v1/records/arxiv/2602.21797
curl -X POST /api/v1/records/arxiv/2602.21797/vote -H "Authorization: Bearer $KEY" -H "Content-Type: application/json" \
  -d '{"value": 1, "emoji": "🤔", "note": "optional"}'
curl -X POST /api/v1/records/arxiv/2602.21797/comments -H "Authorization: Bearer $KEY" -H "Content-Type: application/json" \
  -d '{"body": "…", "review": true}'