Pith. sign in

REVIEW 8 cited by

Separable Self-attention for Mobile Vision Transformers

Not yet reviewed by Pith; the record is open.

This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.

SPECIMEN: schema-true, not a live event

T0 review · schema-true

One-sentence machine reading of the paper's core claim.

pith:XXXXXXXX · record.json · timestamp

arxiv 2206.02680 v1 pith:RPLVPQTU submitted 2022-06-06 cs.CV cs.AIcs.LG

classification cs.CVcs.AIcs.LG
keywords mobileself-attentionvisionmobilevittransformersclassificationcomplexitycomputing
verification ladder T0 review T1 audit T2 compute T3 formal
0 comments
abstract

Mobile vision transformers (MobileViT) can achieve state-of-the-art performance across several mobile vision tasks, including classification and detection. Though these models have fewer parameters, they have high latency as compared to convolutional neural network-based models. The main efficiency bottleneck in MobileViT is the multi-headed self-attention (MHA) in transformers, which requires $O(k^2)$ time complexity with respect to the number of tokens (or patches) $k$. Moreover, MHA requires costly operations (e.g., batch-wise matrix multiplication) for computing self-attention, impacting latency on resource-constrained devices. This paper introduces a separable self-attention method with linear complexity, i.e. $O(k)$. A simple yet effective characteristic of the proposed method is that it uses element-wise operations for computing self-attention, making it a good choice for resource-constrained devices. The improved model, MobileViTv2, is state-of-the-art on several mobile vision tasks, including ImageNet object classification and MS-COCO object detection. With about three million parameters, MobileViTv2 achieves a top-1 accuracy of 75.6% on the ImageNet dataset, outperforming MobileViT by about 1% while running $3.2\times$ faster on a mobile device. Our source code is available at: \url{https://github.com/apple/ml-cvnets}

Discussion (0). Sign in to comment.

Forward citations

Cited by 8 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score. Full citation record

  1. UltraViT: Latency-Optimized On-device Vision Encoder for Large Vision-Language Models

    cs.CV 2026-07 conditional novelty 6.0 of 10

    A heterogeneous pyramidal vision encoder plus two-stage dense-then-generative pre-training yields faster, stronger on-device LVLM vision encoding than FastVLM.

  2. CoAtNeXt:An Attention-Enhanced ConvNeXtV2-Transformer Hybrid Model for Gastric Tissue Classification

    cs.CV 2025-09 conditional novelty 5.0 of 10

    CoAtNeXt, a CoAtNet variant using ConvNeXtV2 blocks and CBAM attention, reports state-of-the-art accuracy on two gastric histopathology datasets.

  3. Lightweight Backbone Networks Only Require Adaptive Lightweight Self-Attention Mechanisms

    cs.CV 2025-08 conditional novelty 5.0 of 10

    LOLViT, a GhostNet-based lightweight backbone using adaptive window attention, reports CNN-like CPU speed with MobileViT-level accuracy.

  4. Exploring Superposition and Interference in State-of-the-Art Low-Parameter Vision Models

    cs.CV 2025-07 conditional novelty 5.0 of 10

    Vision networks whose bottlenecks avoid feature-map interference scale better than MobileNet-style designs at very low parameter counts, and the new NoDepth block demonstrates this on ImageNet.

  5. VCMamba: Bridging Convolutions with Multi-Directional Mamba for Efficient Visual Representation

    cs.CV 2025-09 conditional novelty 4.0 of 10

    VCMamba reports that using convolutional feed-forward blocks for the first three stages followed by multi-directional Mamba blocks in the final stage yields 82.6% ImageNet-1K and 47.1 ADE20K mIoU at 31.5M parameters, ...

  6. Foundation Models and Transformers for Anomaly Detection: A Survey

    cs.LG 2025-07 reject novelty 4.0 of 10

    A taxonomy and literature review of Transformer-based visual anomaly detection, compromised by fabricated citations with dummy arXiv IDs.

  7. LAID: Lightweight AI-Generated Image Detection in Spatial and Spectral Domains

    cs.CV 2025-07 conditional novelty 4.0 of 10

    A benchmark showing lightweight off-the-shelf models can exceed 99% accuracy on GenImage for AI-image detection, with spectral inputs and decision fusion, at a fraction of larger models' compute.

  8. DeepTraverse: A Depth-First Search Inspired Network for Algorithmic Visual Understanding

    cs.CV 2025-06 reject novelty 4.0 of 10

    DeepTraverse is a weight-tied residual network plus squeeze-and-excitation attention, framed as depth-first search, with claimed efficiency gains that rest on a questionable ImageNet subset comparison.

Pith tools