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

UChicago Information

Division(s)
Physical Sciences Division
Department(s)
Physics