TEFormer: Structured Bidirectional Temporal Enhancement Modeling in Spiking Transformers

· Source: cs.NE updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Emerging Technologies & Innovation · Depth: Expert, extended

Summary

TEFormer is a novel Spiking Transformer framework designed to enhance temporal modeling by implementing structured bidirectional temporal fusion. Inspired by the human visual pathway, TEFormer decouples temporal processing into two core components: a lightweight, hyperparameter-free Temporal Enhanced Attention (TEA) module for parallel forward temporal fusion in the attention mechanism, and a backward gated recurrent Temporal MLP (T-MLP) for reverse-order temporal information aggregation. This approach addresses limitations in existing Spiking Transformers' ability to exploit spatiotemporal dependencies. Extensive experiments demonstrate TEFormer's consistent and significant outperformance against strong SNN and Spiking Transformer baselines across static (e.g., >96% on CIFAR10), neuromorphic (e.g., 81.90% on CIFAR10-DVS, 95.95% on NCARS), and temporally complicated datasets. Its performance gains remain stable across various neural encoding schemes, establishing TEFormer as an effective and general framework for temporal modeling.

Key takeaway

For Machine Learning Engineers developing Spiking Neural Networks for sequence modeling, you should consider implementing TEFormer's structured bidirectional temporal fusion. This approach, which decouples forward and backward temporal processing, consistently improves performance across static, neuromorphic, and complex temporal datasets, even with varied neuron encoding schemes. Adopting this method can significantly enhance your SNNs' ability to capture spatiotemporal dependencies efficiently, offering a robust path to higher accuracy in energy-constrained applications.

Key insights

TEFormer introduces structured bidirectional temporal fusion to Spiking Transformers, significantly improving spatiotemporal dependency exploitation.

Principles

Method

TEFormer integrates Temporal Enhanced Attention (TEA) for parallel forward fusion using a single learnable scalar and a temporal mask matrix. It also uses a Temporal MLP (T-MLP) with a gated recurrent update for backward temporal tracing, starting from the last time step.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Hardware Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.NE updates on arXiv.org.