Published March 4, 2025
| Version v1
Journal article
Open
Temporal Contrastive Learning through implicit non-equilibrium memory
- 1. University of Chicago
- 2. Rain AI
Description
The backpropagation method has enabled transformative uses of neural networks. Alternatively, for energy-based models, local learning methods involving only nearby neurons offer benefits in terms of decentralized training, and allow for the possibility of learning in computationally-constrained substrates. One class of local learning methods contrasts the desired, clamped behavior with spontaneous, free behavior. However, directly contrasting free and clamped behaviors requires explicit memory. Here, we introduce 'Temporal Contrastive Learning', an approach that uses integral feedback in each learning degree of freedom to provide a simple form of implicit non-equilibrium memory. During training, free and clamped behaviors are shown in a sawtooth-like protocol over time. When combined with integral feedback dynamics, these alternating temporal protocols generate an implicit memory necessary for comparing free and clamped behaviors, broadening the range of physical and biological systems capable of contrastive learning. Finally, we show that non-equilibrium dissipation improves learning quality and determine a Landauer-like energy cost of contrastive learning through physical dynamics.
Data availability
The data generated in this study may be found in the following figshare database: https://doi.org/10.6084/m9.figshare.25057793.
Code for training neural networks to perform MNIST recognition can be found at https://github.com/falkma/ContrastiveMemory-Exp. Code for understanding tradeoffs between protocol time and dissipation can be found at https://github.com/atstrupp/ContrastiveMemory-SynapseAnalysis.
Files
Temporal-Contrastive-Learning-through-implicit-non-equilibrium-memory.pdf
Files
(3.1 MB)
| Name | Size | Download all |
|---|---|---|
|
md5:26c31cb05b516972d7bcf80a673a1a37
|
2.0 MB | Preview Download |
|
Article md5:6924a22a15f15a57f180c43e9930c08b |
1.1 MB | Preview Download |
Additional details
Identifiers
- DOI
- 10.1038/s41467-025-57043-x
- Other
- oai:uchicago.tind.io:14692
Funding
- National Science Foundation
- DMR-2239801
- NIGMS
- R35GM151211
- Schmidt Sciences
- Eric and Wendy Schmidt AI in Science Postdoctoral Fellowship
- Center for Living Systems, National Science Foundation
- 2317138