TTTIR: Unlocking Instance-Specific State Evolution via Test-Time Training for Image Restoration
Not on the map. Why?
Abstract (arXiv)
Image restoration is inherently challenging due to the diverse and highly input-dependent nature of real-world degradations. While recent architectures like Transformers and state-space models have advanced the field, they predominantly rely on static, globally shared parameters, which struggle to fully accommodate instance-specific degradation patterns. Test-Time Training (TTT) offers a promising paradigm for generating data-dependent operators, yet its standard self-supervised inner loop lacks the explicit guidance required to transition degraded features toward clean structures. To address this, we propose TTTIR, a novel framework that reformulates image restoration as an instance-specific state evolution process. Specifically, we design Progressive State Sequence Generation (PSSG) to construct complementary spatial-frequency target states (defining what to recover), and State Transition Evolution (STE) to adapt lightweight transition operators via a restoration-oriented TTT inner loop (determining how the features should evolve). Extensive experiments demonstrate that TTTIR consistently outperforms state-of-the-art models across multiple image restoration benchmarks, achieving dynamic instance-specific recovery with favorable computational scalability. The code is available at https://github.com/Elysiaaaaaaaa/TTTIR.git.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 39 of its 56 references that are in the map. First posted on arXiv 15 Sep 2026.
Builds on
39 of 56 references.
References · 39
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2021SwinIR: Image Restoration Using Swin Transformercited by 1,869
- 2021Uformer: A General U-Shaped Transformer for Image Restorationcited by 864
- 2022Simple Baselines for Image Restorationcited by 701
- 2019Kindling the Darkness: A Practical Low-light Image Enhancercited by 355
- 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Modelcited by 349
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 329
- 2022Vision Transformers for Single Image Dehazingcited by 255
- 2023Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancementcited by 228
- 2022Test-Time Training with Masked Autoencoderscited by 208
- 2023Learning A Sparse Transformer Network for Effective Image Derainingcited by 204
27 more
- 2025Test-Time Training Done Rightcited by 121
- 2024AdaIR: Adaptive All-in-One Image Restoration via Frequency Mining and Modulationcited by 108
- 2025One-Minute Video Generation with Test-Time Trainingcited by 100
- 2023DEA-Net: Single image dehazing based on detail-enhanced convolution and content-guided attentioncited by 90
- 2023Embedding Fourier for Ultra-High-Definition Low-Light Image Enhancementcited by 86
- 2023MB-TaylorFormer: Multi-branch Efficient Transformer Expanded by Taylor Formula for Image Dehazingcited by 81
- 2023FourLLIE: Boosting Low-Light Image Enhancement by Fourier Frequency Informationcited by 64
- 2024Retinexmamba: Retinex-based Mamba for Low-light Image Enhancementcited by 59
- 2024Bidirectional Multi-Scale Implicit Neural Representations for Image Derainingcited by 54
- 2024Wave-Mamba: Wavelet State Space Model for Ultra-High-Definition Low-Light Image Enhancementcited by 48
- 2026ZipMap: Linear-Time Stateful 3D Reconstruction via Test-Time Trainingcited by 23
- 2023MixDehazeNet : Mix Structure Block For Image Dehazing Networkcited by 18
- 2022Magic ELF: Image Deraining Meets Association Learning and Transformercited by 17
- 2024DMFourLLIE: Dual-Stage and Multi-Branch Fourier Network for Low-Light Image Enhancementcited by 15
- 2025ViT3: Unlocking Test-Time Training in Visioncited by 15
- 2022LKD-Net: Large Kernel Convolution Network for Single Image Dehazingcited by 13
- 2025EAMamba: Efficient All-Around Vision State Space Model for Image Restorationcited by 10
- 2025CWNet: Causal Wavelet Network for Low-Light Image Enhancementcited by 10
- 2026tttLRM: Test-Time Training for Long Context and Autoregressive 3D Reconstructioncited by 10
- 2026Spatial-TTT: Streaming Visual-based Spatial Intelligence with Test-Time Trainingcited by 8
- 2025Enhancing Image Restoration Transformer via Adaptive Translation Equivariancecited by 5
- 2026SF-Mamba: Rethinking State Space Model for Visioncited by 4
- 2026ATD: Improved Transformer with Adaptive Token Dictionary for Image Restorationcited by 3
- 2024Test-time Training for Hyperspectral Image Super-resolutioncited by 2
- 2025Cross Paradigm Representation and Alignment Transformer for Image Derainingcited by 1
- 2025Moir'eXNet: Adaptive Multi-Scale Demoir'eing with Linear Attention Test-Time Training and Truncated Flow Matching Priorcited by 1
- 2026Progressive Split Mamba: Effective State Space Modelling for Image Restorationcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2023Test-Time Degradation Adaptation for Open-Set Image Restoration
- 2025DPMambaIR: All-in-One Image Restoration via Degradation-Aware Prompt State Space Model
- 2026Continuous Expert Assembly: Instance-Conditioned Low-Rank Residuals for All-in-One Image Restoration
- 2026UnSCAR: Universal, Scalable, Controllable, and Adaptable Image Restoration
- 2023Parameter Efficient Adaptation for Image Restoration with Heterogeneous Mixture-of-Experts
- 2026Learning Adaptive Dynamical Features via Multi-τ Liquid-Mamba for All-in-one Image Restoration
- 2024Chain-of-Restoration: Multi-Task Image Restoration Models are Zero-Shot Step-by-Step Universal Image Restorers
- 2026NanoSD: Edge Efficient Foundation Model for Real Time Image Restoration