ViTAR: Vision-Transformer mit beliebiger AuflösungViTAR: Vision Transformer with Any Resolution
Dieses Papier behandelt eine bedeutende Herausforderung, der Vision-Transformer (ViTs) gegenüberstehen: ihre eingeschränkte Skalierbarkeit über verschiedene Bildauflösungen hinweg. Typischerweise erleben ViTs einen Leistungsabfall, wenn sie Auflösungen verarbeiten, die sich von denen während des Trainings unterscheiden. Unsere Arbeit führt zwei wichtige Innovationen ein, um dieses Problem anzugehen. Erstens schlagen wir ein neuartiges Modul für die dynamische Auflösungsanpassung vor, das mit einem einzelnen Transformer-Block entworfen ist, speziell um eine äußerst effiziente inkrementelle Token-Integration zu erreichen. Zweitens führen wir unscharfe Positionsencodierung im Vision Transformer ein, um eine konsistente Positionskenntnis über mehrere Auflösungen hinweg bereitzustellen und damit Überanpassung an eine einzelne Trainingsauflösung zu verhindern. Unser resultierendes Modell, ViTAR (Vision Transformer mit beliebiger Auflösung), zeigt beeindruckende Anpassungsfähigkeit, erreicht eine Top-1 Genauigkeit von 83,3\% bei einer Auflösung von 1120x1120 und eine Genauigkeit von 80,4\% bei einer Auflösung von 4032x4032, und das alles bei reduzierten Rechenkosten. ViTAR zeigt auch starke Leistung bei nachgelagerten Aufgaben wie Instanz- und semantischer Segmentierung und kann problemlos mit selbstüberwachten Lerntechniken wie Masked AutoEncoder kombiniert werden. Unsere Arbeit bietet eine kostengünstige Lösung zur Verbesserung der Auflösungsskalierbarkeit von ViTs und ebnet den Weg für vielseitigere und effizientere Bildverarbeitung mit hoher Auflösung.