mlsanityData to 7 Oct 2026

Test-Time Registers as Global Priors for Tokenized Image Generation

arXiv 2607.16824 ↗First posted 18 Jul 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Attention-based models often develop attention sinks, where a small number of tokens repeatedly attract attention and accumulate unusually large activations. In vision transformers, these outliers are closely related to registers, which have been diagnostically linked to global, low-frequency image structure. Existing work has largely studied registers through interpretability analyses and linear probes, leaving open whether they can be operationalized as plug-and-play signals for generation without retraining. We revisit this question in tokenized image generation. Using OpenCLIP and DINOv2 on ImageNet, we find that test-time register features exhibit stronger low-frequency concentration than both [CLS] readouts and patch-mean features, and show a consistent (albeit moderate) correlation with pixel-space DCT low-frequency energy. Motivated by these diagnostics, we introduce RegToken, a training-free procedure that converts register structure into a small set of global prior tokens by (i) NFN-based layer localization, (ii) TokenRank-guided subspace extraction, and (iii) a projection-and-conservation update on the register subspace. Inserted into a frozen compact 1D token generation pipeline, RegToken improves ImageNet generation and alignment metrics (e.g., FID-5k 20.5 to 20.1, SigLIP 3.6 to 3.9) without modifying pretrained weights, and accelerates test-time optimization (Steps@τ 74 to 52). Overall, our results suggest that structures often viewed as attention artifacts can be repurposed as lightweight global priors for tokenized generation.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 33 of its 36 references that are in the map. First posted on arXiv 18 Jul 2026.

Builds on

33 of 36 references.

References · 33

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  2. 2020Language Models are Few-Shot Learnerscited by 32,199
  3. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  4. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
  5. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
  6. 2025DINOv3cited by 1,297
  7. 2022Reproducible scaling laws for contrastive language-image learningcited by 1,163
  8. 2022MaskGIT: Masked Generative Image Transformercited by 1,082
  9. 2023Vision Transformers Need Registerscited by 844
  10. 2024An Image is Worth 32 Tokens for Reconstruction and Generationcited by 295
  11. 2024Massive Activations in Large Language Modelscited by 277
  12. 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 200
21 more
  1. 2025See What You Are Told: Visual Attention Sink in Large Multimodal Modelscited by 127
  2. 2025Democratizing Text-to-Image Masked Generative Models with Compact Text-Aware One-Dimensional Tokenscited by 48
  3. 2021Auto-Encoding Knowledge Graph for Unsupervised Medical Report Generationcited by 47
  4. 2024LoRA Land: 310 Fine-tuned LLMs that Rival GPT-4, A Technical Reportcited by 47
  5. 2025Vision Transformers Don't Need Trained Registerscited by 39
  6. 2024Language-Guided Image Tokenization for Generationcited by 36
  7. 2024Calibrating Multi-modal Representations: A Pursuit of Group Robustness without Annotationscited by 30
  8. 2025AToken: A Unified Tokenizer for Visioncited by 30
  9. 2024Cross-Modal Conditioned Reconstruction for Language-guided Medical Image Segmentationcited by 27
  10. 2025Beyond Matryoshka: Revisiting Sparse Coding for Adaptive Representationcited by 25
  11. 2025Uncovering Memorization Effect in the Presence of Spurious Correlationscited by 20
  12. 2024CoMA: Compositional Human Motion Generation with Multi-modal Agentscited by 18
  13. 2025To Sink or Not to Sink: Visual Information Pathways in Large Vision-Language Modelscited by 18
  14. 2025Highly Compressed Tokenizer Can Generate Without Trainingcited by 17
  15. 2022Aligning Source Visual and Target Language Domains for Unpaired Video Captioningcited by 14
  16. 2025PLoP: Precise LoRA Placement for Efficient Finetuning of Large Modelscited by 13
  17. 2025Attention (as Discrete-Time Markov) Chainscited by 11
  18. 2023Hybrid Neural Diffeomorphic Flow for Shape Representation and Generation via Triplanecited by 6
  19. 2026Let EEG Models Learn EEGcited by 6
  20. 2026CSRv2: Unlocking Ultra-Sparse Embeddingscited by 5
  21. 2026NeuroSonic: Conditional Flow Matching for EEG-to-Speech Reconstructioncited by 1

Built on by

Not cited yet.