New training losses make ViTs develop specialized spatial attention heads

Yuji Nozawa · hf · 2026-11-10

Kioxia researchers introduce two visual focusing losses that align a Vision Transformer's multi-head attention with image regions without labels, yielding specialized spatial attention heads and improving dense prediction tasks like zero-shot segmentation.

Original post →

More from Research

Research channel →