arXiv Machine Learning

Broadcast Product: Redefining Shape-aligned Element-wise Multiplication and Beyond

arXiv:2409. 17502v2 Announce Type: replace Abstract: Broadcast operations are widely used in scientific computing libraries, yet their mathematical formulation is often implicit and inconsistently represented in machine learning literature.

arXiv AI
Jul 10

Multi-agent Autoformalization of Tensor Network Theory

arXiv:2607. 07857v1 Announce Type: cross Abstract: We build a team of specialized large language-model agents and present an agent-driven workflow for research-level formalization in theoretical physics, with the autoformalization of the fundamental theorem of matrix-product states as a demonstration.

By Sirui Lu, Erickson Tjoa, J. Ignacio Cirac
arXiv Machine Learning
1d ago

A Typed Tensor Language for Shared-State Federated Computation

The paper introduces a typed tensor language designed for shared-state federated computations, distinguishing between client-partitioned data and globally available values through two tensor sorts. It demonstrates that typed one-round programs factor through shared tensors whose shapes are program-dependent but client-independent, and extends this construction to multi-round programs with persistent shared state. The language supports server-side first-order updates, curvature-block updates, and covers federated analytics and FedSGD, while excluding more complex schemes like multi-local-step FedAvg and persistent private client state.

By Theofilos Mailis, Theodore Papamarkou, Andreas Ktenidis, Kalliopi-Christina Despotidou, Konstantinos Filippopolitis, Yannis Foufoulas, Thanasis-Michail Karampatsis, Evdokia Mailli, Yannis Ioannidis
arXiv Machine Learning
Jun 25

Tensorion: A Tensor-Aware Generalization of the Muon Optimizer

arXiv:2606. 25975v1 Announce Type: new Abstract: Common first-order optimizers, such as Adam, implicitly treat each parameter block as an unstructured vector, which disregards the multilinear weight structure present in many modern machine learning models.

By Vladimir Bogachev, Vladimir Aletov, Alexander Molozhavenko, Sergei Kudriashov, Maxim Rakhuba