ビジョン トランスフォーマーは、コンピューター ビジョン タスクにおいて大きな可能性を示しています。最近の研究では、パフォーマンス向上のために空間トークン ミキサーを精緻化することに重点を置いています。ただし、適切に設計された一般的なアーキテクチャは、どの空間トークン ミキサーが装備されているかに関係なく、バックボーン全体のパフォーマンスを大幅に向上できることがわかりました。この論文では、ビジョン バックボーンの改良された一般アーキテクチャである UniNeXt を提案します。その有効性を検証するために、畳み込みモジュールとアテンション モジュールの両方を含む、さまざまな典型的および最新の設計で空間トークン ミキサーをインスタンス化します。最初に提案されたアーキテクチャと比較して、私たちの UniNeXt アーキテクチャはすべての空間トークン ミキサーのパフォーマンスを着実に向上させ、それらの間のパフォーマンス ギャップを狭めることができます。驚くべきことに、単純なローカル ウィンドウ アテンションを備えた当社の UniNeXt は、以前の最先端のものよりも優れたパフォーマンスを発揮します。興味深いことに、これらの空間トークン ミキサーのランキングも UniNeXt の下では変化しており、優れた空間トークン ミキサーが最適ではない一般的なアーキテクチャによって抑制される可能性があることを示唆しています。これは、ビジョン バックボーンの一般的なアーキテクチャに関する研究の重要性をさらに示しています。
Vision Transformers have shown great potential in computer vision tasks. Most recent works have focused on elaborating the spatial token mixer for performance gains. However, we observe that a well-designed general architecture can significantly improve the performance of the entire backbone, regardless of which spatial token mixer is equipped. In this paper, we propose UniNeXt, an improved general architecture for the vision backbone. To verify its effectiveness, we instantiate the spatial token mixer with various typical and modern designs, including both convolution and attention modules. Compared with the architecture in which they are first proposed, our UniNeXt architecture can steadily boost the performance of all the spatial token mixers, and narrows the performance gap among them. Surprisingly, our UniNeXt equipped with naive local window attention even outperforms the previous state-of-the-art. Interestingly, the ranking of these spatial token mixers also changes under our UniNeXt, suggesting that an excellent spatial token mixer may be stifled due to a suboptimal general architecture, which further shows the importance of the study on the general architecture of vision backbone.