Convolutions and More as Einsum: A Tensor Network Perspective with Advances for Second-Order Methodscs.LG
Simplifies convolutions using tensor networks and einsum for efficient second-order methods.
problem Complexity in analyzing and applying convolutions in deep learning.
method Viewing convolutions as tensor networks, drawing diagrams, and using einsum for efficient computation.
result Accelerates a KFAC variant up to 4.5x with reduced memory overhead.