Table 4: Influence of the optimal grouping on Group Window Attention. Group size g

We provide a diagram in Figure 6 for an intuitive illustration of our method. ViT to obtain representations for each visible patch. Illustration of the Group Window Attention scheme with shifted windows. B.2 Group Window Attention scheme with shifted windows. We provide a Python implementation of the Dynamic-Programming-based Optimal Grouping algorithm in Algorithm 2. As we can see, the two components of the Optimal Grouping algorithm Note the padding operations are omitted here for simplicity.