Physics-inspired Energy Transition Neural Network for Sequence Learning
Zhou Wu, Junyi An, Baile Xu, Furao Shen, Jian Zhao
TL;DR
PETNN addresses long-term dependency challenges in sequence learning by grounding a recurrent architecture in physics energy transition concepts. It introduces the Remaining Time $T_t$, Cell State $C_t$, and Hidden State $S_t$, with updates driven by energy injection and a self-selective information mixing mechanism. Empirical results on time-series forecasting and text sentiment classification show PETNN competitive with Transformer-based models and often superior, while requiring lower computational resources due to its recurrence. This physics-inspired approach offers a memory-centric, generalizable foundation for future recurrence-based architectures beyond Transformer-dominated regimes.
Abstract
Recently, the superior performance of Transformers has made them a more robust and scalable solution for sequence modeling than traditional recurrent neural networks (RNNs). However, the effectiveness of Transformer in capturing long-term dependencies is primarily attributed to their comprehensive pair-modeling process rather than inherent inductive biases toward sequence semantics. In this study, we explore the capabilities of pure RNNs and reassess their long-term learning mechanisms. Inspired by the physics energy transition models that track energy changes over time, we propose a effective recurrent structure called the``Physics-inspired Energy Transition Neural Network" (PETNN). We demonstrate that PETNN's memory mechanism effectively stores information over long-term dependencies. Experimental results indicate that PETNN outperforms transformer-based methods across various sequence tasks. Furthermore, owing to its recurrent nature, PETNN exhibits significantly lower complexity. Our study presents an optimal foundational recurrent architecture and highlights the potential for developing effective recurrent neural networks in fields currently dominated by Transformer.
