mlsanityData to 7 Oct 2026

TTTIR: Unlocking Instance-Specific State Evolution via Test-Time Training for Image Restoration

arXiv 2609.26151 ↗First posted 15 Sep 2026eess.IV cs.AI cs.CVcited by 0

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

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  2. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
  3. 2021SwinIR: Image Restoration Using Swin Transformercited by 1,869
  4. 2021Uformer: A General U-Shaped Transformer for Image Restorationcited by 864
  5. 2022Simple Baselines for Image Restorationcited by 701
  6. 2019Kindling the Darkness: A Practical Low-light Image Enhancercited by 355
  7. 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Modelcited by 349
  8. 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 329
  9. 2022Vision Transformers for Single Image Dehazingcited by 255
  10. 2023Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancementcited by 228
  11. 2022Test-Time Training with Masked Autoencoderscited by 208
  12. 2023Learning A Sparse Transformer Network for Effective Image Derainingcited by 204
27 more
  1. 2025Test-Time Training Done Rightcited by 121
  2. 2024AdaIR: Adaptive All-in-One Image Restoration via Frequency Mining and Modulationcited by 108
  3. 2025One-Minute Video Generation with Test-Time Trainingcited by 100
  4. 2023DEA-Net: Single image dehazing based on detail-enhanced convolution and content-guided attentioncited by 90
  5. 2023Embedding Fourier for Ultra-High-Definition Low-Light Image Enhancementcited by 86
  6. 2023MB-TaylorFormer: Multi-branch Efficient Transformer Expanded by Taylor Formula for Image Dehazingcited by 81
  7. 2023FourLLIE: Boosting Low-Light Image Enhancement by Fourier Frequency Informationcited by 64
  8. 2024Retinexmamba: Retinex-based Mamba for Low-light Image Enhancementcited by 59
  9. 2024Bidirectional Multi-Scale Implicit Neural Representations for Image Derainingcited by 54
  10. 2024Wave-Mamba: Wavelet State Space Model for Ultra-High-Definition Low-Light Image Enhancementcited by 48
  11. 2026ZipMap: Linear-Time Stateful 3D Reconstruction via Test-Time Trainingcited by 23
  12. 2023MixDehazeNet : Mix Structure Block For Image Dehazing Networkcited by 18
  13. 2022Magic ELF: Image Deraining Meets Association Learning and Transformercited by 17
  14. 2024DMFourLLIE: Dual-Stage and Multi-Branch Fourier Network for Low-Light Image Enhancementcited by 15
  15. 2025ViT3: Unlocking Test-Time Training in Visioncited by 15
  16. 2022LKD-Net: Large Kernel Convolution Network for Single Image Dehazingcited by 13
  17. 2025EAMamba: Efficient All-Around Vision State Space Model for Image Restorationcited by 10
  18. 2025CWNet: Causal Wavelet Network for Low-Light Image Enhancementcited by 10
  19. 2026tttLRM: Test-Time Training for Long Context and Autoregressive 3D Reconstructioncited by 10
  20. 2026Spatial-TTT: Streaming Visual-based Spatial Intelligence with Test-Time Trainingcited by 8
  21. 2025Enhancing Image Restoration Transformer via Adaptive Translation Equivariancecited by 5
  22. 2026SF-Mamba: Rethinking State Space Model for Visioncited by 4
  23. 2026ATD: Improved Transformer with Adaptive Token Dictionary for Image Restorationcited by 3
  24. 2024Test-time Training for Hyperspectral Image Super-resolutioncited by 2
  25. 2025Cross Paradigm Representation and Alignment Transformer for Image Derainingcited by 1
  26. 2025Moir'eXNet: Adaptive Multi-Scale Demoir'eing with Linear Attention Test-Time Training and Truncated Flow Matching Priorcited by 1
  27. 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

  1. 2023Test-Time Degradation Adaptation for Open-Set Image Restoration
  2. 2025DPMambaIR: All-in-One Image Restoration via Degradation-Aware Prompt State Space Model
  3. 2026Continuous Expert Assembly: Instance-Conditioned Low-Rank Residuals for All-in-One Image Restoration
  4. 2026UnSCAR: Universal, Scalable, Controllable, and Adaptable Image Restoration
  5. 2023Parameter Efficient Adaptation for Image Restoration with Heterogeneous Mixture-of-Experts
  6. 2026Learning Adaptive Dynamical Features via Multi-τ Liquid-Mamba for All-in-one Image Restoration
  7. 2024Chain-of-Restoration: Multi-Task Image Restoration Models are Zero-Shot Step-by-Step Universal Image Restorers
  8. 2026NanoSD: Edge Efficient Foundation Model for Real Time Image Restoration