Probing Length Generalization in Mamba via Image Reconstruction
We explore Mamba's length generalization capabilities through a toy vision task in which Mamba is trained to reconstruct images from patch sequences.
@inproceedings{RathjensSchiewerWiskottEtAl2027,
author = {Rathjens, Jan and Schiewer, Robin and Wiskott, Laurenz and Subramoney, Anand},
title = {Probing Length Generalization in Mamba via Image Reconstruction},
booktitle = {Artificial Neural Networks and Machine Learning – ICANN 2026},
editor = {Pasa, Luca and Lintas, Alessandra and Tetko, Igor V. and Micheli, Alessio and Navarin, Nicolò and Villa, Alessandro E.P. and Tortorella, Domenico and Polato, Mirko},
pages = {235–246},
publisher = {Springer Nature Switzerland},
address = {Cham},
year = {2027},
}
@article{RathjensReyhanianKappelEtAl2026,
author = {Rathjens, Jan and Reyhanian, Shirin and Kappel, David and Wiskott, Laurenz},
title = {Understanding Transformer-Based Vision Models via Modular Feature Inversion},
journal = {Transactions on Machine Learning Research},
month = {September},
year = {2026},
}
@inproceedings{FayyazRathjensChengEtAl2026,
author = {Fayyaz, Z. and Rathjens, J. and Cheng, S. and Wiskott, L.},
title = {Learning What to Remember: Semantic Attention in Generative Episodic Memory Model},
booktitle = {Proceedings of the International Joint Conference on Neural Networks (IJCNN)},
year = {2026},
}
@misc{RathjensSchiewerWiskottEtAl2026,
author = {Rathjens, Jan and Schiewer, Robin and Wiskott, Laurenz and Subramoney, Anand},
title = {Probing Length Generalization in Mamba via Image Reconstruction},
year = {2026},
}
@misc{RathjensReyhanianKappelEtAl2025,
author = {Rathjens, Jan and Reyhanian, Shirin and Kappel, David and Wiskott, Laurenz},
title = {Understanding Transformer-based Vision Models through Inversion},
howpublished = {arXiv e-print},
month = {March},
year = {2025},
doi = {10.48550/arXiv.2412.06534},
}
@inproceedings{RathjensWiskott2024b,
author = {Rathjens, Jan and Wiskott, Laurenz},
title = {Antagonism between Classification and Reconstruction Processes in Deep Predictive Coding Networks},
booktitle = {ESANN 2024 proceesdings},
pages = {161–166},
publisher = {Ciaco - i6doc.com},
address = {Bruges (Belgium) and online},
year = {2024},
doi = {10.14428/esann/2024.ES2024-59},
}
@misc{RathjensWiskott2024,
author = {Rathjens, Jan and Wiskott, Laurenz},
title = {Classification and Reconstruction Processes in Deep Predictive Coding Networks: Antagonists or Allies?},
year = {2024},
}
| Lab courses | An Introduction to Python for Data Analysis |
| Lab courses | An Introduction to Python for Data Analysis |
| Lab courses | An Introduction to Python for Data Analysis |
| Lectures | Database Systems |
We explore Mamba's length generalization capabilities through a toy vision task in which Mamba is trained to reconstruct images from patch sequences.
In this study, we revisit feature inversion, introducing a novel, modular variation that enables significantly more efficient application of the technique. We demonstrate how our method can be systematically applied to the large-scale transformer-based vision models, Detection Transformer and Vision Transformer, and how reconstructed images can be qualitatively interpreted in a meaningful way.
We analyze the interaction between classification- and reconstruction-driven processes in deep neural networks, finding an antagonistic rather than a synergistic relationship.
The Institut für Neuroinformatik (INI) is a research unit of the Faculties of Computer Science and Medicine at the Ruhr-Universität Bochum. Its scientific goal is to understand the fundamental principles through which organisms generate behavior and cognition while linked to their environments through sensory and effector systems. Inspired by our insights into such natural cognitive systems, we seek new solutions to problems of information processing in artificial cognitive systems. We draw from a variety of disciplines that include experimental psychology and neurophysiology as well as machine learning, neural artificial intelligence, computer vision, and robotics.
Universitätsstr. 150, Building NB, Room 3/32
D-44801 Bochum, Germany
Tel: (+49) 234 32-28967
Fax: (+49) 234 32-14210