Easing Optimization Paths: a Circuit Perspective

Ambroise Odonnat, Wassim Bouaziz, Vivien Cabannes

Preprint: not peer reviewed. Results may change in later versions.

Abstract

Gradient descent is the method of choice for training large artificial intelligence systems. As these systems become larger, a better understanding of the mechanisms behind gradient training would allow us to alleviate compute costs and help steer these systems away from harmful behaviors. To that end, we suggest utilizing the circuit perspective brought forward by mechanistic interpretability. After laying out our intuition, we illustrate how it enables us to design a curriculum for efficient learning in a controlled setting. The code is available at \url{https://github.com/facebookresearch/pal}.

Abstract: arXiv descriptive metadata (CC0). Thank you to arXiv for use of its open access interoperability. This service was not reviewed or approved by, nor does it necessarily express or reflect the policies or opinions of, arXiv.

Record

Archive ID
arxiv:2501.02362
Published
Jan 4, 2025
arXiv
2501.02362
Categories
cs.LG, eess.SP, stat.ML
Provenance
arxiv